BgpManager JUnits for add/delete fib entries

[vpnservice.git] / bgpmanager / bgpmanager-impl / src / test / java / org / opendaylight / vpnservice / bgpmanager / test / MockFibManager.java

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;
+    }
+}