Merge "Factor common code"

[netvirt.git] / openstack / net-virt / src / test / java / org / opendaylight / ovsdb / openstack / netvirt / FWaasHandlerTest.java

diff --git a/openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/FWaasHandlerTest.java b/openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/FWaasHandlerTest.java
index 551640a2ff2ba4650b1f457d4c4c150356cca0b2..9640e603d0881896952e83739e4eb8967af05061 100644 (file)
--- a/openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/FWaasHandlerTest.java
+++ b/openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/FWaasHandlerTest.java
@@ -17,20 +17,35 @@ import static org.mockito.Mockito.verifyNoMoreInteractions;
 
 import java.net.HttpURLConnection;
 
+import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
 import org.mockito.InjectMocks;
 import org.mockito.Mockito;
-import org.opendaylight.neutron.spi.NeutronFirewall;
-import org.opendaylight.neutron.spi.NeutronFirewallPolicy;
-import org.opendaylight.neutron.spi.NeutronFirewallRule;
+import org.opendaylight.ovsdb.openstack.netvirt.translator.NeutronFirewall;
+import org.opendaylight.ovsdb.openstack.netvirt.translator.NeutronFirewallPolicy;
+import org.opendaylight.ovsdb.openstack.netvirt.translator.NeutronFirewallRule;
+import org.opendaylight.ovsdb.openstack.netvirt.api.EventDispatcher;
+import org.opendaylight.ovsdb.utils.servicehelper.ServiceHelper;
+import org.osgi.framework.ServiceReference;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
 
 
 /**
  * Unit test for {@link FWaasHandler}
  */
+@RunWith(PowerMockRunner.class)
+@PrepareForTest(ServiceHelper.class)
 public class FWaasHandlerTest {
 
-    @InjectMocks FWaasHandler fwaasHandler = mock(FWaasHandler.class, Mockito.CALLS_REAL_METHODS);
+    @InjectMocks FWaasHandler fwaasHandler;
+
+    @Before
+    public void setUp() {
+        fwaasHandler = mock(FWaasHandler.class, Mockito.CALLS_REAL_METHODS);
+    }
 
     @Test
     public void testCanCreateNeutronFirewall(){
@@ -127,4 +142,16 @@ public class FWaasHandlerTest {
 //        when(ev.getAction()).thenReturn(Action.ADD);
 //        fwaasHandler.processEvent(ev);
     }
+
+    @Test
+    public void testSetDependencies() {
+        EventDispatcher eventDispatcher = mock(EventDispatcher.class);
+
+        PowerMockito.mockStatic(ServiceHelper.class);
+        PowerMockito.when(ServiceHelper.getGlobalInstance(EventDispatcher.class, fwaasHandler)).thenReturn(eventDispatcher);
+
+        fwaasHandler.setDependencies(mock(ServiceReference.class));
+
+        assertEquals("Error, did not return the correct object", fwaasHandler.eventDispatcher, eventDispatcher);
+    }
 }