opendaylight/md-sal/statistics-manager/src/test/java/test/mock/NodeRegistrationTest.java

   1 package test.mock;
   2
   3 import static org.junit.Assert.assertFalse;
   4 import static org.junit.Assert.assertTrue;
   5
   6 import java.util.concurrent.ExecutionException;
   7
   8 import org.opendaylight.controller.md.statistics.manager.StatisticsManagerActivator;
   9 import org.opendaylight.controller.md.statistics.manager.StatisticsManagerProvider;
  10 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  11 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  12 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  13
  14 import test.mock.util.StatisticsManagerTest;
  15
  16 public class NodeRegistrationTest extends StatisticsManagerTest {
  17
  18 //    @Test
  19     public void nodeRegistrationTest() throws ExecutionException, InterruptedException {
  20         final StatisticsManagerActivator activator = new StatisticsManagerActivator();
  21         final StatisticsManagerProvider statisticsManagerProvider = new StatisticsManagerProvider(activator);
  22         activator.onSessionInitiated(providerContext);
  23
  24         addFlowCapableNode(s1Key);
  25         Thread.sleep(1000);
  26         final InstanceIdentifier<Node> nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, s1Key);
  27
  28         assertTrue(statisticsManagerProvider.getStatisticsManager().isProvidedFlowNodeActive(nodeII));
  29     }
  30
  31 //    @Test
  32     public void nodeUnregistrationTest() throws ExecutionException, InterruptedException {
  33         final StatisticsManagerActivator activator = new StatisticsManagerActivator();
  34         final StatisticsManagerProvider statisticsManagerProvider = new StatisticsManagerProvider(activator);
  35         activator.onSessionInitiated(providerContext);
  36
  37         addFlowCapableNode(s1Key);
  38         Thread.sleep(1000);
  39         final InstanceIdentifier<Node> nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, s1Key);
  40
  41         assertTrue(statisticsManagerProvider.getStatisticsManager().isProvidedFlowNodeActive(nodeII));
  42
  43         removeNode(s1Key);
  44         Thread.sleep(1000);
  45         assertFalse(statisticsManagerProvider.getStatisticsManager().isProvidedFlowNodeActive(nodeII));
  46     }
  47 }
  48