- verifyRaftState(datastore, shardName, new RaftStateVerifier() {
- @Override
- public void verify(OnDemandRaftState raftState) {
- assertTrue(String.format("Peer(s) %s not found for shard %s. Actual: %s", peerIds, shardName,
- raftState.getPeerAddresses().keySet()),
- raftState.getPeerAddresses().keySet().containsAll(peerIds));
+ verifyRaftState(datastore, shardName, raftState -> assertEquals("Peers for shard " + shardName, peerIds,
+ raftState.getPeerAddresses().keySet()));
+ }
+
+ public static void verifyNoShardPresent(final AbstractDataStore datastore, final String shardName) {
+ Stopwatch sw = Stopwatch.createStarted();
+ while (sw.elapsed(TimeUnit.SECONDS) <= 5) {
+ Optional<ActorRef> shardReply = datastore.getActorContext().findLocalShard(shardName);
+ if (!shardReply.isPresent()) {
+ return;