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
  19 import org.opendaylight.vpnservice.VpnManager;
  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 ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
  43     MockDataChangedEvent event;
  44
  45     @Before
  46     public void setUp() throws Exception {
  47         when(dataBroker.registerDataChangeListener(
  48                 any(LogicalDatastoreType.class),
  49                 any(InstanceIdentifier.class),
  50                 any(DataChangeListener.class),
  51                 any(DataChangeScope.class)))
  52                 .thenReturn(dataChangeListenerRegistration);
  53         event = new MockDataChangedEvent();
  54     }
  55
  56     @Test
  57     public void test() {
  58         VpnInstanceBuilder builder = new VpnInstanceBuilder().setKey(new VpnInstanceKey("Vpn1")).
  59                 setIpv4Family(new Ipv4FamilyBuilder().setRouteDistinguisher("100:1").setImportRoutePolicy("100:2").
  60                     setExportRoutePolicy("100:1").setApplyLabel(new ApplyLabelBuilder().setApplyLabelMode(
  61                         new PerRouteBuilder().setApplyLabelPerRoute(true).build()).build()).build());
  62         VpnInstance instance = builder.build();
  63         VpnManager vpnManager = new VpnManager(dataBroker);
  64         event.created.put(createVpnId("Vpn1"), instance);
  65         vpnManager.onDataChanged(event);
  66     }
  67
  68     private InstanceIdentifier<VpnInstance> createVpnId(String name) {
  69        InstanceIdentifierBuilder<VpnInstance> idBuilder =
  70            InstanceIdentifier.builder(VpnInstances.class).child(VpnInstance.class, new VpnInstanceKey(name));
  71        InstanceIdentifier<VpnInstance> id = idBuilder.build();
  72        return id;
  73     }
  74
  75 }