vpnmanager/vpnmanager-impl/src/test/java/org/opendaylight/vpnservice/test/VpnServiceTest.java

   1 /*
   2  * Copyright (c) 2015 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.yangtools.concepts.ListenerRegistration;
  25 import org.opendaylight.yangtools.yang.binding.DataObject;
  26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.Interfaces;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceKey;
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.VpnInstances;
  32 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.*;
  33 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.*;
  34 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.apply.label.apply.label.mode.*;
  35 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.apply.label.ApplyLabelMode;
  36 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.af.config.ApplyLabelBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.huawei.params.xml.ns.yang.l3vpn.rev140815.vpn.instances.vpn.instance.Ipv4FamilyBuilder;
  38
  39 @RunWith(MockitoJUnitRunner.class)
  40 public class VpnServiceTest {
  41     @Mock DataBroker dataBroker;
  42     @Mock IBgpManager bgpManager;
  43     @Mock ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  44     MockDataChangedEvent event;
  45
  46     @Before
  47     public void setUp() throws Exception {
  48         when(dataBroker.registerDataChangeListener(
  49                 any(LogicalDatastoreType.class),
  50                 any(InstanceIdentifier.class),
  51                 any(DataChangeListener.class),
  52                 any(DataChangeScope.class)))
  53                 .thenReturn(dataChangeListenerRegistration);
  54         event = new MockDataChangedEvent();
  55     }
  56
  57     @Test
  58     public void test() {
  59         VpnInstanceBuilder builder = new VpnInstanceBuilder().setKey(new VpnInstanceKey("Vpn1")).
  60                 setIpv4Family(new Ipv4FamilyBuilder().setRouteDistinguisher("100:1").setImportRoutePolicy("100:2").
  61                     setExportRoutePolicy("100:1").setApplyLabel(new ApplyLabelBuilder().setApplyLabelMode(
  62                         new PerRouteBuilder().setApplyLabelPerRoute(true).build()).build()).build());
  63         VpnInstance instance = builder.build();
  64         VpnManager vpnManager = new VpnManager(dataBroker, bgpManager);
  65         event.created.put(createVpnId("Vpn1"), instance);
  66         //TODO: Need to enhance the test case to handle ds read/write ops
  67         //vpnManager.onDataChanged(event);
  68     }
  69
  70     private InstanceIdentifier<VpnInstance> createVpnId(String name) {
  71        InstanceIdentifierBuilder<VpnInstance> idBuilder =
  72            InstanceIdentifier.builder(VpnInstances.class).child(VpnInstance.class, new VpnInstanceKey(name));
  73        InstanceIdentifier<VpnInstance> id = idBuilder.build();
  74        return id;
  75     }
  76
  77 }