vpnmanager/impl/src/test/java/org/opendaylight/netvirt/vpnmanager/test/VpnServiceTest.java

   1 /*
   2  * Copyright (c) 2015 - 2016 Ericsson India Global Services Pvt Ltd. 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.netvirt.vpnmanager.test;
   9
  10 import java.util.ArrayList;
  11 import java.util.Arrays;
  12 import java.util.List;
  13 import org.junit.Before;
  14 import org.junit.Test;
  15 import org.junit.runner.RunWith;
  16 import org.mockito.Mock;
  17 import org.mockito.junit.MockitoJUnitRunner;
  18 import org.opendaylight.mdsal.binding.api.DataBroker;
  19 import org.opendaylight.netvirt.bgpmanager.api.IBgpManager;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.VpnInstances;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.VpnInstance;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.VpnInstanceBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.VpnInstanceKey;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.vpn.instance.VpnTargets;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.vpn.instance.VpnTargetsBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.vpn.instance.vpntargets.VpnTarget;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.vpn.instance.vpntargets.VpnTargetBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.neutronvpn.l3vpn.rev200204.vpn.instances.vpn.instance.vpntargets.VpnTargetKey;
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  31
  32 @RunWith(MockitoJUnitRunner.class)
  33 public class VpnServiceTest {
  34     @Mock
  35     DataBroker dataBroker;
  36     @Mock
  37     IBgpManager bgpManager;
  38
  39     @Before
  40     public void setUp() throws Exception {
  41     }
  42
  43     @Test
  44     public void test() {
  45
  46         List<VpnTarget> vpnTargetList = new ArrayList<>();
  47
  48         VpnTarget vpneRTarget = new VpnTargetBuilder().withKey(new VpnTargetKey("100:1")).setVrfRTValue("100:1")
  49             .setVrfRTType(VpnTarget.VrfRTType.ExportExtcommunity).build();
  50         VpnTarget vpniRTarget = new VpnTargetBuilder().withKey(new VpnTargetKey("100:2")).setVrfRTValue("100:2")
  51             .setVrfRTType(VpnTarget.VrfRTType.ImportExtcommunity).build();
  52
  53         vpnTargetList.add(vpneRTarget);
  54         vpnTargetList.add(vpniRTarget);
  55
  56         VpnTargets vpnTargets = new VpnTargetsBuilder().setVpnTarget(vpnTargetList).build();
  57
  58         VpnInstanceBuilder builder =
  59                 new VpnInstanceBuilder().withKey(new VpnInstanceKey("Vpn1"))
  60                         .setRouteDistinguisher(Arrays.asList("100:1","100:2:")).setVpnTargets(vpnTargets);
  61         VpnInstance instance = builder.build();
  62         //TODO: Need to enhance the test case to handle ds read/write ops
  63         //vpnManager.onDataChanged(event);
  64     }
  65
  66     private InstanceIdentifier<VpnInstance> createVpnId(String name) {
  67         InstanceIdentifierBuilder<VpnInstance> idBuilder =
  68             InstanceIdentifier.builder(VpnInstances.class).child(VpnInstance.class, new VpnInstanceKey(name));
  69         InstanceIdentifier<VpnInstance> id = idBuilder.build();
  70         return id;
  71     }
  72
  73 }