Search in sources :

Example 6 with DOMEntityOwnershipCandidateRegistration

use of org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration in project controller by opendaylight.

the class DistributedEntityOwnershipIntegrationTest method testFunctionalityWithThreeNodes.

@Test
public void testFunctionalityWithThreeNodes() throws Exception {
    String name = "testFunctionalityWithThreeNodes";
    MemberNode leaderNode = MemberNode.builder(memberNodes).akkaConfig("Member1").testName(name).moduleShardsConfig(MODULE_SHARDS_CONFIG).schemaContext(SCHEMA_CONTEXT).createOperDatastore(false).datastoreContextBuilder(leaderDatastoreContextBuilder).build();
    MemberNode follower1Node = MemberNode.builder(memberNodes).akkaConfig("Member2").testName(name).moduleShardsConfig(MODULE_SHARDS_CONFIG).schemaContext(SCHEMA_CONTEXT).createOperDatastore(false).datastoreContextBuilder(followerDatastoreContextBuilder).build();
    MemberNode follower2Node = MemberNode.builder(memberNodes).akkaConfig("Member3").testName(name).moduleShardsConfig(MODULE_SHARDS_CONFIG).schemaContext(SCHEMA_CONTEXT).createOperDatastore(false).datastoreContextBuilder(followerDatastoreContextBuilder).build();
    AbstractDataStore leaderDistributedDataStore = leaderNode.configDataStore();
    leaderDistributedDataStore.waitTillReady();
    follower1Node.configDataStore().waitTillReady();
    follower2Node.configDataStore().waitTillReady();
    final DOMEntityOwnershipService leaderEntityOwnershipService = newOwnershipService(leaderDistributedDataStore);
    final DOMEntityOwnershipService follower1EntityOwnershipService = newOwnershipService(follower1Node.configDataStore());
    final DOMEntityOwnershipService follower2EntityOwnershipService = newOwnershipService(follower2Node.configDataStore());
    leaderNode.kit().waitUntilLeader(leaderNode.configDataStore().getActorContext(), ENTITY_OWNERSHIP_SHARD_NAME);
    leaderEntityOwnershipService.registerListener(ENTITY_TYPE1, leaderMockListener);
    leaderEntityOwnershipService.registerListener(ENTITY_TYPE2, leaderMockListener2);
    follower1EntityOwnershipService.registerListener(ENTITY_TYPE1, follower1MockListener);
    // Register leader candidate for entity1 and verify it becomes owner
    leaderEntityOwnershipService.registerCandidate(ENTITY1);
    verify(leaderMockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY1, false, true, true));
    verify(follower1MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY1, false, false, true));
    reset(leaderMockListener, follower1MockListener);
    verifyGetOwnershipState(leaderEntityOwnershipService, ENTITY1, EntityOwnershipState.IS_OWNER);
    verifyGetOwnershipState(follower1EntityOwnershipService, ENTITY1, EntityOwnershipState.OWNED_BY_OTHER);
    // Register leader candidate for entity1_2 (same id, different type) and verify it becomes owner
    leaderEntityOwnershipService.registerCandidate(ENTITY1_2);
    verify(leaderMockListener2, timeout(5000)).ownershipChanged(ownershipChange(ENTITY1_2, false, true, true));
    Uninterruptibles.sleepUninterruptibly(300, TimeUnit.MILLISECONDS);
    verify(leaderMockListener, never()).ownershipChanged(ownershipChange(ENTITY1_2));
    reset(leaderMockListener2);
    // Register follower1 candidate for entity1 and verify it gets added but doesn't become owner
    follower1EntityOwnershipService.registerCandidate(ENTITY1);
    verifyCandidates(leaderDistributedDataStore, ENTITY1, "member-1", "member-2");
    verifyOwner(leaderDistributedDataStore, ENTITY1, "member-1");
    verifyOwner(follower2Node.configDataStore(), ENTITY1, "member-1");
    Uninterruptibles.sleepUninterruptibly(300, TimeUnit.MILLISECONDS);
    verify(leaderMockListener, never()).ownershipChanged(ownershipChange(ENTITY1));
    verify(follower1MockListener, never()).ownershipChanged(ownershipChange(ENTITY1));
    // Register follower1 candidate for entity2 and verify it becomes owner
    final DOMEntityOwnershipCandidateRegistration follower1Entity2Reg = follower1EntityOwnershipService.registerCandidate(ENTITY2);
    verify(follower1MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, false, true, true));
    verify(leaderMockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, false, false, true));
    verifyOwner(follower2Node.configDataStore(), ENTITY2, "member-2");
    reset(leaderMockListener, follower1MockListener);
    // Register follower2 candidate for entity2 and verify it gets added but doesn't become owner
    follower2EntityOwnershipService.registerListener(ENTITY_TYPE1, follower2MockListener);
    verify(follower2MockListener, timeout(5000).times(2)).ownershipChanged(or(ownershipChange(ENTITY1, false, false, true), ownershipChange(ENTITY2, false, false, true)));
    follower2EntityOwnershipService.registerCandidate(ENTITY2);
    verifyCandidates(leaderDistributedDataStore, ENTITY2, "member-2", "member-3");
    verifyOwner(leaderDistributedDataStore, ENTITY2, "member-2");
    // Unregister follower1 candidate for entity2 and verify follower2 becomes owner
    follower1Entity2Reg.close();
    verifyCandidates(leaderDistributedDataStore, ENTITY2, "member-3");
    verifyOwner(leaderDistributedDataStore, ENTITY2, "member-3");
    verify(follower1MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, true, false, true));
    verify(leaderMockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, false, false, true));
    // Depending on timing, follower2MockListener could get ownershipChanged with "false, false, true" if
    // if the original ownership change with "member-2 is replicated to follower2 after the listener is
    // registered.
    Uninterruptibles.sleepUninterruptibly(500, TimeUnit.MILLISECONDS);
    verify(follower2MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, false, true, true));
    // Register follower1 candidate for entity3 and verify it becomes owner
    follower1EntityOwnershipService.registerCandidate(ENTITY3);
    verifyOwner(leaderDistributedDataStore, ENTITY3, "member-2");
    verify(follower1MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY3, false, true, true));
    verify(follower2MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY3, false, false, true));
    verify(leaderMockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY3, false, false, true));
    // Register follower2 candidate for entity4 and verify it becomes owner
    follower2EntityOwnershipService.registerCandidate(ENTITY4);
    verifyOwner(leaderDistributedDataStore, ENTITY4, "member-3");
    verify(follower2MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY4, false, true, true));
    verify(follower1MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY4, false, false, true));
    verify(leaderMockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY4, false, false, true));
    reset(follower1MockListener, follower2MockListener);
    // Register follower1 candidate for entity4 and verify it gets added but doesn't become owner
    follower1EntityOwnershipService.registerCandidate(ENTITY4);
    verifyCandidates(leaderDistributedDataStore, ENTITY4, "member-3", "member-2");
    verifyOwner(leaderDistributedDataStore, ENTITY4, "member-3");
    // Shutdown follower2 and verify it's owned entities (entity 4) get re-assigned
    reset(leaderMockListener, follower1MockListener);
    follower2Node.cleanup();
    verify(follower1MockListener, timeout(15000)).ownershipChanged(ownershipChange(ENTITY4, false, true, true));
    verify(leaderMockListener, timeout(15000)).ownershipChanged(ownershipChange(ENTITY4, false, false, true));
    // Register leader candidate for entity2 and verify it becomes owner
    DOMEntityOwnershipCandidateRegistration leaderEntity2Reg = leaderEntityOwnershipService.registerCandidate(ENTITY2);
    verifyOwner(leaderDistributedDataStore, ENTITY2, "member-1");
    verify(leaderMockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, false, true, true));
    // Unregister leader candidate for entity2 and verify the owner is cleared
    leaderEntity2Reg.close();
    verifyOwner(leaderDistributedDataStore, ENTITY2, "");
    verify(leaderMockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, true, false, false));
    verify(follower1MockListener, timeout(5000)).ownershipChanged(ownershipChange(ENTITY2, false, false, false));
}
Also used : DOMEntityOwnershipService(org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipService) MemberNode(org.opendaylight.controller.cluster.datastore.MemberNode) DOMEntityOwnershipCandidateRegistration(org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration) AbstractDataStore(org.opendaylight.controller.cluster.datastore.AbstractDataStore) Test(org.junit.Test)

