2 * Copyright (c) 2015 Inocybe Technologies 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
9 package org.opendaylight.ovsdb.southbound;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Matchers.anyString;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.times;
16 import static org.mockito.Mockito.verify;
17 import static org.mockito.Mockito.when;
19 import org.junit.Before;
20 import org.junit.Test;
21 import org.junit.runner.RunWith;
22 import org.mockito.Mock;
23 import org.mockito.Mockito;
24 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
25 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
26 import org.opendaylight.controller.md.sal.common.api.clustering.Entity;
27 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipCandidateRegistration;
28 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipListener;
29 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipListenerRegistration;
30 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipService;
31 import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipState;
32 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
33 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
34 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
35 import org.opendaylight.ovsdb.southbound.transactions.md.TransactionInvoker;
36 import org.opendaylight.ovsdb.southbound.transactions.md.TransactionInvokerImpl;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopologyBuilder;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
42 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
43 import org.powermock.api.mockito.PowerMockito;
44 import org.powermock.api.support.membermodification.MemberMatcher;
45 import org.powermock.api.support.membermodification.MemberModifier;
46 import org.powermock.core.classloader.annotations.PrepareForTest;
47 import org.powermock.modules.junit4.PowerMockRunner;
48 import org.powermock.reflect.Whitebox;
50 import com.google.common.base.Optional;
51 import com.google.common.util.concurrent.CheckedFuture;
53 @PrepareForTest({SouthboundProvider.class, InstanceIdentifier.class, LogicalDatastoreType.class})
54 @RunWith(PowerMockRunner.class)
55 public class SouthboundProviderTest {
56 @Mock private DataBroker db;
57 @Mock private OvsdbConnectionManager cm;
58 @Mock private TransactionInvoker txInvoker;
59 @Mock private OvsdbDataChangeListener ovsdbDataChangeListener;
60 @Mock private SouthboundProvider southboundProvider;
61 @Mock private EntityOwnershipService entityOwnershipService;
62 @Mock private EntityOwnershipCandidateRegistration registration;
63 @Mock private EntityOwnershipListener entityOwnershipListener;
66 public void setUp() throws Exception {
67 southboundProvider = PowerMockito.mock(SouthboundProvider.class, Mockito.CALLS_REAL_METHODS);
68 MemberModifier.field(SouthboundProvider.class, "cm").set(southboundProvider, cm);
69 MemberModifier.field(SouthboundProvider.class, "ovsdbDataChangeListener").set(southboundProvider, ovsdbDataChangeListener);
70 MemberModifier.field(SouthboundProvider.class, "db").set(southboundProvider, db);
71 MemberModifier.field(SouthboundProvider.class, "entityOwnershipService").set(southboundProvider, entityOwnershipService);
72 MemberModifier.field(SouthboundProvider.class, "registration").set(southboundProvider, registration);
76 public void testOnSessionInitiated() throws Exception {
77 ProviderContext session = mock(ProviderContext.class);
78 when(session.getSALService(DataBroker.class)).thenReturn(db);
79 TransactionInvokerImpl transactionInvokerImpl = mock(TransactionInvokerImpl.class);
80 PowerMockito.whenNew(TransactionInvokerImpl.class).withArguments(any(DataBroker.class)).thenReturn(transactionInvokerImpl);
81 PowerMockito.whenNew(OvsdbConnectionManager.class).withArguments(any(DataBroker.class), any(TransactionInvoker.class), any(EntityOwnershipService.class)).thenReturn(cm);
82 PowerMockito.whenNew(OvsdbDataChangeListener.class).withArguments(any(DataBroker.class), any(OvsdbConnectionManager.class)).thenReturn(ovsdbDataChangeListener);
84 when(entityOwnershipService.registerListener(anyString(), any(EntityOwnershipListener.class))).thenReturn(mock(EntityOwnershipListenerRegistration.class));
85 when(entityOwnershipService.registerCandidate(any(Entity.class))).thenReturn(registration);
86 EntityOwnershipState entityOwnershipState = mock(EntityOwnershipState.class);
87 when(entityOwnershipService.getOwnershipState(any(Entity.class))).thenReturn(Optional.of(entityOwnershipState));
89 southboundProvider.onSessionInitiated(session);
91 verify(entityOwnershipService, times(2)).registerListener(anyString(), any(EntityOwnershipListener.class));
92 verify(entityOwnershipService).registerCandidate(any(Entity.class));
95 @SuppressWarnings("unchecked")
97 public void testInitializeTopology() throws Exception {
98 InstanceIdentifier<NetworkTopology> path = mock(InstanceIdentifier.class);
99 PowerMockito.mockStatic(InstanceIdentifier.class);
100 when(InstanceIdentifier.create(NetworkTopology.class)).thenReturn(path);
102 CheckedFuture<Optional<NetworkTopology>, ReadFailedException> topology = mock(CheckedFuture.class);
103 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
104 when(db.newReadWriteTransaction()).thenReturn(transaction);
105 when(transaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(topology);
107 Optional<NetworkTopology> optNetTopo = mock(Optional.class);
108 when(topology.get()).thenReturn(optNetTopo);
109 when(optNetTopo.isPresent()).thenReturn(false);
110 NetworkTopologyBuilder ntb = mock(NetworkTopologyBuilder.class);
111 PowerMockito.whenNew(NetworkTopologyBuilder.class).withNoArguments().thenReturn(ntb);
112 NetworkTopology networkTopology = mock(NetworkTopology.class);
113 when(ntb.build()).thenReturn(networkTopology);
114 doNothing().when(transaction).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(NetworkTopology.class));
115 when(transaction.submit()).thenReturn(mock(CheckedFuture.class));
117 LogicalDatastoreType type = PowerMockito.mock(LogicalDatastoreType.class);
118 Whitebox.invokeMethod(southboundProvider, "initializeTopology", type);
122 @SuppressWarnings("unchecked")
124 public void testInitializeOvsdbTopology() throws Exception {
125 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
126 when(db.newReadWriteTransaction()).thenReturn(transaction);
128 //suppress calls to initializeTopology()
129 MemberModifier.suppress(MemberMatcher.method(SouthboundProvider.class, "initializeTopology", LogicalDatastoreType.class));
131 CheckedFuture<Optional<Topology>, ReadFailedException> ovsdbTp = mock(CheckedFuture.class);
132 when(transaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(ovsdbTp);
135 Optional<Topology> optTopo = mock(Optional.class);
136 when(ovsdbTp.get()).thenReturn(optTopo);
137 when(optTopo.isPresent()).thenReturn(false);
138 TopologyBuilder tpb = mock(TopologyBuilder.class);
139 PowerMockito.whenNew(TopologyBuilder.class).withNoArguments().thenReturn(tpb);
140 when(tpb.setTopologyId(any(TopologyId.class))).thenReturn(tpb);
141 Topology data = mock(Topology.class);
142 when(tpb.build()).thenReturn(data);
143 doNothing().when(transaction).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(Topology.class));
144 when(transaction.submit()).thenReturn(mock(CheckedFuture.class));
146 LogicalDatastoreType type = PowerMockito.mock(LogicalDatastoreType.class);
147 Whitebox.invokeMethod(southboundProvider, "initializeOvsdbTopology", type);
148 PowerMockito.verifyPrivate(southboundProvider).invoke("initializeTopology", any(LogicalDatastoreType.class));
149 verify(tpb).setTopologyId(any(TopologyId.class));
153 when(optTopo.isPresent()).thenReturn(false);
154 when(transaction.cancel()).thenReturn(true);
155 Whitebox.invokeMethod(southboundProvider, "initializeOvsdbTopology", type);
156 PowerMockito.verifyPrivate(southboundProvider, times(2)).invoke("initializeTopology", any(LogicalDatastoreType.class));