BUG-2613: Migrating Openflow Specific NSF from controller project to openflowplugin...

[openflowplugin.git] / applications / statistics-manager / src / test / java / test / mock / NodeRegistrationTest.java

diff --git a/applications/statistics-manager/src/test/java/test/mock/NodeRegistrationTest.java b/applications/statistics-manager/src/test/java/test/mock/NodeRegistrationTest.java
new file mode 100644 (file)
index 0000000..79eb236
--- /dev/null
+++ b/applications/statistics-manager/src/test/java/test/mock/NodeRegistrationTest.java
@@ -0,0 +1,43 @@
+package test.mock;
+
+import org.junit.Test;
+import org.opendaylight.controller.md.statistics.manager.StatisticsManager;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import test.mock.util.StatisticsManagerTest;
+
+import java.util.concurrent.ExecutionException;
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+public class NodeRegistrationTest extends StatisticsManagerTest {
+
+    @Test
+    public void nodeRegistrationTest() throws ExecutionException, InterruptedException {
+        StatisticsManager statisticsManager = setupStatisticsManager();
+
+        addFlowCapableNode(s1Key);
+        Thread.sleep(2000);
+        final InstanceIdentifier<Node> nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, s1Key);
+
+        assertTrue(statisticsManager.isProvidedFlowNodeActive(nodeII));
+    }
+
+    @Test
+    public void nodeUnregistrationTest() throws ExecutionException, InterruptedException {
+        StatisticsManager statisticsManager = setupStatisticsManager();
+
+        addFlowCapableNode(s1Key);
+        Thread.sleep(2000);
+        final InstanceIdentifier<Node> nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, s1Key);
+
+        assertTrue(statisticsManager.isProvidedFlowNodeActive(nodeII));
+
+        removeNode(s1Key);
+        Thread.sleep(2000);
+        assertFalse(statisticsManager.isProvidedFlowNodeActive(nodeII));
+    }
+}
+