maybe the notification is already received

Author: ndyakov

maintnotifications/e2e/scenario_endpoint_types_test.go

@@ -60,18 +60,20 @@ func TestEndpointTypesPushNotifications(t *testing.T) {
 		logCollector.Clear()
 	}()
 
+	// Create fault injector
+	faultInjector, err := CreateTestFaultInjector()
+	if err != nil {
+		t.Fatalf("[ERROR] Failed to create fault injector: %v", err)
+	}
+
+
 	// Create client factory from configuration
 	factory, err := CreateTestClientFactory("standalone")
 	if err != nil {
 		t.Skipf("Enterprise cluster not available, skipping endpoint types test: %v", err)
 	}
 	endpointConfig := factory.GetConfig()
 
-	// Create fault injector
-	faultInjector, err := CreateTestFaultInjector()
-	if err != nil {
-		t.Fatalf("[ERROR] Failed to create fault injector: %v", err)
-	}
 
 	defer func() {
 		if dump {
@@ -230,7 +232,7 @@ func TestEndpointTypesPushNotifications(t *testing.T) {
 			p("[FI] Migrate action completed for %s: %+v", endpointTest.name, status)
 
 			// Wait for MIGRATING notification
-			match, found = logCollector.WaitForLogMatchFunc(func(s string) bool {
+			match, found = logCollector.MatchOrWaitForLogMatchFunc(func(s string) bool {
 				return strings.Contains(s, logs2.ProcessingNotificationMessage) && strings.Contains(s, "MIGRATING")
 			}, 60*time.Second)
 			if !found {

maintnotifications/e2e/scenario_push_notifications_test.go

@@ -182,7 +182,7 @@ func TestPushNotifications(t *testing.T) {
 		ef("Failed to trigger migrate action: %v", err)
 	}
 	go func() {
-		match, found = logCollector.WaitForLogMatchFunc(func(s string) bool {
+		match, found = logCollector.MatchOrWaitForLogMatchFunc(func(s string) bool {
 			return strings.Contains(s, logs2.ProcessingNotificationMessage) && strings.Contains(s, "MIGRATING")
 		}, 60*time.Second)
 		commandsRunner.Stop()

maintnotifications/e2e/scenario_timeout_configs_test.go

@@ -239,7 +239,7 @@ func TestTimeoutConfigurationsPushNotifications(t *testing.T) {
 			p("[FI] Migrate action completed for %s: %+v", timeoutTest.name, status)
 
 			// Wait for MIGRATING notification
-			match, found = logCollector.WaitForLogMatchFunc(func(s string) bool {
+			match, found = logCollector.MatchOrWaitForLogMatchFunc(func(s string) bool {
 				return strings.Contains(s, logs2.ProcessingNotificationMessage) && strings.Contains(s, "MIGRATING")
 			}, 60*time.Second)
 			if !found {

maintnotifications/e2e/scenario_tls_configs_test.go

@@ -182,7 +182,7 @@ func ТestTLSConfigurationsPushNotifications(t *testing.T) {
 			}
 
 			// Wait for MIGRATING notification
-			match, found := logCollector.WaitForLogMatchFunc(func(s string) bool {
+			match, found := logCollector.MatchOrWaitForLogMatchFunc(func(s string) bool {
 				return strings.Contains(s, logs2.ProcessingNotificationMessage) && strings.Contains(s, "MIGRATING")
 			}, 60*time.Second)
 			if !found {