vpnmanager/vpnmanager-impl/src/test/java/org/opendaylight/vpnservice/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.vpnservice.test;
   9
  10 import static org.junit.Assert.*;
  11 import static org.mockito.Matchers.*;
  12 import static org.mockito.Mockito.when;
  13
  14 import org.junit.*;
  15 import org.junit.runner.RunWith;
  16 import org.mockito.Mock;
  17 import org.mockito.runners.MockitoJUnitRunner;
  18 import org.opendaylight.vpnservice.VpnManager;
  19 import org.opendaylight.bgpmanager.api.IBgpManager;
  20 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  21 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  22 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  24 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.vpntargets.VpnTarget;
  25 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.vpntargets.VpnTargetBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.vpntargets.VpnTargetKey;
  27 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.vpn.instance.Ipv4Family;
  28 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  29 import org.opendaylight.yangtools.yang.binding.DataObject;
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.Interfaces;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceKey;
  34 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.VpnInstances;
  36 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.*;
  37 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.*;
  38 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.apply.label.apply.label.mode.*;
  39 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.apply.label.ApplyLabelMode;
  40 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.ApplyLabelBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.vpn.instance.Ipv4FamilyBuilder;
  42
  43 import java.util.ArrayList;
  44 import java.util.List;
  45
  46 @RunWith(MockitoJUnitRunner.class)
  47 public class VpnServiceTest {
  48     @Mock DataBroker dataBroker;
  49     @Mock IBgpManager bgpManager;
  50     @Mock ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  51     MockDataChangedEvent event;
  52
  53     @Before
  54     public void setUp() throws Exception {
  55         when(dataBroker.registerDataChangeListener(
  56                 any(LogicalDatastoreType.class),
  57                 any(InstanceIdentifier.class),
  58                 any(DataChangeListener.class),
  59                 any(DataChangeScope.class)))
  60                 .thenReturn(dataChangeListenerRegistration);
  61         event = new MockDataChangedEvent();
  62     }
  63
  64     @Test
  65     public void test() {
  66
  67         List<VpnTarget> vpnTargetList = new ArrayList<VpnTarget>();
  68
  69         VpnTarget vpneRTarget = new VpnTargetBuilder().setKey(new VpnTargetKey("100:1")).setVrfRTValue("100:1")
  70                 .setVrfRTType(VpnTarget.VrfRTType.ExportExtcommunity).build();
  71         VpnTarget vpniRTarget = new VpnTargetBuilder().setKey(new VpnTargetKey("100:2")).setVrfRTValue("100:2")
  72                 .setVrfRTType(VpnTarget.VrfRTType.ImportExtcommunity).build();
  73
  74         vpnTargetList.add(vpneRTarget);
  75         vpnTargetList.add(vpniRTarget);
  76
  77         VpnTargets vpnTargets = new VpnTargetsBuilder().setVpnTarget(vpnTargetList).build();
  78
  79         Ipv4Family ipv4Family = new Ipv4FamilyBuilder().setRouteDistinguisher("100:1").setVpnTargets(vpnTargets)
  80                 .setApplyLabel(new ApplyLabelBuilder().setApplyLabelMode(
  81                         new PerRouteBuilder().setApplyLabelPerRoute(true).build()).build()).build();
  82
  83         VpnInstanceBuilder builder = new VpnInstanceBuilder().setKey(new VpnInstanceKey("Vpn1")).setIpv4Family
  84                 (ipv4Family);
  85         VpnInstance instance = builder.build();
  86         VpnManager vpnManager = new VpnManager(dataBroker, bgpManager);
  87         event.created.put(createVpnId("Vpn1"), instance);
  88         //TODO: Need to enhance the test case to handle ds read/write ops
  89         //vpnManager.onDataChanged(event);
  90     }
  91
  92     private InstanceIdentifier<VpnInstance> createVpnId(String name) {
  93        InstanceIdentifierBuilder<VpnInstance> idBuilder =
  94            InstanceIdentifier.builder(VpnInstances.class).child(VpnInstance.class, new VpnInstanceKey(name));
  95        InstanceIdentifier<VpnInstance> id = idBuilder.build();
  96        return id;
  97     }
  98
  99 }