Fix NPE triggered after disabling SG on a port
[netvirt.git] / openstack / net-virt / src / test / java / org / opendaylight / netvirt / openstack / netvirt / PortSecurityHandlerTest.java
1 /*
2  * Copyright (c) 2015, 2016 Inocybe and others.  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.netvirt.openstack.netvirt;
10
11 import static org.junit.Assert.assertEquals;
12 import static org.mockito.Matchers.any;
13 import static org.mockito.Mockito.mock;
14 import static org.mockito.Mockito.times;
15 import static org.mockito.Mockito.verify;
16
17 import java.lang.reflect.Field;
18 import java.net.HttpURLConnection;
19
20 import org.junit.Assert;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.junit.runner.RunWith;
24 import org.mockito.InjectMocks;
25 import org.mockito.Mock;
26 import org.mockito.Mockito;
27 import org.mockito.runners.MockitoJUnitRunner;
28 import org.opendaylight.netvirt.openstack.netvirt.api.EventDispatcher;
29 import org.opendaylight.netvirt.openstack.netvirt.translator.NeutronSecurityGroup;
30 import org.opendaylight.netvirt.openstack.netvirt.translator.NeutronSecurityRule;
31 import org.opendaylight.netvirt.utils.servicehelper.ServiceHelper;
32 import org.osgi.framework.ServiceReference;
33
34 /**
35  * Unit test {@link PortSecurityHandler}
36  */
37 @RunWith(MockitoJUnitRunner.class)
38 public class PortSecurityHandlerTest {
39
40     @InjectMocks private PortSecurityHandler portSecurityHandler;
41     private PortSecurityHandler posrtSecurityHandlerSpy;
42     @Mock
43     EventDispatcher eventDispatcher;
44
45     @Before
46     public void setUp() {
47         posrtSecurityHandlerSpy = Mockito.spy(portSecurityHandler);
48     }
49
50     @Test
51     public void testCanCreateNeutronSecurityGroup() {
52         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_CREATED, portSecurityHandler.canCreateNeutronSecurityGroup(mock(NeutronSecurityGroup.class)));
53
54         posrtSecurityHandlerSpy.neutronSecurityGroupCreated(any(NeutronSecurityGroup.class));
55         verify(posrtSecurityHandlerSpy, times(1)).canCreateNeutronSecurityGroup(any(NeutronSecurityGroup.class));
56     }
57
58     @Test
59     public void testCanUpdateNeutronSecurityGroup() {
60         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, portSecurityHandler.canUpdateNeutronSecurityGroup(mock(NeutronSecurityGroup.class), mock(NeutronSecurityGroup.class)));
61     }
62
63     @Test
64     public void testCanDeleteNeutronSecurityGroup() {
65         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, portSecurityHandler.canDeleteNeutronSecurityGroup(mock(NeutronSecurityGroup.class)));
66
67         posrtSecurityHandlerSpy.neutronSecurityGroupDeleted(any(NeutronSecurityGroup.class));
68         verify(posrtSecurityHandlerSpy, times(1)).canDeleteNeutronSecurityGroup(any(NeutronSecurityGroup.class));
69     }
70
71     @Test
72     public void testCanCreateNeutronSecurityRule() {
73         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_CREATED, portSecurityHandler.canCreateNeutronSecurityRule(mock(NeutronSecurityRule.class)));
74
75         posrtSecurityHandlerSpy.neutronSecurityRuleCreated(any(NeutronSecurityRule.class));
76         verify(posrtSecurityHandlerSpy, times(1)).enqueueEvent(any(AbstractEvent.class));
77     }
78
79     @Test
80     public void testCanUpdateNeutronSecurityRule() {
81         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, portSecurityHandler.canUpdateNeutronSecurityRule(mock(NeutronSecurityRule.class), mock(NeutronSecurityRule.class)));
82     }
83
84     @Test
85     public void testCanDeleteNeutronSecurityRule() {
86         assertEquals("Error, did not return the correct HTTP flag", HttpURLConnection.HTTP_OK, portSecurityHandler.canDeleteNeutronSecurityRule(mock(NeutronSecurityRule.class)));
87
88         posrtSecurityHandlerSpy.neutronSecurityRuleDeleted(any(NeutronSecurityRule.class));
89         verify(posrtSecurityHandlerSpy, times(1)).enqueueEvent(any(AbstractEvent.class));
90    }
91
92     @Test
93     public void testProcessEvent() {
94         // TODO see PortSecurityHandler#processEvent()
95     }
96
97     @Test
98     public void testSetDependencies() throws Exception {
99         EventDispatcher eventDispatcher = mock(EventDispatcher.class);
100
101         ServiceHelper.overrideGlobalInstance(EventDispatcher.class, eventDispatcher);
102
103         portSecurityHandler.setDependencies(mock(ServiceReference.class));
104
105         Assert.assertEquals("Error, did not return the correct object", portSecurityHandler.eventDispatcher, eventDispatcher);
106     }
107
108
109     private Object getField(String fieldName) throws Exception {
110         Field field = PortSecurityHandler.class.getDeclaredField(fieldName);
111         field.setAccessible(true);
112         return field.get(portSecurityHandler);
113     }
114
115 }