2 * Copyright (c) 2015 Inocybe 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.openstack.netvirt;
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertTrue;
13 import static org.mockito.Matchers.any;
14 import static org.mockito.Matchers.anyString;
15 import static org.mockito.Matchers.same;
16 import static org.mockito.Mockito.mock;
17 import static org.mockito.Mockito.times;
18 import static org.mockito.Mockito.verify;
19 import static org.mockito.Mockito.when;
21 import java.lang.reflect.Field;
22 import java.net.HttpURLConnection;
23 import java.util.ArrayList;
24 import java.util.List;
26 import java.util.Map.Entry;
28 import org.junit.Before;
29 import org.junit.Test;
30 import org.junit.runner.RunWith;
31 import org.mockito.InjectMocks;
32 import org.mockito.Mock;
33 import org.mockito.Mockito;
34 import org.opendaylight.neutron.spi.INeutronLoadBalancerCRUD;
35 import org.opendaylight.neutron.spi.INeutronLoadBalancerPoolCRUD;
36 import org.opendaylight.neutron.spi.INeutronNetworkCRUD;
37 import org.opendaylight.neutron.spi.INeutronPortCRUD;
38 import org.opendaylight.neutron.spi.INeutronSubnetCRUD;
39 import org.opendaylight.neutron.spi.NeutronLoadBalancer;
40 import org.opendaylight.neutron.spi.NeutronLoadBalancerPool;
41 import org.opendaylight.neutron.spi.NeutronLoadBalancerPoolMember;
42 import org.opendaylight.ovsdb.openstack.netvirt.api.Action;
43 import org.opendaylight.ovsdb.openstack.netvirt.api.EventDispatcher;
44 import org.opendaylight.ovsdb.openstack.netvirt.api.LoadBalancerConfiguration;
45 import org.opendaylight.ovsdb.openstack.netvirt.api.LoadBalancerConfiguration.LoadBalancerPoolMember;
46 import org.opendaylight.ovsdb.openstack.netvirt.api.LoadBalancerProvider;
47 import org.opendaylight.ovsdb.openstack.netvirt.api.NodeCacheManager;
48 import org.opendaylight.ovsdb.utils.servicehelper.ServiceHelper;
49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
50 import org.osgi.framework.BundleContext;
51 import org.osgi.framework.ServiceReference;
52 import org.powermock.api.mockito.PowerMockito;
53 import org.powermock.core.classloader.annotations.PrepareForTest;
54 import org.powermock.modules.junit4.PowerMockRunner;
57 * Unit test for {@link LBaaSHandler}
59 @RunWith(PowerMockRunner.class)
60 @PrepareForTest({NeutronCacheUtils.class, ServiceHelper.class})
61 public class LBaaSHandlerTest {
63 @InjectMocks private LBaaSHandler lbaasHandler;
64 private LBaaSHandler lbaasHandlerSpy;
66 @Mock private INeutronLoadBalancerCRUD neutronLBCache;
67 @Mock private INeutronLoadBalancerPoolCRUD neutronLBPoolCache;
68 @Mock private LoadBalancerProvider loadBalancerProvider;
69 @Mock private NodeCacheManager nodeCacheManager;
70 @Mock private NeutronLoadBalancer neutronLB;
74 NorthboundEvent ev = mock(NorthboundEvent.class);
75 Map.Entry<String,String> providerInfo = mock(Entry.class);
76 NeutronLoadBalancerPoolMember neutronLBPoolMember = mock(NeutronLoadBalancerPoolMember.class);
77 NeutronLoadBalancerPool neutronLBPool = mock(NeutronLoadBalancerPool.class);
78 List<NeutronLoadBalancerPoolMember> members = new ArrayList();
79 List<NeutronLoadBalancerPool> list_neutronLBPool = new ArrayList();
80 List<NeutronLoadBalancer> list_neutronLB = new ArrayList();
81 List<Node> list_node = new ArrayList();
83 lbaasHandlerSpy = Mockito.spy(lbaasHandler);
85 neutronLB = mock(NeutronLoadBalancer.class);
86 when(neutronLB.getLoadBalancerName()).thenReturn("load_balancer_name");
87 when(neutronLB.getLoadBalancerVipAddress()).thenReturn("vip_address");
88 when(neutronLB.getLoadBalancerVipSubnetID()).thenReturn("subnetID");
90 when(ev.getLoadBalancer()).thenReturn(neutronLB);
92 when(providerInfo.getKey()).thenReturn("key");
93 when(providerInfo.getValue()).thenReturn("value");
95 PowerMockito.mockStatic(NeutronCacheUtils.class);
96 when(NeutronCacheUtils.getProviderInformation(any(INeutronNetworkCRUD.class), any(INeutronSubnetCRUD.class), anyString())).thenReturn(providerInfo);
97 when(NeutronCacheUtils.getMacAddress(any(INeutronPortCRUD.class), anyString(), anyString())).thenReturn("mac_address");
99 when(neutronLBPoolMember.getPoolMemberAdminStateIsUp()).thenReturn(true);
100 when(neutronLBPoolMember.getPoolMemberSubnetID()).thenReturn("subnetID");
101 when(neutronLBPoolMember.getID()).thenReturn("pool_memberID");
102 when(neutronLBPoolMember.getPoolMemberAddress()).thenReturn("pool_member_address");
103 when(neutronLBPoolMember.getPoolMemberProtoPort()).thenReturn(1);
104 members.add(neutronLBPoolMember);
106 when(neutronLBPool.getLoadBalancerPoolMembers()).thenReturn(members);
107 when(neutronLBPool.getLoadBalancerPoolProtocol()).thenReturn(LoadBalancerConfiguration.PROTOCOL_TCP);
108 list_neutronLBPool.add(neutronLBPool);
109 when(neutronLBPoolCache.getAllNeutronLoadBalancerPools()).thenReturn(list_neutronLBPool);
111 list_neutronLB.add(neutronLB);
112 when(neutronLBCache.getAllNeutronLoadBalancers()).thenReturn(list_neutronLB );
114 list_node.add(mock(Node.class));
115 when(nodeCacheManager.getBridgeNodes()).thenReturn(list_node);
119 public void testCanCreateNeutronLoadBalancer(){
120 assertEquals("Error, canCreateNeutronLoadBalancer() did not return the correct value ", HttpURLConnection.HTTP_OK, lbaasHandler.canCreateNeutronLoadBalancer(any(NeutronLoadBalancer.class)));
124 public void testCanUpdateNeutronLoadBalancer(){
125 assertEquals("Error, canUpdateNeutronLoadBalancer() did not return the correct value ", HttpURLConnection.HTTP_OK, lbaasHandler.canUpdateNeutronLoadBalancer(any(NeutronLoadBalancer.class), any(NeutronLoadBalancer.class)));
129 public void testCanDeleteNeutronLoadBalancer(){
130 assertEquals("Error, canDeleteNeutronLoadBalancer() did not return the correct value ", HttpURLConnection.HTTP_OK, lbaasHandler.canDeleteNeutronLoadBalancer(any(NeutronLoadBalancer.class)));
134 * Test method {@link LBaaSHandler#processEvent(AbstractEvent)}
137 public void testProcessEvent(){
138 NorthboundEvent ev = mock(NorthboundEvent.class);
139 when(ev.getLoadBalancer()).thenReturn(neutronLB);
141 when(ev.getAction()).thenReturn(Action.ADD);
142 lbaasHandlerSpy.processEvent(ev);
143 verify(lbaasHandlerSpy, times(1)).extractLBConfiguration(any(NeutronLoadBalancer.class));
144 verify(loadBalancerProvider, times(1)).programLoadBalancerRules(any(Node.class), any(LoadBalancerConfiguration.class), same(Action.ADD));
146 when(ev.getAction()).thenReturn(Action.DELETE);
147 lbaasHandlerSpy.processEvent(ev);
148 verify(lbaasHandlerSpy, times(2)).extractLBConfiguration(any(NeutronLoadBalancer.class)); // 1 + 1 above
149 verify(loadBalancerProvider, times(1)).programLoadBalancerRules(any(Node.class), any(LoadBalancerConfiguration.class), same(Action.DELETE));
151 when(ev.getAction()).thenReturn(Action.UPDATE);
152 lbaasHandlerSpy.processEvent(ev);
153 verify(lbaasHandlerSpy, times(4)).extractLBConfiguration(any(NeutronLoadBalancer.class)); // 2 + 2 above
154 verify(loadBalancerProvider, times(2)).programLoadBalancerRules(any(Node.class), any(LoadBalancerConfiguration.class), same(Action.DELETE)); // 1 + 1 above
155 verify(loadBalancerProvider, times(2)).programLoadBalancerRules(any(Node.class), any(LoadBalancerConfiguration.class), same(Action.ADD)); // 1 + 1 above
159 * Test method {@link LBaaSHandler#extractLBConfiguration(NeutronLoadBalancer)}
162 public void testExtractLBConfiguration(){
163 LoadBalancerConfiguration lbConfig = lbaasHandler.extractLBConfiguration(neutronLB);
165 verify(neutronLBPoolCache, times(1)).getAllNeutronLoadBalancerPools();
167 // make sure the load balancer configuration was correctly populated
168 assertEquals("Error, did not return the correct value", "key", lbConfig.getProviderNetworkType());
169 assertEquals("Error, did not return the correct value", "value", lbConfig.getProviderSegmentationId());
170 assertEquals("Error, did not return the correct value", "mac_address", lbConfig.getVmac());
172 // make sure the load balancer pool member was correctly populated
173 LoadBalancerPoolMember member = lbConfig.getMembers().get("pool_memberID");
174 assertEquals("Error, did not return the correct value", "pool_member_address", member.getIP());
175 assertEquals("Error, did not return the correct value", "mac_address", member.getMAC());
176 assertEquals("Error, did not return the correct value", LoadBalancerConfiguration.PROTOCOL_TCP, member.getProtocol());
177 assertTrue("Error, did not return the correct value", 1 == member.getPort());
181 * Test method {@link LBaaSHandler#notifyNode(Node, Action)}
184 public void testNotifyNode() {
185 lbaasHandlerSpy.notifyNode(mock(Node.class), Action.ADD);
187 verify(lbaasHandlerSpy, times(1)).extractLBConfiguration(any(NeutronLoadBalancer.class));
188 verify(neutronLBCache, times(1)).getAllNeutronLoadBalancers();
189 verify(neutronLBPoolCache, times(1)).getAllNeutronLoadBalancerPools();
190 verify(loadBalancerProvider, times(1)).programLoadBalancerRules(any(Node.class), any(LoadBalancerConfiguration.class), any(Action.class));
194 public void testSetDependencies() throws Exception {
195 EventDispatcher eventDispatcher = mock(EventDispatcher.class);
196 LoadBalancerProvider loadBalancerProvider = mock(LoadBalancerProvider.class);
197 NodeCacheManager nodeCacheManager = mock(NodeCacheManager.class);
199 PowerMockito.mockStatic(ServiceHelper.class);
200 PowerMockito.when(ServiceHelper.getGlobalInstance(EventDispatcher.class, lbaasHandler)).thenReturn(eventDispatcher);
201 PowerMockito.when(ServiceHelper.getGlobalInstance(LoadBalancerProvider.class, lbaasHandler)).thenReturn(loadBalancerProvider);
202 PowerMockito.when(ServiceHelper.getGlobalInstance(NodeCacheManager.class, lbaasHandler)).thenReturn(nodeCacheManager);
204 lbaasHandler.setDependencies(mock(BundleContext.class), mock(ServiceReference.class));
206 assertEquals("Error, did not return the correct object", lbaasHandler.eventDispatcher, eventDispatcher);
207 assertEquals("Error, did not return the correct object", getField("loadBalancerProvider"), loadBalancerProvider);
208 assertEquals("Error, did not return the correct object", getField("nodeCacheManager"), nodeCacheManager);
212 public void testSetDependenciesObject() throws Exception{
213 INeutronNetworkCRUD iNeutronNetworkCRUD = mock(INeutronNetworkCRUD.class);
214 lbaasHandler.setDependencies(iNeutronNetworkCRUD);
215 assertEquals("Error, did not return the correct object", getField("neutronNetworkCache"), iNeutronNetworkCRUD);
217 INeutronPortCRUD iNeutronPortCRUD = mock(INeutronPortCRUD.class);
218 lbaasHandler.setDependencies(iNeutronPortCRUD);
219 assertEquals("Error, did not return the correct object", getField("neutronPortCache"), iNeutronPortCRUD);
221 INeutronSubnetCRUD iNeutronSubnetCRUD = mock(INeutronSubnetCRUD.class);
222 lbaasHandler.setDependencies(iNeutronSubnetCRUD);
223 assertEquals("Error, did not return the correct object", getField("neutronSubnetCache"), iNeutronSubnetCRUD);
225 INeutronLoadBalancerCRUD iNeutronLoadBalancerCRUD = mock(INeutronLoadBalancerCRUD.class);
226 lbaasHandler.setDependencies(iNeutronLoadBalancerCRUD);
227 assertEquals("Error, did not return the correct object", getField("neutronLBCache"), iNeutronLoadBalancerCRUD);
229 INeutronLoadBalancerPoolCRUD iNeutronLoadBalancerPoolCRUD = mock(INeutronLoadBalancerPoolCRUD.class);
230 lbaasHandler.setDependencies(iNeutronLoadBalancerPoolCRUD);
231 assertEquals("Error, did not return the correct object", getField("neutronLBPoolCache"), iNeutronLoadBalancerPoolCRUD);
233 LoadBalancerProvider loadBalancerProvider = mock(LoadBalancerProvider.class);
234 lbaasHandler.setDependencies(loadBalancerProvider);
235 assertEquals("Error, did not return the correct object", getField("loadBalancerProvider"), loadBalancerProvider);
238 private Object getField(String fieldName) throws Exception {
239 Field field = LBaaSHandler.class.getDeclaredField(fieldName);
240 field.setAccessible(true);
241 return field.get(lbaasHandler);