engine: fix flakey TestRemoveEvents

since we "Emit a couple of events for the task before cleanup finishes." We should also add corresponding client.EXPECT().StopContainer()
aws · Jan 31, 2018 · 7f1c65d · 7f1c65d
1 parent 0c65868
commit 7f1c65d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/agent/engine/docker_task_engine_test.go b/agent/engine/docker_task_engine_test.go
@@ -470,6 +470,8 @@ func TestRemoveEvents(t *testing.T) {
 			eventStream <- createDockerEvent(api.ContainerStopped)
 			eventStream <- createDockerEvent(api.ContainerStopped)
 		}).Return(nil)
+
+	client.EXPECT().StopContainer(gomock.Any(), gomock.Any()).AnyTimes()
 	imageManager.EXPECT().RemoveContainerReferenceFromImageState(gomock.Any())
 
 	// This ensures that managedTask.waitForStopReported makes progress