BgpManager JUnits for add/delete fib entries

Change-Id: I6a2f3b0a75e30a15665cefc64b60b9892782df7f
Signed-off-by: Manisha Malla <manisha.malla@ericsson.com>

diff --git a/bgpmanager/bgpmanager-impl/pom.xml b/bgpmanager/bgpmanager-impl/pom.xml
index 78469fd1ec2bdc562e39aa69c645a7a98800ce29..ffdf1ea093385c51051add8e5cb06d2121cd77b6 100644 (file)
--- a/bgpmanager/bgpmanager-impl/pom.xml
+++ b/bgpmanager/bgpmanager-impl/pom.xml
@@ -43,6 +43,34 @@ and is available at http://www.eclipse.org/legal/epl-v10.html
       <artifactId>libthrift</artifactId>
       <version>0.9.1</version>
     </dependency>
+
+    <!--Dependencies for Unit Test -->
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <scope>test</scope>
+    </dependency>
+
+    <dependency>
+      <groupId>org.mockito</groupId>
+      <artifactId>mockito-all</artifactId>
+      <scope>test</scope>
+    </dependency>
+
+    <dependency>
+      <groupId>org.opendaylight.controller</groupId>
+      <artifactId>sal-binding-broker-impl</artifactId>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>org.opendaylight.controller</groupId>
+      <artifactId>sal-binding-broker-impl</artifactId>
+      <version>${mdsal.version}</version>
+      <scope>test</scope>
+      <type>test-jar</type>
+    </dependency>
+
+
   </dependencies>
 
 </project>

diff --git a/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/AbstractMockFibManager.java b/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/AbstractMockFibManager.java
new file mode 100644 (file)
index 0000000..5aab67b
--- /dev/null
+++ b/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/AbstractMockFibManager.java
@@ -0,0 +1,24 @@
+package org.opendaylight.vpnservice.test;
+
+import java.util.Collection;
+
+import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
+import org.opendaylight.controller.md.sal.binding.api.DataTreeChangeListener;
+import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+import com.google.common.base.Preconditions;
+
+public abstract class AbstractMockFibManager<D extends DataObject> implements DataTreeChangeListener<D> {
+
+    public AbstractMockFibManager() {
+        // Do Nothing
+    }
+
+
+    public void onDataTreeChanged(Collection<DataTreeModification<D>> changes) {
+        // TODO Auto-generated method stub
+    }
+
+}

diff --git a/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/BgpManagerTest.java b/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/BgpManagerTest.java
new file mode 100644 (file)
index 0000000..1c7ee26
--- /dev/null
+++ b/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/BgpManagerTest.java
@@ -0,0 +1,102 @@
+package org.opendaylight.vpnservice.test;
+
+import java.math.BigInteger;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Collections;
+import java.util.concurrent.ExecutionException;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+
+import org.mockito.runners.MockitoJUnitRunner;
+import org.opendaylight.bgpmanager.FibDSWriter;
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
+import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+import static org.junit.Assert.assertEquals;
+
+@RunWith(MockitoJUnitRunner.class)
+
+public class BgpManagerTest extends AbstractDataBrokerTest {
+     DataBroker dataBroker;
+     FibDSWriter bgpFibWriter = null ;
+     MockFibManager fibManager = null ;
+
+     @Before
+     public void setUp() throws Exception {
+         dataBroker = getDataBroker() ;
+         bgpFibWriter = new FibDSWriter(dataBroker);
+         fibManager = new MockFibManager(dataBroker);
+     }
+
+    @Test
+    public void testAddSinglePrefix() {
+        String rd = "101";
+        String prefix = "10.10.10.10/32";
+        String nexthop = "100.100.100.100";
+        int label = 1234;
+
+        bgpFibWriter.addFibEntryToDS(rd, prefix, nexthop, label);
+        assertEquals(1, fibManager.getDataChgCount());
+    }
+
+    @Test
+    public void testAddPrefixesInRd() {
+        String rd = "101";
+        String prefix = "10.10.10.10/32";
+        String nexthop = "100.100.100.100";
+        int label = 1234;
+
+        bgpFibWriter.addFibEntryToDS(rd, prefix, nexthop, label);
+        assertEquals(1, fibManager.getDataChgCount());
+
+        prefix = "10.10.10.11/32";
+        label = 3456;
+        bgpFibWriter.addFibEntryToDS(rd, prefix, nexthop, label);
+        assertEquals(2, fibManager.getDataChgCount());
+
+
+    }
+
+    @Test
+    public void testAddPrefixesAcrossRd() {
+        String rd = "101";
+        String prefix = "10.10.10.10/32";
+        String nexthop = "100.100.100.100";
+        int label = 1234;
+
+        bgpFibWriter.addFibEntryToDS(rd, prefix, nexthop, label);
+        assertEquals(1, fibManager.getDataChgCount());
+
+        rd = "102";
+        prefix = "10.10.10.11/32";
+        nexthop = "200.200.200.200";
+        label = 3456;
+        bgpFibWriter.addFibEntryToDS(rd, prefix, nexthop, label);
+        assertEquals(2, fibManager.getDataChgCount());
+
+    }
+
+
+    @Test
+    public void testRemovePrefix() {
+        String rd = "101";
+        String prefix = "10.10.10.10/32";
+        String nexthop = "100.100.100.100";
+        int label = 1234;
+
+        //add and then remove prefix
+        bgpFibWriter.addFibEntryToDS(rd, prefix, nexthop, label);
+        assertEquals(1, fibManager.getDataChgCount());
+        bgpFibWriter.removeFibEntryFromDS(rd, prefix);
+        assertEquals(0, fibManager.getDataChgCount());
+
+    }
+
+}

