build: Replace clock library

keboola · Jan 6, 2025 · 95084a5 · 95084a5
1 parent e38c6b6
commit 95084a5
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/internal/pkg/service/stream/storage/metacleanup/metacleanup_test.go b/internal/pkg/service/stream/storage/metacleanup/metacleanup_test.go
@@ -93,10 +93,11 @@ func TestMetadataCleanup(t *testing.T) {
 	// -----------------------------------------------------------------------------------------------------------------
 	var doCleanup func()
 	{
+		startTime := clk.Now()
 		cleanupAttempt := 0
 		doCleanup = func() {
 			cleanupAttempt++
-			clk.Advance(cleanupInterval)
+			clk.Advance(startTime.Add(time.Duration(cleanupAttempt) * cleanupInterval).Sub(clk.Now()))
 			assert.EventuallyWithT(t, func(c *assert.CollectT) {
 				logger.AssertJSONMessages(c, `{"level":"info","message":"deleted \"%d\" files"}`)
 			}, 2*time.Second, 100*time.Millisecond)
@@ -234,9 +235,12 @@ func TestMetadataCleanup(t *testing.T) {
 	}
 
 	// Delete success job as it has finished
+	jobCleanupAttempt := 0
 	{
 		logger.Truncate()
-		clk.Advance(cleanupInterval)
+		startTime := clk.Now()
+		jobCleanupAttempt++
+		clk.Advance(startTime.Add(time.Duration(jobCleanupAttempt) * cleanupInterval).Sub(clk.Now()))
 		assert.EventuallyWithT(t, func(c *assert.CollectT) {
 			logger.AssertJSONMessages(c, `{"level":"info","message":"deleted \"%d\" jobs"}`)
 		}, 2*time.Second, 100*time.Millisecond)
@@ -325,9 +329,12 @@ func TestMetadataProcessingJobCleanup(t *testing.T) {
 	}
 
 	// Delete processing job as it has finished
+	jobCleanupAttempt := 0
 	{
 		logger.Truncate()
-		clk.Advance(cleanupInterval)
+		startTime := clk.Now()
+		jobCleanupAttempt++
+		clk.Advance(startTime.Add(time.Duration(jobCleanupAttempt) * cleanupInterval).Sub(clk.Now()))
 		assert.EventuallyWithT(t, func(c *assert.CollectT) {
 			logger.AssertJSONMessages(c, `{"level":"info","message":"deleted \"%d\" jobs"}`)
 		}, 2*time.Second, 100*time.Millisecond)