2 * Copyright (c) 2015 - 2016 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the terms of the Eclipse
5 * Public License v1.0 which accompanies this distribution, and is available at
6 * http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.vpnservice.fibmanager.test;
10 import java.math.BigInteger;
12 import org.junit.Before;
13 import org.junit.Test;
15 import static org.mockito.Matchers.any;
16 import static org.mockito.Mockito.when;
18 import java.util.ArrayList;
19 import java.util.List;
21 import org.junit.runner.RunWith;
22 import org.mockito.Mock;
23 import org.mockito.runners.MockitoJUnitRunner;
24 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
25 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
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.AsyncDataBroker.DataChangeScope;
29 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
30 import org.opendaylight.vpnmanager.api.IVpnManager;
31 import org.opendaylight.vpnservice.fibmanager.FibManager;
32 import org.opendaylight.vpnservice.mdsalutil.interfaces.IMdsalApiManager;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntry;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.VpnInstanceOpDataEntryKey;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.vpn.instance.op.data.entry.VpnToDpnList;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.vpn.instance.op.data.entry.VpnToDpnListKey;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.l3vpn.rev130911.vpn.instance.op.data.vpn.instance.op.data.entry.vpn.to.dpn.list.VpnInterfaces;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.FibEntries;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTables;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTablesKey;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntry;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntryBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntryKey;
44 import org.opendaylight.yangtools.concepts.ListenerRegistration;
45 import org.opendaylight.yangtools.yang.binding.Augmentation;
46 import org.opendaylight.yangtools.yang.binding.DataContainer;
47 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
48 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
50 @RunWith(MockitoJUnitRunner.class)
51 public class FibManagerTest {
54 DataBroker dataBroker;
56 ListenerRegistration<DataChangeListener> dataChangeListenerRegistration;
58 ReadOnlyTransaction mockReadTx;
60 WriteTransaction mockWriteTx;
62 IMdsalApiManager mdsalManager;
64 IVpnManager vpnmanager;
66 VrfTablesKey vrfTableKey;
68 MockDataChangedEvent dataChangeEvent;
70 private static final Long EgressPointer = 11L;
72 InstanceIdentifier<VrfEntry> identifier;
73 VrfEntryBuilder vrfbuilder;
74 private static final String testRd = "100:1";
75 private static final String prefix = "1.1.2.3";
76 private static final String nexthop = "1.1.1.1";
77 private static final int label = 10;
79 private static final long vpnId = 101L;
80 private static final long vpnIntfCnt = 2;
82 private void SetupMocks() {
83 Dpn = BigInteger.valueOf(100000L);
84 identifier = buildVrfEntryId(testRd, prefix);
85 vrfEntry = buildVrfEntry(testRd, prefix, nexthop, label);
86 fibmgr.setMdsalManager(mdsalManager);
87 fibmgr.setVpnmanager(vpnmanager);
88 when(vrfTableKey.getRouteDistinguisher()).thenReturn(testRd);
92 public void setUp() throws Exception {
94 dataBroker.registerDataChangeListener(any(LogicalDatastoreType.class),
95 any(InstanceIdentifier.class), any(DataChangeListener.class),
96 any(DataChangeScope.class))).thenReturn(dataChangeListenerRegistration);
97 dataChangeEvent = new MockDataChangedEvent();
98 vrfbuilder = new VrfEntryBuilder();
99 fibmgr = new FibManager(dataBroker) {
101 protected VpnInstanceOpDataEntry getVpnInstance(String rd) {
102 return new VpnInstanceOpDataEntry() {
105 public <E extends Augmentation<VpnInstanceOpDataEntry>> E getAugmentation(Class<E> aClass) {
110 public Long getVpnId() {
115 public String getVrfId() {
120 public Long getVpnInterfaceCount() { return vpnIntfCnt; }
123 public List<VpnToDpnList> getVpnToDpnList() {
124 List <VpnToDpnList> vpnToDpnLists = new ArrayList<>();
125 vpnToDpnLists.add(new VpnToDpnList() {
127 public BigInteger getDpnId() {
132 public List<VpnInterfaces> getVpnInterfaces() {
137 public VpnToDpnListKey getKey() {
138 return new VpnToDpnListKey(Dpn);
142 public <E extends Augmentation<VpnToDpnList>> E getAugmentation(
143 Class<E> augmentationType) {
148 public Class<? extends DataContainer> getImplementedInterface() {
152 return vpnToDpnLists;
156 public VpnInstanceOpDataEntryKey getKey() {
157 return new VpnInstanceOpDataEntryKey(testRd);
161 public List<Long> getRouteEntryId() {
166 public Class<? extends DataContainer> getImplementedInterface() {
176 public void testAdd() {
177 dataChangeEvent.created.put(identifier, vrfEntry);
178 //fibmgr.onDataChanged(dataChangeEvent);
179 //Mockito.verify(mdsalManager, Mockito.times(2)).installFlow(any(FlowEntity.class));
182 private VrfEntry buildVrfEntry(String rd, String prefix, String nexthop, int label) {
183 return new VrfEntryBuilder().setDestPrefix(prefix).setNextHopAddress(nexthop)
184 .setLabel((long) label).build();
187 public static InstanceIdentifier<VrfTables> buildVrfTableId(String rd) {
188 InstanceIdentifierBuilder<VrfTables> idBuilder =
189 InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd));
190 InstanceIdentifier<VrfTables> vrfTableId = idBuilder.build();
194 public static InstanceIdentifier<VrfEntry> buildVrfEntryId(String rd, String prefix) {
195 InstanceIdentifierBuilder<VrfEntry> idBuilder =
196 InstanceIdentifier.builder(FibEntries.class).child(VrfTables.class, new VrfTablesKey(rd))
197 .child(VrfEntry.class, new VrfEntryKey(prefix));
198 InstanceIdentifier<VrfEntry> vrfEntryId = idBuilder.build();