diff --git a/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/MockFibManager.java b/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/MockFibManager.java
new file mode 100644 (file)
index 0000000..dfbbf95
--- /dev/null
+++ b/bgpmanager/bgpmanager-impl/src/test/java/org/opendaylight/vpnservice/bgpmanager/test/MockFibManager.java
@@ -0,0 +1,66 @@
+package org.opendaylight.vpnservice.test;
+
+import java.util.Collection;
+
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
+import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
+import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.FibEntries;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.fibentries.VrfTables;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.vpnservice.fibmanager.rev150330.vrfentries.VrfEntry;
+import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+
+public class MockFibManager extends AbstractMockFibManager<VrfEntry> {
+
+    private int nFibEntries = 0;
+
+    private ListenerRegistration<MockFibManager> listenerRegistration;
+
+    public MockFibManager( final DataBroker db) {
+        super() ;
+        registerListener(db) ;
+    }
+
+    private void registerListener(final DataBroker db) {
+        final DataTreeIdentifier<VrfEntry> treeId = new DataTreeIdentifier<>(LogicalDatastoreType.CONFIGURATION, getWildCardPath());
+        try {
+            listenerRegistration = db.registerDataTreeChangeListener(treeId, MockFibManager.this);
+        } catch (final Exception e) {
+            throw new IllegalStateException("FibManager registration Listener fail! System needs restart.", e);
+        }
+    }
+
+    private InstanceIdentifier<VrfEntry> getWildCardPath() {
+            return InstanceIdentifier.create(FibEntries.class).child(VrfTables.class).child(VrfEntry.class);
+     }
+
+    @Override
+    public void onDataTreeChanged(Collection<DataTreeModification<VrfEntry>> changes) {
+        for (DataTreeModification<VrfEntry> change : changes) {
+            final InstanceIdentifier<VrfEntry> key = change.getRootPath().getRootIdentifier();
+            final DataObjectModification<VrfEntry> mod = change.getRootNode();
+
+                switch (mod.getModificationType()) {
+                case DELETE:
+                    nFibEntries -= 1;
+                    break;
+                case WRITE:
+                    if (mod.getDataBefore() == null) {
+                        nFibEntries += 1;
+                    } else {
+                        // UPDATE COUNT UNCHANGED
+                    }
+                    break;
+                default:
+                    throw new IllegalArgumentException("Unhandled modification type " + mod.getModificationType());
+                }
+            }
+     }
+
+    public int getDataChgCount() {
+        return nFibEntries;
+    }
+}