neutron/src/test/java/org/opendaylight/ovsdb/neutron/NetworkHandlerTest.java

   1 /*******************************************************************************
   2  * Copyright (c) 2013 Hewlett-Packard Development Company, L.P.
   3  * All rights reserved. This program and the accompanying materials
   4  * are made available under the terms of the Eclipse Public License v1.0
   5  * which accompanies this distribution, and is available at
   6  * http://www.eclipse.org/legal/epl-v10.html
   7  *
   8  * Contributors:
   9  *    Dave Tucker (HP) - Added unit tests for the NetworkHandler class.
  10  *******************************************************************************/
  11
  12 package org.opendaylight.ovsdb.neutron;
  13
  14 import static org.junit.Assert.assertEquals;
  15 import static org.mockito.Mockito.mock;
  16 import static org.mockito.Mockito.spy;
  17 import static org.mockito.Mockito.verify;
  18 import static org.mockito.Mockito.when;
  19
  20 import java.net.HttpURLConnection;
  21
  22 import org.junit.Test;
  23 import org.junit.runner.RunWith;
  24 import org.opendaylight.controller.networkconfig.neutron.NeutronNetwork;
  25 import org.powermock.core.classloader.annotations.PrepareForTest;
  26 import org.powermock.modules.junit4.PowerMockRunner;
  27 import org.powermock.reflect.Whitebox;
  28
  29 @RunWith(PowerMockRunner.class)
  30 @PrepareForTest(TenantNetworkManager.class)
  31
  32 public class NetworkHandlerTest {
  33
  34     NetworkHandler testNetworkHandler = new NetworkHandler();
  35
  36     @Test
  37     public void testCanCreateNetwork() throws Exception {
  38         NeutronNetwork mockNet = mock(NeutronNetwork.class);
  39
  40         when(mockNet.isShared())
  41                 .thenReturn(true)
  42                 .thenReturn(false);
  43
  44         assertEquals(HttpURLConnection.HTTP_NOT_ACCEPTABLE, testNetworkHandler.canCreateNetwork(mockNet));
  45         assertEquals(HttpURLConnection.HTTP_CREATED, testNetworkHandler.canCreateNetwork(mockNet));
  46
  47     }
  48
  49     @Test
  50     public void testNeutronNetworkCreated() {
  51         String netId = "6cfdb7";
  52         NeutronNetwork mockNet = mock(NeutronNetwork.class);
  53         when(mockNet.isShared()).thenReturn(false);
  54         when(mockNet.getID()).thenReturn(netId);
  55
  56         TenantNetworkManager tenantNetworkManager = mock(TenantNetworkManager.class);
  57         Whitebox.setInternalState(TenantNetworkManager.class, "tenantHelper", tenantNetworkManager);
  58
  59         testNetworkHandler.neutronNetworkCreated(mockNet);
  60
  61         verify(tenantNetworkManager).networkCreated(netId);
  62     }
  63
  64     @Test
  65     public void testCanUpdateNetwork() {
  66         NeutronNetwork delta = new NeutronNetwork();
  67         NeutronNetwork original = new NeutronNetwork();
  68
  69         assertEquals(HttpURLConnection.HTTP_OK, testNetworkHandler.canUpdateNetwork(delta, original));
  70     }
  71
  72     @Test
  73     public void testCanDeleteNetwork() throws Exception {
  74         NeutronNetwork network = new NeutronNetwork();
  75
  76         assertEquals(HttpURLConnection.HTTP_OK, testNetworkHandler.canDeleteNetwork(network));
  77     }
  78
  79     @Test
  80     public void testNeutronNetworkDeleted() throws Exception {
  81         String netId = "6cfdb7";
  82
  83         NeutronNetwork mockNet = mock(NeutronNetwork.class);
  84         when(mockNet.isShared()).thenReturn(false);
  85         when(mockNet.getID()).thenReturn(netId);
  86
  87         NetworkHandler spy = spy(testNetworkHandler);
  88
  89         TenantNetworkManager tenantNetworkManager = mock(TenantNetworkManager.class);
  90         Whitebox.setInternalState(TenantNetworkManager.class, "tenantHelper", tenantNetworkManager);
  91
  92         when(spy.canDeleteNetwork(mockNet))
  93                 .thenReturn(HttpURLConnection.HTTP_BAD_REQUEST)
  94                 .thenCallRealMethod();
  95
  96         spy.neutronNetworkDeleted(mockNet);
  97         spy.neutronNetworkDeleted(mockNet);
  98
  99         verify(tenantNetworkManager).networkDeleted(netId);
 100
 101     }
 102 }