@Test
public void testJoinsAndWatchesSurviveExpiredSession() throws Exception {
onLoseMembership.execute();
replay(onLoseMembership);
assertEmptyMembershipObserved();
Membership membership = joinGroup.join(onLoseMembership);
String originalMemberId = membership.getMemberId();
assertMembershipObserved(originalMemberId);
expireSession(zkClient);
// We should have lost our group membership and then re-gained it with a new ephemeral node.
// We may or may-not see the intermediate state change but we must see the final state
Iterable<String> members = listener.take();
if (Iterables.isEmpty(members)) {
members = listener.take();
}
assertEquals(1, Iterables.size(members));
assertNotEqual(originalMemberId, Iterables.getOnlyElement(members));
assertNotEqual(originalMemberId, membership.getMemberId());
listener.assertEmpty();
verify(onLoseMembership);
reset(onLoseMembership); // Turn off expectations during ZK server shutdown.
}
@Test
public void testRecover() throws Exception {
expect(snapshotStore.createSnapshot()).andReturn(SNAPSHOT1);
Capture<MutateWork<Object, Exception>> transaction = createCapture();
expect(primaryStorage.write(capture(transaction))).andReturn(null);
distributedStore.persist(SNAPSHOT1);
shutDownNow.execute();
control.replay();
assertEquals(ImmutableSet.<String>of(), recovery.listBackups());
clock.advance(INTERVAL);
storageBackup.createSnapshot();
String backup1 = storageBackup.createBackupName();
assertEquals(ImmutableSet.of(backup1), recovery.listBackups());
recovery.stage(backup1);
assertEquals(
IScheduledTask.setFromBuilders(SNAPSHOT1.getTasks()), recovery.query(Query.unscoped()));
recovery.commit();
transaction.getValue().apply(storeProvider);
}
@Test
public void testStopWatching() throws Exception {
replay(onLoseMembership);
assertEmptyMembershipObserved();
Membership member1 = joinGroup.join();
String memberId1 = member1.getMemberId();
assertMembershipObserved(memberId1);
Membership member2 = joinGroup.join();
String memberId2 = member2.getMemberId();
assertMembershipObserved(memberId1, memberId2);
stopWatching.execute();
member1.cancel();
Membership member3 = joinGroup.join();
member2.cancel();
member3.cancel();
listener.assertEmpty();
}