66
77 "github.com/aws/amazon-ecs-agent/agent/dockerclient/dockerapi"
88 mock_dockerapi "github.com/aws/amazon-ecs-agent/agent/dockerclient/dockerapi/mocks"
9+ "github.com/aws/amazon-ecs-agent/ecs-agent/doctor"
910 "github.com/docker/docker/api/types"
1011 "github.com/golang/mock/gomock"
1112 "github.com/stretchr/testify/assert"
@@ -16,33 +17,33 @@ func TestNewDockerRuntimeHealthCheck(t *testing.T) {
1617 defer ctrl .Finish ()
1718 mockDockerClient := mock_dockerapi .NewMockDockerClient (ctrl )
1819 dockerRuntimeHealthCheck := NewDockerRuntimeHealthcheck (mockDockerClient )
19- assert .Equal (t , HealthcheckStatusInitializing , dockerRuntimeHealthCheck .Status )
20+ assert .Equal (t , doctor . HealthcheckStatusInitializing , dockerRuntimeHealthCheck .Status )
2021}
2122
2223func TestRunCheck (t * testing.T ) {
2324 testcases := []struct {
2425 name string
2526 dockerPingResponse * dockerapi.PingResponse
26- expectedStatus HealthcheckStatus
27- expectedLastStatus HealthcheckStatus
27+ expectedStatus doctor. HealthcheckStatus
28+ expectedLastStatus doctor. HealthcheckStatus
2829 }{
2930 {
3031 name : "empty checks" ,
3132 dockerPingResponse : & dockerapi.PingResponse {
3233 Response : & types.Ping {APIVersion : "test_api_version" },
3334 Error : nil ,
3435 },
35- expectedStatus : HealthcheckStatusOk ,
36- expectedLastStatus : HealthcheckStatusInitializing ,
36+ expectedStatus : doctor . HealthcheckStatusOk ,
37+ expectedLastStatus : doctor . HealthcheckStatusInitializing ,
3738 },
3839 {
3940 name : "all true checks" ,
4041 dockerPingResponse : & dockerapi.PingResponse {
4142 Response : nil ,
4243 Error : & dockerapi.DockerTimeoutError {},
4344 },
44- expectedStatus : HealthcheckStatusImpaired ,
45- expectedLastStatus : HealthcheckStatusInitializing ,
45+ expectedStatus : doctor . HealthcheckStatusImpaired ,
46+ expectedLastStatus : doctor . HealthcheckStatusInitializing ,
4647 },
4748 }
4849 ctrl := gomock .NewController (t )
@@ -66,9 +67,9 @@ func TestSetHealthCheckStatus(t *testing.T) {
6667 defer ctrl .Finish ()
6768 dockerClient := mock_dockerapi .NewMockDockerClient (ctrl )
6869 dockerRuntimeHealthCheck := NewDockerRuntimeHealthcheck (dockerClient )
69- healthCheckStatus := HealthcheckStatusOk
70+ healthCheckStatus := doctor . HealthcheckStatusOk
7071 dockerRuntimeHealthCheck .SetHealthcheckStatus (healthCheckStatus )
71- assert .Equal (t , HealthcheckStatusOk , dockerRuntimeHealthCheck .Status )
72+ assert .Equal (t , doctor . HealthcheckStatusOk , dockerRuntimeHealthCheck .Status )
7273}
7374
7475func TestSetHealthcheckStatusChange (t * testing.T ) {
@@ -78,21 +79,21 @@ func TestSetHealthcheckStatusChange(t *testing.T) {
7879 dockerRuntimeHealthcheck := NewDockerRuntimeHealthcheck (dockerClient )
7980
8081 // we should start in initializing status
81- assert .Equal (t , HealthcheckStatusInitializing , dockerRuntimeHealthcheck .Status )
82+ assert .Equal (t , doctor . HealthcheckStatusInitializing , dockerRuntimeHealthcheck .Status )
8283 initializationChangeTime := dockerRuntimeHealthcheck .GetStatusChangeTime ()
8384
8485 // we update to initializing again; our StatusChangeTime remains the same
85- dockerRuntimeHealthcheck .SetHealthcheckStatus (HealthcheckStatusInitializing )
86+ dockerRuntimeHealthcheck .SetHealthcheckStatus (doctor . HealthcheckStatusInitializing )
8687 updateChangeTime := dockerRuntimeHealthcheck .GetStatusChangeTime ()
87- assert .Equal (t , HealthcheckStatusInitializing , dockerRuntimeHealthcheck .Status )
88+ assert .Equal (t , doctor . HealthcheckStatusInitializing , dockerRuntimeHealthcheck .Status )
8889 assert .Equal (t , initializationChangeTime , updateChangeTime )
8990
9091 // add a sleep so we know time has elapsed between the initial status and status change time
9192 time .Sleep (1 * time .Millisecond )
9293
9394 // change status. This should change the update time too
94- dockerRuntimeHealthcheck .SetHealthcheckStatus (HealthcheckStatusOk )
95- assert .Equal (t , HealthcheckStatusOk , dockerRuntimeHealthcheck .Status )
95+ dockerRuntimeHealthcheck .SetHealthcheckStatus (doctor . HealthcheckStatusOk )
96+ assert .Equal (t , doctor . HealthcheckStatusOk , dockerRuntimeHealthcheck .Status )
9697 okChangeTime := dockerRuntimeHealthcheck .GetStatusChangeTime ()
9798 // have we updated our change time?
9899 assert .True (t , okChangeTime .After (initializationChangeTime ))
0 commit comments