fix flaky TestHandleEBSAckTimeout test for windows (#3904)

aws · Sep 20, 2023 · 19c3a32 · 19c3a32
1 parent e317379
commit 19c3a32
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agent/ebs/watcher_test.go b/agent/ebs/watcher_test.go
@@ -296,7 +296,7 @@ func TestHandleEBSAckTimeout(t *testing.T) {
 		apiebs.VolumeIdName:     volumeID,
 	}
 
-	expiresAt := time.Now().Add(time.Millisecond * 5)
+	expiresAt := time.Now().Add(time.Millisecond * testconst.WaitTimeoutMillis)
 	ebsAttachment := &apiebs.ResourceAttachment{
 		AttachmentInfo: attachmentinfo.AttachmentInfo{
 			TaskARN:              taskARN,
@@ -311,7 +311,7 @@ func TestHandleEBSAckTimeout(t *testing.T) {
 	watcher := newTestEBSWatcher(ctx, taskEngineState, eventChannel, mockDiscoveryClient)
 
 	watcher.HandleResourceAttachment(ebsAttachment)
-	time.Sleep(10 * time.Millisecond)
+	time.Sleep(time.Millisecond * testconst.WaitTimeoutMillis * 2)
 	assert.Len(t, taskEngineState.(*dockerstate.DockerTaskEngineState).GetAllEBSAttachments(), 0)
 	ebsAttachment, ok := taskEngineState.(*dockerstate.DockerTaskEngineState).GetEBSByVolumeId(volumeID)
 	assert.False(t, ok)