Example 7 with DOMEntityOwnershipCandidateRegistration

use of org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration 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 8 with DOMEntityOwnershipCandidateRegistration

use of org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration in project controller by opendaylight.

the class DistributedEntityOwnershipServiceTest method testRegisterCandidate.

@Test
public void testRegisterCandidate() throws Exception {
    DistributedEntityOwnershipService service = spy(DistributedEntityOwnershipService.start(dataStore.getActorContext(), EntityOwnerSelectionStrategyConfig.newBuilder().build()));
    YangInstanceIdentifier entityId = YangInstanceIdentifier.of(QNAME);
    DOMEntity entity = new DOMEntity(ENTITY_TYPE, entityId);
    DOMEntityOwnershipCandidateRegistration reg = service.registerCandidate(entity);
    verifyRegisterCandidateLocal(service, entity);
    verifyEntityOwnershipCandidateRegistration(entity, reg);
    verifyEntityCandidate(service.getLocalEntityOwnershipShard(), ENTITY_TYPE, entityId, dataStore.getActorContext().getCurrentMemberName().getName());
    try {
        service.registerCandidate(entity);
        fail("Expected CandidateAlreadyRegisteredException");
    } catch (CandidateAlreadyRegisteredException e) {
        // expected
        assertEquals("getEntity", entity, e.getEntity());
    }
    // Register a different entity - should succeed
    reset(service);
    DOMEntity entity2 = new DOMEntity(ENTITY_TYPE2, entityId);
    DOMEntityOwnershipCandidateRegistration reg2 = service.registerCandidate(entity2);
    verifyEntityOwnershipCandidateRegistration(entity2, reg2);
    verifyEntityCandidate(service.getLocalEntityOwnershipShard(), ENTITY_TYPE2, entityId, dataStore.getActorContext().getCurrentMemberName().getName());
    service.close();
}
Also used : CandidateAlreadyRegisteredException(org.opendaylight.mdsal.eos.common.api.CandidateAlreadyRegisteredException) DOMEntityOwnershipCandidateRegistration(org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration) DOMEntity(org.opendaylight.mdsal.eos.dom.api.DOMEntity) YangInstanceIdentifier(org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)8 DOMEntityOwnershipCandidateRegistration (org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration)8 AbstractDataStore (org.opendaylight.controller.cluster.datastore.AbstractDataStore)5 MemberNode (org.opendaylight.controller.cluster.datastore.MemberNode)5 DOMEntityOwnershipService (org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipService)5 DOMEntity (org.opendaylight.mdsal.eos.dom.api.DOMEntity)2 Failure (akka.actor.Status.Failure)1 UnregisterCandidateLocal (org.opendaylight.controller.cluster.datastore.entityownership.messages.UnregisterCandidateLocal)1 AddShardReplica (org.opendaylight.controller.cluster.datastore.messages.AddShardReplica)1 EntityOwnershipCandidateRegistration (org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipCandidateRegistration)1 CandidateAlreadyRegisteredException (org.opendaylight.mdsal.eos.common.api.CandidateAlreadyRegisteredException)1 EntityOwnershipState (org.opendaylight.mdsal.eos.common.api.EntityOwnershipState)1 DOMEntityOwnershipChange (org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipChange)1 YangInstanceIdentifier (org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier)1