Ensure tests wait for leadership movement

Signed-off-by: Kyle Liberti <kliberti.us@gmail.com>

Author: kyguy

cruise-control/src/test/java/com/linkedin/kafka/cruisecontrol/executor/ExecutorTest.java

@@ -869,8 +869,13 @@ private void executeAndVerifyProposals(AdminClient adminClient,
                   topic.partitions().get(tp.partition()).replicas().stream().anyMatch(rep -> r.brokerId().equals(rep.id())));
         }
       }
+
+      // Wait for leader movement
+      TopicDescription topicAfterLeaderMove = _cluster.waitForTopicMetadata(tp.topic(), Duration.ofSeconds(60),
+          topicDescription -> topicDescription.partitions().get(tp.partition()).leader().id()
+            == proposal.newLeader().brokerId());
       assertEquals("The leader should have moved for " + tp,
-                   proposal.newLeader().brokerId().intValue(), topic.partitions().get(tp.partition()).leader().id());
+                   proposal.newLeader().brokerId().intValue(), topicAfterLeaderMove.partitions().get(tp.partition()).leader().id());
 
     }
     if (isTriggeredByUserRequest) {