3 import static org.junit.Assert.assertFalse;
4 import static org.junit.Assert.assertTrue;
6 import java.util.concurrent.ExecutionException;
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;
14 import test.mock.util.StatisticsManagerTest;
16 public class NodeRegistrationTest extends StatisticsManagerTest {
19 public void nodeRegistrationTest() throws ExecutionException, InterruptedException {
20 final StatisticsManagerActivator activator = new StatisticsManagerActivator();
21 final StatisticsManagerProvider statisticsManagerProvider = new StatisticsManagerProvider(activator);
22 activator.onSessionInitiated(providerContext);
24 addFlowCapableNode(s1Key);
26 final InstanceIdentifier<Node> nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, s1Key);
28 assertTrue(statisticsManagerProvider.getStatisticsManager().isProvidedFlowNodeActive(nodeII));
32 public void nodeUnregistrationTest() throws ExecutionException, InterruptedException {
33 final StatisticsManagerActivator activator = new StatisticsManagerActivator();
34 final StatisticsManagerProvider statisticsManagerProvider = new StatisticsManagerProvider(activator);
35 activator.onSessionInitiated(providerContext);
37 addFlowCapableNode(s1Key);
39 final InstanceIdentifier<Node> nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, s1Key);
41 assertTrue(statisticsManagerProvider.getStatisticsManager().isProvidedFlowNodeActive(nodeII));
45 assertFalse(statisticsManagerProvider.getStatisticsManager().isProvidedFlowNodeActive(nodeII));