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

   1 package test.mock;
   2
   3 import static org.junit.Assert.assertEquals;
   4 import static org.junit.Assert.assertTrue;
   5
   6 import java.util.concurrent.ExecutionException;
   7
   8 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
   9 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  10 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
  11 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  12 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  13 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  14 import org.opendaylight.controller.md.statistics.manager.StatisticsManagerActivator;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowFeatureCapabilityTableStats;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.table.statistics.rev131215.FlowTableStatisticsData;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.TableId;
  23 import org.opendaylight.yangtools.yang.binding.DataObject;
  24 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  25
  26 import test.mock.util.StatisticsManagerTest;
  27
  28 import com.google.common.base.Optional;
  29
  30 public class TableStatisticsTest extends StatisticsManagerTest {
  31     private final Object waitObject = new Object();
  32
  33 //    @Test(timeout = 23000)
  34     public void getTableStatisticsTest() throws ExecutionException, InterruptedException, ReadFailedException {
  35         final StatisticsManagerActivator activator = new StatisticsManagerActivator();
  36         activator.onSessionInitiated(providerContext);
  37
  38         addFlowCapableNodeWithFeatures(s1Key, false, FlowFeatureCapabilityTableStats.class);
  39
  40         final TableId tableId = getTableId();
  41         final InstanceIdentifier<Table> tableII = InstanceIdentifier.create(Nodes.class).child(Node.class, s1Key)
  42                 .augmentation(FlowCapableNode.class).child(Table.class, new TableKey(tableId.getValue()));
  43
  44         getDataBroker().registerDataChangeListener(LogicalDatastoreType.OPERATIONAL,
  45                 tableII.augmentation(FlowTableStatisticsData.class), new ChangeListener(), AsyncDataBroker.DataChangeScope.BASE);
  46
  47         synchronized (waitObject) {
  48             waitObject.wait();
  49         }
  50
  51         final ReadOnlyTransaction readTx = getDataBroker().newReadOnlyTransaction();
  52         final Optional<FlowTableStatisticsData> flowTableStatisticsDataOptional = readTx.read(
  53                 LogicalDatastoreType.OPERATIONAL, tableII.augmentation(FlowTableStatisticsData.class)).checkedGet();
  54         assertTrue(flowTableStatisticsDataOptional.isPresent());
  55         assertEquals(COUNTER_32_TEST_VALUE,
  56                 flowTableStatisticsDataOptional.get().getFlowTableStatistics().getActiveFlows());
  57         assertEquals(COUNTER_64_TEST_VALUE,
  58                 flowTableStatisticsDataOptional.get().getFlowTableStatistics().getPacketsLookedUp());
  59     }
  60
  61     private class ChangeListener implements DataChangeListener {
  62
  63         @Override
  64         public void onDataChanged(final AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
  65             synchronized (waitObject) {
  66                 waitObject.notify();
  67             }
  68         }
  69     }
  70 }
  71