2 * Copyright (c) 2016 Brocade Communications Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.md.sal.dom.clustering.impl;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12 import static org.mockito.Matchers.any;
13 import static org.mockito.Matchers.eq;
14 import static org.mockito.Mockito.doNothing;
15 import static org.mockito.Mockito.doReturn;
16 import static org.mockito.Mockito.doThrow;
17 import static org.mockito.Mockito.mock;
18 import static org.mockito.Mockito.verify;
20 import com.google.common.base.Optional;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.mockito.ArgumentCaptor;
24 import org.mockito.Mock;
25 import org.mockito.MockitoAnnotations;
26 import org.opendaylight.controller.md.sal.common.api.clustering.CandidateAlreadyRegisteredException;
27 import org.opendaylight.controller.md.sal.common.api.clustering.Entity;
28 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipCandidateRegistration;
29 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipChange;
30 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipListener;
31 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipListenerRegistration;
32 import org.opendaylight.mdsal.eos.common.api.EntityOwnershipChangeState;
33 import org.opendaylight.mdsal.eos.common.api.EntityOwnershipState;
34 import org.opendaylight.mdsal.eos.dom.api.DOMEntity;
35 import org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipCandidateRegistration;
36 import org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipChange;
37 import org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipListener;
38 import org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipListenerRegistration;
39 import org.opendaylight.mdsal.eos.dom.api.DOMEntityOwnershipService;
42 * Unit tests for PreBoronEntityOwnershipServiceAdapter.
44 * @author Thomas Pantelis
47 public class LegacyEntityOwnershipServiceAdapterTest {
48 static Entity LEGACY_ENTITY = new Entity("foo", "bar");
49 static DOMEntity DOM_ENTITY = new DOMEntity("foo", LEGACY_ENTITY.getId());
52 private DOMEntityOwnershipService mockDOMService;
54 private LegacyEntityOwnershipServiceAdapter adapter;
58 MockitoAnnotations.initMocks(this);
59 adapter = new LegacyEntityOwnershipServiceAdapter(mockDOMService);
63 public void testRegisterCandidate() throws Exception {
64 DOMEntityOwnershipCandidateRegistration mockDOMReg = mock(DOMEntityOwnershipCandidateRegistration.class);
65 doNothing().when(mockDOMReg).close();
66 doReturn(mockDOMReg).when(mockDOMService).registerCandidate(DOM_ENTITY);
68 EntityOwnershipCandidateRegistration reg = adapter.registerCandidate(LEGACY_ENTITY);
70 assertNotNull("registerCandidate returned null", reg);
71 assertEquals("getInstance", LEGACY_ENTITY, reg.getInstance());
74 verify(mockDOMReg).close();
77 @Test(expected=CandidateAlreadyRegisteredException.class)
78 public void testAlreadyRegisteredCandidate() throws Exception {
79 doThrow(new org.opendaylight.mdsal.eos.common.api.CandidateAlreadyRegisteredException(DOM_ENTITY)).
80 when(mockDOMService).registerCandidate(DOM_ENTITY);
82 adapter.registerCandidate(LEGACY_ENTITY);
86 public void testRegisterListener() {
87 DOMEntityOwnershipListenerRegistration mockDOMReg = mock(DOMEntityOwnershipListenerRegistration.class);
88 doNothing().when(mockDOMReg).close();
89 doReturn(mockDOMReg).when(mockDOMService).registerListener(eq(DOM_ENTITY.getType()),
90 any(DOMEntityOwnershipListener.class));
91 EntityOwnershipListener mockListener = mock(EntityOwnershipListener.class);
92 doNothing().when(mockListener).ownershipChanged(any(EntityOwnershipChange.class));
94 EntityOwnershipListenerRegistration reg = adapter.registerListener(LEGACY_ENTITY.getType(), mockListener);
96 assertNotNull("registerListener returned null", reg);
97 assertEquals("getInstance", mockListener, reg.getInstance());
98 assertEquals("getEntityType", LEGACY_ENTITY.getType(), reg.getEntityType());
100 ArgumentCaptor<DOMEntityOwnershipListener> domListenerCaptor = ArgumentCaptor.forClass(DOMEntityOwnershipListener.class);
101 verify(mockDOMService).registerListener(eq(DOM_ENTITY.getType()), domListenerCaptor.capture());
103 DOMEntityOwnershipChange domOwnershipChange = new DOMEntityOwnershipChange(DOM_ENTITY,
104 EntityOwnershipChangeState.LOCAL_OWNERSHIP_GRANTED);
105 domListenerCaptor.getValue().ownershipChanged(domOwnershipChange );
107 ArgumentCaptor<EntityOwnershipChange> ownershipChangeCaptor = ArgumentCaptor.forClass(EntityOwnershipChange.class);
108 verify(mockListener).ownershipChanged(ownershipChangeCaptor.capture());
110 EntityOwnershipChange change = ownershipChangeCaptor.getValue();
111 assertEquals("getEntity", LEGACY_ENTITY, change.getEntity());
112 assertEquals("wasOwner", false, change.wasOwner());
113 assertEquals("isOwner", true, change.isOwner());
114 assertEquals("hasOwner", true, change.hasOwner());
117 verify(mockDOMReg).close();
121 public void testGetOwnershipState() {
122 testGetOwnershipState(EntityOwnershipState.IS_OWNER, true, true);
123 testGetOwnershipState(EntityOwnershipState.OWNED_BY_OTHER, false, true);
124 testGetOwnershipState(EntityOwnershipState.NO_OWNER, false, false);
126 doReturn(Optional.absent()).when(mockDOMService).getOwnershipState(DOM_ENTITY);
127 assertEquals("isPresent", false, adapter.getOwnershipState(LEGACY_ENTITY).isPresent());
131 public void testIsCandidateRegistered() {
132 doReturn(true).when(mockDOMService).isCandidateRegistered(DOM_ENTITY);
133 assertEquals("isCandidateRegistered", true, adapter.isCandidateRegistered(LEGACY_ENTITY));
136 private void testGetOwnershipState(EntityOwnershipState state, boolean expIsOwner, boolean expHasOwner) {
137 doReturn(Optional.of(state)).when(mockDOMService).getOwnershipState(DOM_ENTITY);
139 Optional<org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipState> actualState =
140 adapter.getOwnershipState(LEGACY_ENTITY);
142 assertEquals("isPresent", true, actualState.isPresent());
143 assertEquals("isOwner", expIsOwner, actualState.get().isOwner());
144 assertEquals("hasOwner", expHasOwner, actualState.get().hasOwner());