neutronvpn/impl/src/test/java/org/opendaylight/netvirt/neutronvpn/NeutronPortChangeListenerTest.java

   1 /*
   2  * Copyright (c) 2016 Hewlett-Packard Development Company, L.P. 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.neutronvpn;
  10
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.when;
  14
  15 import com.google.common.base.Optional;
  16 import com.google.common.util.concurrent.Futures;
  17 import java.util.ArrayList;
  18 import java.util.List;
  19 import org.junit.After;
  20 import org.junit.Before;
  21 import org.junit.Test;
  22 import org.junit.runner.RunWith;
  23 import org.mockito.Mock;
  24 import org.mockito.runners.MockitoJUnitRunner;
  25 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  26 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  27 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  28 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  29 import org.opendaylight.genius.datastoreutils.listeners.DataTreeEventCallbackRegistrar;
  30 import org.opendaylight.infrautils.caches.baseimpl.internal.CacheManagersRegistryImpl;
  31 import org.opendaylight.infrautils.caches.guava.internal.GuavaCacheProvider;
  32 import org.opendaylight.infrautils.jobcoordinator.internal.JobCoordinatorImpl;
  33 import org.opendaylight.infrautils.metrics.MetricProvider;
  34 import org.opendaylight.infrautils.metrics.testimpl.TestMetricProviderImpl;
  35 import org.opendaylight.netvirt.elanmanager.api.IElanService;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv6Address;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.genius.idmanager.rev160406.IdManagerService;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.ElanInstances;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstance;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.elan.rev150602.elan.instances.ElanInstanceKey;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIps;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.port.attributes.FixedIpsBuilder;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.Port;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.ports.rev150712.ports.attributes.ports.PortBuilder;
  50 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  51
  52 @RunWith(MockitoJUnitRunner.class)
  53 public class NeutronPortChangeListenerTest {
  54
  55     NeutronPortChangeListener neutronPortChangeListener;
  56
  57     @Mock
  58     DataBroker dataBroker;
  59     @Mock
  60     NeutronvpnManager neutronvpnManager;
  61     @Mock
  62     NeutronvpnNatManager neutronvpnNatManager;
  63     @Mock
  64     WriteTransaction mockWriteTx;
  65     @Mock
  66     ReadOnlyTransaction mockReadTx;
  67     @Mock
  68     Network mockNetwork;
  69     @Mock
  70     ElanInstance elanInstance;
  71     @Mock
  72     NeutronSubnetGwMacResolver gwMacResolver;
  73     @Mock
  74     IElanService elanService;
  75     @Mock
  76     IdManagerService idManager;
  77     @Mock
  78     IPV6InternetDefaultRouteProgrammer ipV6InternetDefRt;
  79     @Mock
  80     DataTreeEventCallbackRegistrar dataTreeEventCallbackRegistrar;
  81
  82     MetricProvider metricProvider = new TestMetricProviderImpl();
  83
  84     private final JobCoordinatorImpl jobCoordinator = new JobCoordinatorImpl(metricProvider);
  85
  86     @Before
  87     public void setUp() {
  88         doReturn(mockReadTx).when(dataBroker).newReadOnlyTransaction();
  89         when(mockReadTx.read(any(LogicalDatastoreType.class), any(InstanceIdentifier.class)))
  90             .thenReturn(Futures.immediateCheckedFuture(Optional.of(mockNetwork)));
  91         neutronPortChangeListener = new NeutronPortChangeListener(dataBroker, neutronvpnManager, neutronvpnNatManager,
  92                 gwMacResolver, elanService, jobCoordinator, new NeutronvpnUtils(dataBroker, idManager, jobCoordinator,
  93                         ipV6InternetDefRt),
  94                 new HostConfigCache(dataBroker, new GuavaCacheProvider(new CacheManagersRegistryImpl())),
  95                 dataTreeEventCallbackRegistrar);
  96         InstanceIdentifier<ElanInstance> elanIdentifierId = InstanceIdentifier.builder(ElanInstances.class)
  97                 .child(ElanInstance.class,
  98                         new ElanInstanceKey(new Uuid("12345678-1234-1234-1234-123456789012").getValue())).build();
  99     }
 100
 101     @After
 102     public void cleanup() {
 103         jobCoordinator.destroy();
 104     }
 105
 106     @Test
 107     public void addPort__Ipv6FixedIps() throws Exception {
 108         PortBuilder pb = new PortBuilder();
 109         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 110         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 111         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 112         IpAddress ipv6 = new IpAddress(new Ipv6Address("1::1"));
 113         FixedIpsBuilder fib = new FixedIpsBuilder();
 114         fib.setIpAddress(ipv6);
 115         List<FixedIps> fixedIps = new ArrayList<>();
 116         fixedIps.add(fib.build());
 117         pb.setFixedIps(fixedIps);
 118         Port port = pb.build();
 119         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 120     }
 121
 122     @Test
 123     public void addPort__Ipv4FixedIps() throws Exception {
 124         PortBuilder pb = new PortBuilder();
 125         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 126         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 127         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 128         IpAddress ipv4 = new IpAddress(new Ipv4Address("2.2.2.2"));
 129         FixedIpsBuilder fib = new FixedIpsBuilder();
 130         fib.setIpAddress(ipv4);
 131         List<FixedIps> fixedIps = new ArrayList<>();
 132         fixedIps.add(fib.build());
 133         pb.setFixedIps(fixedIps);
 134         Port port = pb.build();
 135         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 136     }
 137
 138     @Test
 139     public void addPort__NoFixedIps() throws Exception {
 140         PortBuilder pb = new PortBuilder();
 141         pb.setUuid(new Uuid("12345678-1234-1234-1234-123456789012"));
 142         pb.setNetworkId(new Uuid("12345678-1234-1234-1234-123456789012"));
 143         pb.setMacAddress(new MacAddress("AA:BB:CC:DD:EE:FF"));
 144         List<FixedIps> fixedIps = new ArrayList<>();
 145         pb.setFixedIps(fixedIps);
 146         Port port = pb.build();
 147         neutronPortChangeListener.add(InstanceIdentifier.create(Port.class), port);
 148     }
 149
 150 }