fix: timeout setting

myzhan · Sep 13, 2023 · 67d96b2 · 67d96b2
1 parent e54c24a
commit 67d96b2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/stats_test.go b/stats_test.go
@@ -160,6 +160,6 @@ var _ = Describe("Test states", func() {
 			error:        "500 error",
 		}
 
-		Eventually(newStats.messageToRunnerChan, slaveReportInterval+500*time.Millisecond).Should(Receive())
+		Eventually(newStats.messageToRunnerChan).WithTimeout(slaveReportInterval + 500*time.Millisecond).Should(Receive())
 	})
 })
diff --git a/utils_test.go b/utils_test.go
@@ -54,7 +54,7 @@ var _ = Describe("Test utils", func() {
 
 		err := StartMemoryProfile("mem.pprof", 2*time.Second)
 		Expect(err).NotTo(HaveOccurred())
-		Eventually("mem.pprof", 3*time.Second).Should(BeAnExistingFile())
+		Eventually("mem.pprof").WithTimeout(3 * time.Second).Should(BeAnExistingFile())
 	})
 
 	It("test start CPU profile", func() {
@@ -64,6 +64,6 @@ var _ = Describe("Test utils", func() {
 
 		err := StartCPUProfile("cpu.pprof", 2*time.Second)
 		Expect(err).NotTo(HaveOccurred())
-		Eventually("cpu.pprof", 3*time.Second).Should(BeAnExistingFile())
+		Eventually("cpu.pprof").WithTimeout(3 * time.Second).Should(BeAnExistingFile())
 	})
 })