Merge pull request hazelcast#13678 from metanet/fix/maint/13582

[BACKPORT] Fix MigrationInvocationsSafetyTest failure
sancar · Sep 3, 2018 · b9cb772 · b9cb772
2 parents 268d381 + 0cc690f
commit b9cb772
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/hazelcast/src/test/java/com/hazelcast/internal/partition/MigrationInvocationsSafetyTest.java b/hazelcast/src/test/java/com/hazelcast/internal/partition/MigrationInvocationsSafetyTest.java
@@ -275,6 +275,11 @@ private void setMigrationListenerToDropCommitResponse(final HazelcastInstance ma
         getPartitionServiceImpl(destination).setInternalMigrationListener(new InternalMigrationListener() {
             final AtomicReference<MigrationInfo> committedMigrationInfoRef = new AtomicReference<MigrationInfo>();
 
+            @Override
+            public void onMigrationStart(MigrationParticipant participant, MigrationInfo migrationInfo) {
+                assertClusterStateEventually(ClusterState.ACTIVE, destination);
+            }
+
             @Override
             public void onMigrationCommit(MigrationParticipant participant, MigrationInfo migrationInfo) {
                 if (participant == MigrationParticipant.DESTINATION && committedMigrationInfoRef.compareAndSet(null, migrationInfo)) {