Merge "Use ClusteredDataTreeListener in hwvtepsb"
[netvirt.git] / openstack / net-virt-providers / src / test / java / org / opendaylight / ovsdb / openstack / netvirt / providers / openflow13 / PipelineOrchestratorImplTest.java
1 /*
2  * Copyright (c) 2015 Inocybe Technologies.  All rights reserved.
3  *
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
7  */
8
9 package org.opendaylight.ovsdb.openstack.netvirt.providers.openflow13;
10
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertNull;
13 import static org.mockito.Matchers.anyString;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.when;
16
17 import java.lang.reflect.Field;
18 import java.util.concurrent.ExecutorService;
19
20 import org.junit.Test;
21 import org.junit.runner.RunWith;
22 import org.mockito.InjectMocks;
23 import org.mockito.Mock;
24 import org.opendaylight.ovsdb.openstack.netvirt.api.NodeCacheManager;
25 import org.opendaylight.ovsdb.openstack.netvirt.api.Southbound;
26 import org.opendaylight.ovsdb.utils.servicehelper.ServiceHelper;
27 import org.osgi.framework.BundleContext;
28 import org.osgi.framework.ServiceReference;
29 import org.powermock.api.mockito.PowerMockito;
30 import org.powermock.core.classloader.annotations.PrepareForTest;
31 import org.powermock.modules.junit4.PowerMockRunner;
32
33 /**
34  * Unit test for {@link PipelineOrchestratorImplTest}
35  */
36 @PrepareForTest(ServiceHelper.class)
37 @RunWith(PowerMockRunner.class)
38 public class PipelineOrchestratorImplTest {
39     @InjectMocks private PipelineOrchestratorImpl orchestrator;
40
41     @Mock private ExecutorService eventHandler;
42     @Mock private Southbound southbound;
43
44     /**
45      * Test for method {@link PipelineOrchestratorImpl#getTableOffset()}
46      */
47     @Test
48     public void testGetTableOffset() {
49         short tableOffset = 0;
50         assertEquals("tableOffset was not set", tableOffset, orchestrator.getTableOffset());
51     }
52
53     /**
54      * Test for {@link PipelineOrchestratorImpl#getTable(Service)}
55      */
56     @Test
57     public void testGetTableOffsetWithService() {
58         assertEquals("tableOffset was not set", Service.CLASSIFIER.getTable(),
59                 orchestrator.getTable(Service.CLASSIFIER));
60     }
61
62     /***
63      * Registers a mock service and verifies the registration by asking the
64      * pipeline orchestrator to return the associated service from its internal
65      * registry
66      */
67     @Test
68     public void testRegisterAndUnregisterService() {
69         Service service = Service.CLASSIFIER;
70         ServiceReference<?> serviceReference = mock(ServiceReference.class);
71         when(serviceReference.getProperty(anyString())).thenReturn(service);
72
73         AbstractServiceInstance abstractServiceInstance = mock(AbstractServiceInstance.class);
74
75         orchestrator.registerService(serviceReference, abstractServiceInstance);
76         assertEquals("Error, registerService() service registration fails",
77                 abstractServiceInstance,
78                 orchestrator.getServiceInstance(service));
79
80         orchestrator.unregisterService(serviceReference);
81         assertNull("Error, unregisterService() didn't delete the service", orchestrator.getServiceInstance(service));
82     }
83
84     /**
85      * Test method
86      * {@link PipelineOrchestratorImpl#getNextServiceInPipeline(Service)}
87      */
88     @Test
89     public void testGetNextServiceInPipeline() {
90
91         assertEquals(orchestrator.getNextServiceInPipeline(Service.CLASSIFIER),
92                 Service.ARP_RESPONDER);
93         assertEquals(
94                 orchestrator.getNextServiceInPipeline(Service.ARP_RESPONDER),
95                 Service.INBOUND_NAT);
96         assertEquals(
97                 orchestrator.getNextServiceInPipeline(Service.INBOUND_NAT),
98                 Service.EGRESS_ACL);
99         assertEquals(orchestrator.getNextServiceInPipeline(Service.EGRESS_ACL),
100                 Service.LOAD_BALANCER);
101         assertEquals(
102                 orchestrator.getNextServiceInPipeline(Service.LOAD_BALANCER),
103                 Service.ROUTING);
104         assertEquals(orchestrator.getNextServiceInPipeline(Service.ROUTING),
105                 Service.L3_FORWARDING);
106         assertEquals(
107                 orchestrator.getNextServiceInPipeline(Service.L3_FORWARDING),
108                 Service.L2_REWRITE);
109         assertEquals(orchestrator.getNextServiceInPipeline(Service.L2_REWRITE),
110                 Service.INGRESS_ACL);
111         assertEquals(
112                 orchestrator.getNextServiceInPipeline(Service.INGRESS_ACL),
113                 Service.OUTBOUND_NAT);
114         assertEquals(
115                 orchestrator.getNextServiceInPipeline(Service.OUTBOUND_NAT),
116                 Service.L2_FORWARDING);
117         assertNull(orchestrator.getNextServiceInPipeline(Service.L2_FORWARDING));
118     }
119
120   @Test
121   public void testSetDependencies() throws Exception {
122       NodeCacheManager nodeCacheManager = mock(NodeCacheManager.class);
123       Southbound southbound = mock(Southbound.class);
124
125       PowerMockito.mockStatic(ServiceHelper.class);
126       PowerMockito.when(ServiceHelper.getGlobalInstance(NodeCacheManager.class, orchestrator)).thenReturn(nodeCacheManager);
127       PowerMockito.when(ServiceHelper.getGlobalInstance(Southbound.class, orchestrator)).thenReturn(southbound);
128
129       orchestrator.setDependencies(mock(BundleContext.class), mock(ServiceReference.class));
130
131 //      assertEquals("Error, did not return the correct object", getField("nodeCacheManager"), nodeCacheManager);
132       assertEquals("Error, did not return the correct object", getField("southbound"), southbound);
133   }
134
135   private Object getField(String fieldName) throws Exception {
136       Field field = PipelineOrchestratorImpl.class.getDeclaredField(fieldName);
137       field.setAccessible(true);
138       return field.get(orchestrator);
139   }
140 }