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.lib.OvsdbConnection;
36 import org.opendaylight.ovsdb.southbound.transactions.md.TransactionInvoker;
37 import org.opendaylight.ovsdb.southbound.transactions.md.TransactionInvokerImpl;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopologyBuilder;
40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
42 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
43 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
44 import org.powermock.api.mockito.PowerMockito;
45 import org.powermock.api.support.membermodification.MemberMatcher;
46 import org.powermock.api.support.membermodification.MemberModifier;
47 import org.powermock.core.classloader.annotations.PrepareForTest;
48 import org.powermock.modules.junit4.PowerMockRunner;
49 import org.powermock.reflect.Whitebox;
51 import com.google.common.base.Optional;
52 import com.google.common.util.concurrent.CheckedFuture;
54 @PrepareForTest({SouthboundProvider.class, InstanceIdentifier.class, LogicalDatastoreType.class})
55 @RunWith(PowerMockRunner.class)
56 public class SouthboundProviderTest {
57 @Mock private DataBroker db;
58 @Mock private OvsdbConnectionManager cm;
59 @Mock private TransactionInvoker txInvoker;
60 @Mock private OvsdbDataChangeListener ovsdbDataChangeListener;
61 @Mock private SouthboundProvider southboundProvider;
62 @Mock private EntityOwnershipService entityOwnershipService;
63 @Mock private EntityOwnershipCandidateRegistration registration;
64 @Mock private EntityOwnershipListener entityOwnershipListener;
67 public void setUp() throws Exception {
68 southboundProvider = PowerMockito.mock(SouthboundProvider.class, Mockito.CALLS_REAL_METHODS);
69 MemberModifier.field(SouthboundProvider.class, "cm").set(southboundProvider, cm);
70 MemberModifier.field(SouthboundProvider.class, "ovsdbDataChangeListener").set(southboundProvider, ovsdbDataChangeListener);
71 MemberModifier.field(SouthboundProvider.class, "db").set(southboundProvider, db);
72 MemberModifier.field(SouthboundProvider.class, "entityOwnershipService").set(southboundProvider, entityOwnershipService);
73 MemberModifier.field(SouthboundProvider.class, "registration").set(southboundProvider, registration);
77 public void testOnSessionInitiated() throws Exception {
78 ProviderContext session = mock(ProviderContext.class);
79 when(session.getSALService(DataBroker.class)).thenReturn(db);
80 TransactionInvokerImpl transactionInvokerImpl = mock(TransactionInvokerImpl.class);
81 PowerMockito.whenNew(TransactionInvokerImpl.class).withArguments(any(DataBroker.class)).thenReturn(transactionInvokerImpl);
82 PowerMockito.whenNew(OvsdbConnectionManager.class).withArguments(any(DataBroker.class), any(TransactionInvoker.class), any(EntityOwnershipService.class), any(OvsdbConnection.class)).thenReturn(cm);
83 PowerMockito.whenNew(OvsdbDataChangeListener.class).withArguments(any(DataBroker.class), any(OvsdbConnectionManager.class)).thenReturn(ovsdbDataChangeListener);
85 when(entityOwnershipService.registerListener(anyString(), any(EntityOwnershipListener.class))).thenReturn(mock(EntityOwnershipListenerRegistration.class));
86 when(entityOwnershipService.registerCandidate(any(Entity.class))).thenReturn(registration);
87 EntityOwnershipState entityOwnershipState = mock(EntityOwnershipState.class);
88 when(entityOwnershipService.getOwnershipState(any(Entity.class))).thenReturn(Optional.of(entityOwnershipState));
90 southboundProvider.onSessionInitiated(session);
92 verify(entityOwnershipService, times(2)).registerListener(anyString(), any(EntityOwnershipListener.class));
93 verify(entityOwnershipService).registerCandidate(any(Entity.class));
96 @SuppressWarnings("unchecked")
98 public void testInitializeTopology() throws Exception {
99 InstanceIdentifier<NetworkTopology> path = mock(InstanceIdentifier.class);
100 PowerMockito.mockStatic(InstanceIdentifier.class);
101 when(InstanceIdentifier.create(NetworkTopology.class)).thenReturn(path);
103 CheckedFuture<Optional<NetworkTopology>, ReadFailedException> topology = mock(CheckedFuture.class);
104 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
105 when(db.newReadWriteTransaction()).thenReturn(transaction);
106 when(transaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(topology);
108 Optional<NetworkTopology> optNetTopo = mock(Optional.class);
109 when(topology.get()).thenReturn(optNetTopo);
110 when(optNetTopo.isPresent()).thenReturn(false);
111 NetworkTopologyBuilder ntb = mock(NetworkTopologyBuilder.class);
112 PowerMockito.whenNew(NetworkTopologyBuilder.class).withNoArguments().thenReturn(ntb);
113 NetworkTopology networkTopology = mock(NetworkTopology.class);
114 when(ntb.build()).thenReturn(networkTopology);
115 doNothing().when(transaction).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(NetworkTopology.class));
116 when(transaction.submit()).thenReturn(mock(CheckedFuture.class));
118 LogicalDatastoreType type = PowerMockito.mock(LogicalDatastoreType.class);
119 Whitebox.invokeMethod(southboundProvider, "initializeTopology", type);
123 @SuppressWarnings("unchecked")
125 public void testInitializeOvsdbTopology() throws Exception {
126 ReadWriteTransaction transaction = mock(ReadWriteTransaction.class);
127 when(db.newReadWriteTransaction()).thenReturn(transaction);
129 //suppress calls to initializeTopology()
130 MemberModifier.suppress(MemberMatcher.method(SouthboundProvider.class, "initializeTopology", LogicalDatastoreType.class));
132 CheckedFuture<Optional<Topology>, ReadFailedException> ovsdbTp = mock(CheckedFuture.class);
133 when(transaction.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class))).thenReturn(ovsdbTp);
136 Optional<Topology> optTopo = mock(Optional.class);
137 when(ovsdbTp.get()).thenReturn(optTopo);
138 when(optTopo.isPresent()).thenReturn(false);
139 TopologyBuilder tpb = mock(TopologyBuilder.class);
140 PowerMockito.whenNew(TopologyBuilder.class).withNoArguments().thenReturn(tpb);
141 when(tpb.setTopologyId(any(TopologyId.class))).thenReturn(tpb);
142 Topology data = mock(Topology.class);
143 when(tpb.build()).thenReturn(data);
144 doNothing().when(transaction).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(Topology.class));
145 when(transaction.submit()).thenReturn(mock(CheckedFuture.class));
147 LogicalDatastoreType type = PowerMockito.mock(LogicalDatastoreType.class);
148 Whitebox.invokeMethod(southboundProvider, "initializeOvsdbTopology", type);
149 PowerMockito.verifyPrivate(southboundProvider).invoke("initializeTopology", any(LogicalDatastoreType.class));
150 verify(tpb).setTopologyId(any(TopologyId.class));
154 when(optTopo.isPresent()).thenReturn(false);
155 when(transaction.cancel()).thenReturn(true);
156 Whitebox.invokeMethod(southboundProvider, "initializeOvsdbTopology", type);
157 PowerMockito.verifyPrivate(southboundProvider, times(2)).invoke("initializeTopology", any(LogicalDatastoreType.class));