1 package org.opendaylight.controller.test.sal.binding.it;
3 import static org.junit.Assert.*;
5 import java.math.BigInteger;
6 import java.util.Collection;
8 import org.junit.Before;
10 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker;
11 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ConsumerContext;
12 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.ProviderContext;
13 import org.opendaylight.controller.sal.binding.api.BindingAwareBroker.RoutedRpcRegistration;
14 import org.opendaylight.controller.sal.binding.api.BindingAwareConsumer;
15 import org.opendaylight.controller.sal.binding.api.BindingAwareProvider.ProviderFunctionality;
16 import org.opendaylight.controller.sal.binding.api.BindingAwareProvider;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.AddFlowInput;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.AddFlowInputBuilder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.SalFlowService;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeContext;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
27 import org.opendaylight.yangtools.yang.binding.RpcService;
29 import com.google.inject.Inject;
31 import static org.mockito.Mockito.*;
33 public class RoutedServiceTest extends AbstractTest {
35 private SalFlowService first;
36 private SalFlowService second;
38 private SalFlowService consumerService;
40 private RoutedRpcRegistration<SalFlowService> firstReg;
41 private RoutedRpcRegistration<SalFlowService> secondReg;
44 public void setUp() throws Exception {
45 first = mock(SalFlowService.class,"First Flow Service");
46 second = mock(SalFlowService.class,"Second Flow Service");
50 public void testServiceRegistration() {
52 assertNotNull(getBroker());
54 BindingAwareProvider provider1 = new AbstractTestProvider() {
57 public void onSessionInitiated(ProviderContext session) {
58 assertNotNull(session);
59 firstReg = session.addRoutedRpcImplementation(SalFlowService.class, first);
64 * Register first provider, which register first implementation of
68 getBroker().registerProvider(provider1, getBundleContext());
69 assertNotNull("Registration should not be null", firstReg);
70 assertSame(first, firstReg.getInstance());
72 BindingAwareProvider provider2 = new AbstractTestProvider() {
75 public void onSessionInitiated(ProviderContext session) {
76 assertNotNull(session);
77 secondReg = session.addRoutedRpcImplementation(SalFlowService.class, second);
80 getBroker().registerProvider(provider2, getBundleContext());
81 assertNotNull("Registration should not be null", firstReg);
82 assertNotSame(secondReg, firstReg);
85 BindingAwareConsumer consumer = new BindingAwareConsumer() {
87 public void onSessionInitialized(ConsumerContext session) {
88 consumerService = session.getRpcService(SalFlowService.class);
91 broker.registerConsumer(consumer, getBundleContext());
93 assertNotNull("MD-SAL instance of Flow Service should be returned", consumerService);
94 assertNotSame("Provider instance and consumer instance should not be same.", first, consumerService);
96 NodeRef nodeOne = createNodeRef("foo:node:1");
100 * Provider 1 - register itself as provider for SalFlowService
103 firstReg.registerPath(NodeContext.class, nodeOne.getValue());
106 * Consumer creates addFlow Message for node one and sends
110 AddFlowInput firstMessage = createSampleAddFlow(nodeOne,1);
111 consumerService.addFlow(firstMessage);
114 * We verify if implementation of first provider received same
115 * message from MD-SAL.
118 verify(first).addFlow(firstMessage);
121 * Verifies that second instance was not invoked with first
125 verify(second, times(0)).addFlow(firstMessage);
128 * Second provider registers as provider for nodeTwo
131 NodeRef nodeTwo = createNodeRef("foo:node:2");
132 secondReg.registerPath(NodeContext.class, nodeTwo.getValue());
136 * Consumer sends message to nodeTwo for three times.
137 * Should be processed by second instance.
139 AddFlowInput secondMessage = createSampleAddFlow(nodeTwo,2);
140 consumerService.addFlow(secondMessage);
141 consumerService.addFlow(secondMessage);
142 consumerService.addFlow(secondMessage);
145 * We verify that second was invoked 3 times, with message
146 * two as parameter, first was invoked 0 times.
149 verify(second, times(3)).addFlow(secondMessage);
150 verify(first, times(0)).addFlow(secondMessage);
154 * First provider unregisters as implementation of FlowService
158 firstReg.unregisterPath(NodeContext.class, nodeOne.getValue());
162 * Second provider registers as implementation for FlowService
166 secondReg.registerPath(NodeContext.class, nodeOne.getValue());
169 * Consumer sends third message to Node 1, should be processed
170 * by second instance.
173 AddFlowInput thirdMessage = createSampleAddFlow(nodeOne,3);
174 consumerService.addFlow(thirdMessage);
177 * We verify that first provider was invoked 0 times,
178 * second provider 1 time.
180 verify(first,times(0)).addFlow(thirdMessage);
181 verify(second).addFlow(thirdMessage);
185 private static NodeRef createNodeRef(String string) {
186 NodeKey key = new NodeKey(new NodeId(string));
187 InstanceIdentifier<Node> path = InstanceIdentifier.builder().node(Nodes.class).node(Node.class, key)
190 return new NodeRef(path);
193 static AddFlowInput createSampleAddFlow(NodeRef node,int cookie) {
194 AddFlowInputBuilder ret = new AddFlowInputBuilder();
196 ret.setCookie(BigInteger.valueOf(cookie));