fix: unittest

0xPolygon · Nov 11, 2024 · aaf9254 · aaf9254
1 parent b49fc8e
commit aaf9254
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/aggsender/aggsender_test.go b/aggsender/aggsender_test.go
@@ -859,13 +859,8 @@ func TestCheckIfCertificatesAreSettled(t *testing.T) {
 				},
 			}
 
-			ctx, cancel := context.WithCancel(context.Background())
-			defer cancel()
-
-			go aggSender.checkIfCertificatesAreSettled(ctx)
-
-			time.Sleep(2 * time.Second)
-			cancel()
+			ctx := context.TODO()
+			aggSender.checkPendingCertificatesStatus(ctx)
 
 			mockAggLayerClient.AssertExpectations(t)
 			mockStorage.AssertExpectations(t)