Search in sources :

Example 1 with UnregisterCandidateLocal

use of org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal in project controller by opendaylight.

the class EntityOwnershipShardTest method testListenerRegistration.

@Test
public void testListenerRegistration() throws Exception {
    testLog.info("testListenerRegistration starting");
    ShardTestKit kit = new ShardTestKit(getSystem());
    ShardIdentifier leaderId = newShardId(LOCAL_MEMBER_NAME);
    ShardIdentifier peerId = newShardId(PEER_MEMBER_1_NAME);
    TestActorRef<TestEntityOwnershipShard> peer = actorFactory.createTestActor(TestEntityOwnershipShard.props(newShardBuilder(peerId, peerMap(leaderId.toString()), PEER_MEMBER_1_NAME)), peerId.toString());
    peer.underlyingActor().startDroppingMessagesOfType(ElectionTimeout.class);
    TestActorRef<EntityOwnershipShard> leader = actorFactory.createTestActor(newShardProps(leaderId, peerMap(peerId.toString()), LOCAL_MEMBER_NAME), leaderId.toString());
    ShardTestKit.waitUntilLeader(leader);
    String otherEntityType = "otherEntityType";
    final DOMEntity entity1 = new DOMEntity(ENTITY_TYPE, ENTITY_ID1);
    final DOMEntity entity2 = new DOMEntity(ENTITY_TYPE, ENTITY_ID2);
    final DOMEntity entity3 = new DOMEntity(ENTITY_TYPE, ENTITY_ID3);
    final DOMEntity entity4 = new DOMEntity(otherEntityType, ENTITY_ID3);
    DOMEntityOwnershipListener listener = mock(DOMEntityOwnershipListener.class);
    // Register listener
    leader.tell(new RegisterListenerLocal(listener, ENTITY_TYPE), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    // Register a couple candidates for the desired entity type and verify listener is notified.
    leader.tell(new RegisterCandidateLocal(entity1), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verify(listener, timeout(5000)).ownershipChanged(ownershipChange(entity1, false, true, true));
    leader.tell(new RegisterCandidateLocal(entity2), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verify(listener, timeout(5000)).ownershipChanged(ownershipChange(entity2, false, true, true));
    reset(listener);
    // Register another candidate for another entity type and verify listener is not notified.
    leader.tell(new RegisterCandidateLocal(entity4), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
    verify(listener, never()).ownershipChanged(ownershipChange(entity4));
    // Register remote candidate for entity1
    peer.tell(new RegisterCandidateLocal(entity1), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyCommittedEntityCandidate(leader, ENTITY_TYPE, entity1.getIdentifier(), PEER_MEMBER_1_NAME);
    // Unregister the local candidate for entity1 and verify listener is notified
    leader.tell(new UnregisterCandidateLocal(entity1), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verify(listener, timeout(5000)).ownershipChanged(ownershipChange(entity1, true, false, true));
    reset(listener);
    // Unregister the listener, add a candidate for entity3 and verify listener isn't notified
    leader.tell(new UnregisterListenerLocal(listener, ENTITY_TYPE), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    leader.tell(new RegisterCandidateLocal(entity3), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyOwner(leader, ENTITY_TYPE, entity3.getIdentifier(), LOCAL_MEMBER_NAME);
    Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
    verify(listener, never()).ownershipChanged(any(DOMEntityOwnershipChange.class));
    // Re-register the listener and verify it gets notified of currently owned entities
    reset(listener);
    leader.tell(new RegisterListenerLocal(listener, ENTITY_TYPE), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verify(listener, timeout(5000).times(2)).ownershipChanged(or(ownershipChange(entity2, false, true, true), ownershipChange(entity3, false, true, true)));
    Uninterruptibles.sleepUninterruptibly(300, TimeUnit.MILLISECONDS);
    verify(listener, never()).ownershipChanged(ownershipChange(entity4));
    verify(listener, times(1)).ownershipChanged(ownershipChange(entity1));
    testLog.info("testListenerRegistration ending");
}
Also used : DOMEntityOwnershipListener(org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipListener) RegisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.RegisterCandidateLocal) ShardIdentifier(org.opendaylight.controller.cluster.datastore.identifiers.ShardIdentifier) DOMEntity(org.opendaylight.mdsal.eos.dom.api.DOMEntity) RegisterListenerLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.RegisterListenerLocal) UnregisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal) ShardTestKit(org.opendaylight.controller.cluster.datastore.ShardTestKit) DOMEntityOwnershipChange(org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipChange) UnregisterListenerLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterListenerLocal) Test(org.junit.Test)

Example 2 with UnregisterCandidateLocal

use of org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal in project controller by opendaylight.

the class DistributedEntityOwnershipService method unregisterCandidate.

void unregisterCandidate(final DOMEntity entity) {
    LOG.debug("Unregistering candidate for {}", entity);
    executeLocalEntityOwnershipShardOperation(new UnregisterCandidateLocal(entity));
    registeredEntities.remove(entity);
}
Also used : UnregisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal)

Example 3 with UnregisterCandidateLocal

use of org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal in project controller by opendaylight.

the class DistributedEntityOwnershipServiceTest method testCloseCandidateRegistration.

@Test
public void testCloseCandidateRegistration() throws Exception {
    DistributedEntityOwnershipService service = spy(DistributedEntityOwnershipService.start(dataStore.getActorContext(), EntityOwnerSelectionStrategyConfig.newBuilder().build()));
    DOMEntity entity = new DOMEntity(ENTITY_TYPE, YangInstanceIdentifier.of(QNAME));
    DOMEntityOwnershipCandidateRegistration reg = service.registerCandidate(entity);
    verifyEntityOwnershipCandidateRegistration(entity, reg);
    verifyRegisterCandidateLocal(service, entity);
    reset(service);
    reg.close();
    UnregisterCandidateLocal unregCandidate = verifyMessage(service, UnregisterCandidateLocal.class);
    assertEquals("getEntity", entity, unregCandidate.getEntity());
    // Re-register - should succeed.
    reset(service);
    service.registerCandidate(entity);
    verifyRegisterCandidateLocal(service, entity);
    service.close();
}
Also used : UnregisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal) DOMEntityOwnershipCandidateRegistration(org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration) DOMEntity(org.opendaylight.mdsal.eos.dom.api.DOMEntity) Test(org.junit.Test)

Example 4 with UnregisterCandidateLocal

use of org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal in project controller by opendaylight.

the class EntityOwnershipShardTest method testOnUnregisterCandidateLocal.

@Test
public void testOnUnregisterCandidateLocal() throws Exception {
    testLog.info("testOnUnregisterCandidateLocal starting");
    ShardTestKit kit = new ShardTestKit(getSystem());
    TestActorRef<EntityOwnershipShard> shard = actorFactory.createTestActor(newLocalShardProps());
    ShardTestKit.waitUntilLeader(shard);
    DOMEntity entity = new DOMEntity(ENTITY_TYPE, ENTITY_ID1);
    // Register
    shard.tell(new RegisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyCommittedEntityCandidate(shard, ENTITY_TYPE, ENTITY_ID1, LOCAL_MEMBER_NAME);
    verifyOwner(shard, ENTITY_TYPE, ENTITY_ID1, LOCAL_MEMBER_NAME);
    // Unregister
    shard.tell(new UnregisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyOwner(shard, ENTITY_TYPE, ENTITY_ID1, "");
    // Register again
    shard.tell(new RegisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyCommittedEntityCandidate(shard, ENTITY_TYPE, ENTITY_ID1, LOCAL_MEMBER_NAME);
    verifyOwner(shard, ENTITY_TYPE, ENTITY_ID1, LOCAL_MEMBER_NAME);
    testLog.info("testOnUnregisterCandidateLocal ending");
}
Also used : UnregisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal) RegisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.RegisterCandidateLocal) ShardTestKit(org.opendaylight.controller.cluster.datastore.ShardTestKit) DOMEntity(org.opendaylight.mdsal.eos.dom.api.DOMEntity) Test(org.junit.Test)

Example 5 with UnregisterCandidateLocal

use of org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal in project controller by opendaylight.

the class EntityOwnershipShardTest method testOwnershipChanges.

@Test
public void testOwnershipChanges() throws Exception {
    testLog.info("testOwnershipChanges starting");
    final ShardTestKit kit = new ShardTestKit(getSystem());
    dataStoreContextBuilder.shardHeartbeatIntervalInMillis(100).shardElectionTimeoutFactor(2);
    ShardIdentifier leaderId = newShardId(LOCAL_MEMBER_NAME);
    ShardIdentifier peerId1 = newShardId(PEER_MEMBER_1_NAME);
    ShardIdentifier peerId2 = newShardId(PEER_MEMBER_2_NAME);
    TestActorRef<TestEntityOwnershipShard> peer1 = actorFactory.createTestActor(TestEntityOwnershipShard.props(newShardBuilder(peerId1, peerMap(leaderId.toString(), peerId2.toString()), PEER_MEMBER_1_NAME)), peerId1.toString());
    peer1.underlyingActor().startDroppingMessagesOfType(ElectionTimeout.class);
    TestActorRef<TestEntityOwnershipShard> peer2 = actorFactory.createTestActor(TestEntityOwnershipShard.props(newShardBuilder(peerId2, peerMap(leaderId.toString(), peerId1.toString()), PEER_MEMBER_2_NAME)), peerId2.toString());
    peer2.underlyingActor().startDroppingMessagesOfType(ElectionTimeout.class);
    TestActorRef<EntityOwnershipShard> leader = actorFactory.createTestActor(newShardProps(leaderId, peerMap(peerId1.toString(), peerId2.toString()), LOCAL_MEMBER_NAME), leaderId.toString());
    ShardTestKit.waitUntilLeader(leader);
    DOMEntity entity = new DOMEntity(ENTITY_TYPE, ENTITY_ID1);
    // Add a remote candidate
    peer1.tell(new RegisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyCommittedEntityCandidate(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_1_NAME);
    // Register local
    leader.tell(new RegisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    // Verify the remote candidate becomes owner
    verifyCommittedEntityCandidate(leader, entity.getType(), entity.getIdentifier(), LOCAL_MEMBER_NAME);
    verifyOwner(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_1_NAME);
    // Add another remote candidate and verify ownership doesn't change
    peer2.tell(new RegisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyCommittedEntityCandidate(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_2_NAME);
    Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
    verifyOwner(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_1_NAME);
    // Remove the second remote candidate and verify ownership doesn't change
    peer2.tell(new UnregisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyEntityCandidateRemoved(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_2_NAME);
    Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
    verifyOwner(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_1_NAME);
    // Remove the first remote candidate and verify the local candidate becomes owner
    peer1.tell(new UnregisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyEntityCandidateRemoved(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_1_NAME);
    verifyOwner(leader, entity.getType(), entity.getIdentifier(), LOCAL_MEMBER_NAME);
    // Add the second remote candidate back and verify ownership doesn't change
    peer2.tell(new RegisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyCommittedEntityCandidate(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_2_NAME);
    Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
    verifyOwner(leader, entity.getType(), entity.getIdentifier(), LOCAL_MEMBER_NAME);
    // Unregister the local candidate and verify the second remote candidate becomes owner
    leader.tell(new UnregisterCandidateLocal(entity), kit.getRef());
    kit.expectMsgClass(SuccessReply.class);
    verifyEntityCandidateRemoved(leader, entity.getType(), entity.getIdentifier(), LOCAL_MEMBER_NAME);
    verifyOwner(leader, entity.getType(), entity.getIdentifier(), PEER_MEMBER_2_NAME);
    testLog.info("testOwnershipChanges ending");
}
Also used : UnregisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal) RegisterCandidateLocal(org.opendaylight.controller.cluster.datastore.entityownership.messages.RegisterCandidateLocal) ShardTestKit(org.opendaylight.controller.cluster.datastore.ShardTestKit) ShardIdentifier(org.opendaylight.controller.cluster.datastore.identifiers.ShardIdentifier) DOMEntity(org.opendaylight.mdsal.eos.dom.api.DOMEntity) Test(org.junit.Test)

Aggregations

UnregisterCandidateLocal (org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal)5 Test (org.junit.Test)4 DOMEntity (org.opendaylight.mdsal.eos.dom.api.DOMEntity)4 ShardTestKit (org.opendaylight.controller.cluster.datastore.ShardTestKit)3 RegisterCandidateLocal (org.opendaylight.controller.cluster.datastore.entityownership.messages.RegisterCandidateLocal)3 ShardIdentifier (org.opendaylight.controller.cluster.datastore.identifiers.ShardIdentifier)2 RegisterListenerLocal (org.opendaylight.controller.cluster.datastore.entityownership.messages.RegisterListenerLocal)1 UnregisterListenerLocal (org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterListenerLocal)1 DOMEntityOwnershipCandidateRegistration (org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration)1 DOMEntityOwnershipChange (org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipChange)1 DOMEntityOwnershipListener (org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipListener)1