openstack/net-virt/src/test/java/org/opendaylight/ovsdb/openstack/netvirt/impl/ProviderNetworkManagerImplTest.java

   1 /*
   2  * Copyright (c) 2015 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 package org.opendaylight.ovsdb.openstack.netvirt.impl;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.mockito.Mockito.mock;
  12 import static org.mockito.Mockito.when;
  13
  14 import java.lang.reflect.Field;
  15 import java.util.HashMap;
  16
  17 import org.junit.Test;
  18 import org.junit.runner.RunWith;
  19 import org.mockito.InjectMocks;
  20 import org.mockito.Spy;
  21 import org.mockito.runners.MockitoJUnitRunner;
  22 import org.opendaylight.ovsdb.openstack.netvirt.api.NetworkingProvider;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  24 import org.osgi.framework.ServiceReference;
  25
  26 /**
  27  * Unit test for {@link ProviderNetworkManagerImpl}
  28  */
  29 @RunWith(MockitoJUnitRunner.class)
  30 public class ProviderNetworkManagerImplTest {
  31
  32     @InjectMocks private ProviderNetworkManagerImpl providerNetworkManagerImpl;
  33     @Spy private HashMap<Node, NetworkingProvider> nodeToProviderMapping;
  34
  35     /**
  36      * Test method {@link ProviderNetworkManagerImpl#getProvider(Node)}
  37      */
  38     @Test
  39     public void testGetProvider(){
  40         // TODO test the method with no networkingProvider in the map
  41         // Could not be done as ProviderEntry is a private inner class of ProviderNetworkManagerImpl
  42 //        assertNotNull("Error, did not return the networkingProvider of the specified node", providerNetworkManagerImpl.getProvider(any(Node.class));
  43
  44         Node node = mock(Node.class);
  45         NetworkingProvider networkingProvider = mock(NetworkingProvider.class);
  46         nodeToProviderMapping.put(node, networkingProvider);
  47         assertEquals("Error, did not return the networkingProvider of the specified node", networkingProvider, providerNetworkManagerImpl.getProvider(node));
  48     }
  49
  50     /**
  51      * Test methods {@link ProviderNetworkManagerImpl#providerRemoved(ServiceReference)}
  52      * and {@link ProviderNetworkManagerImpl#providerAdded(ServiceReference, NetworkingProvider)}
  53      */
  54     @Test
  55     public void testProviderAddedAndRemoved() throws Exception {
  56         Field field = ProviderNetworkManagerImpl.class.getDeclaredField("providers");
  57         field.setAccessible(true);
  58         HashMap map = (HashMap) field.get(providerNetworkManagerImpl);
  59
  60         ServiceReference ref = mock(ServiceReference.class);
  61         when(ref.getProperty(org.osgi.framework.Constants.SERVICE_ID)).thenReturn(Long.valueOf(1));
  62
  63         providerNetworkManagerImpl.providerAdded(ref, mock(NetworkingProvider.class));
  64
  65         assertEquals("Error, providerAdded() did not add the provider", 1, map.size());
  66
  67         providerNetworkManagerImpl.providerRemoved(ref);
  68
  69         assertEquals("Error, providerRemoved() did not remove the provider", 0, map.size());
  70     }
  71 }