@@ -250,6 +250,17 @@ func TestTimeoutConfigurationsPushNotifications(t *testing.T) {
250
250
ef ("Failed to trigger migrate action for %s: %v" , timeoutTest .name , err )
251
251
}
252
252
253
+ // Wait for migration to complete
254
+ status , err = faultInjector .WaitForAction (ctx , migrateResp .ActionID ,
255
+ WithMaxWaitTime (240 * time .Second ),
256
+ WithPollInterval (2 * time .Second ),
257
+ )
258
+ if err != nil {
259
+ ef ("[FI] Migrate action failed for %s: %v" , timeoutTest .name , err )
260
+ }
261
+
262
+ p ("[FI] Migrate action completed for %s: %s" , timeoutTest .name , status .Status )
263
+
253
264
// Wait for MIGRATING notification
254
265
match , found = logCollector .WaitForLogMatchFunc (func (s string ) bool {
255
266
return strings .Contains (s , logs2 .ProcessingNotificationMessage ) && strings .Contains (s , "MIGRATING" )
@@ -260,15 +271,6 @@ func TestTimeoutConfigurationsPushNotifications(t *testing.T) {
260
271
migrateData := logs2 .ExtractDataFromLogMessage (match )
261
272
p ("MIGRATING notification received for %s: %v" , timeoutTest .name , migrateData )
262
273
263
- // Wait for migration to complete
264
- status , err = faultInjector .WaitForAction (ctx , migrateResp .ActionID ,
265
- WithMaxWaitTime (240 * time .Second ),
266
- WithPollInterval (2 * time .Second ),
267
- )
268
- if err != nil {
269
- ef ("[FI] Migrate action failed for %s: %v" , timeoutTest .name , err )
270
- }
271
- p ("[FI] Migrate action completed for %s: %s" , timeoutTest .name , status .Status )
272
274
273
275
// do a bind action
274
276
bindResp , err := faultInjector .TriggerAction (ctx , ActionRequest {
0 commit comments