diff --git a/device/drain/drainer_test.go b/device/drain/drainer_test.go index 4c0abde1..c30ebed8 100644 --- a/device/drain/drainer_test.go +++ b/device/drain/drainer_test.go @@ -645,6 +645,7 @@ func testDrainFilter(t *testing.T, deviceTypeOne deviceInfo, deviceTypeTwo devic stopCalled = false stop = func() { + time.Sleep(time.Second) stopCalled = true }