Added unit tests for FlowStats and FlowStatsRequest - Issue #158

kytos · Sep 14, 2016 · 0d7a991 · 0d7a991
1 parent 271751e
commit 0d7a991
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/tests/v0x04/test_controller2switch/test_flow_stats.py b/tests/v0x04/test_controller2switch/test_flow_stats.py
@@ -0,0 +1,15 @@
+"""Flow stats message."""
+from pyof.v0x04.controller2switch.common import FlowStats
+from tests.test_struct import TestStruct
+
+
+class TestFlowStats(TestStruct):
+    """Flow stats message."""
+
+    @classmethod
+    def setUpClass(cls):
+        """Configure raw file and its object in parent class (TestDump)."""
+        super().setUpClass()
+        super().set_raw_dump_file('v0x04', 'ofpt_flow_stats')
+        super().set_raw_dump_object(FlowStats)
+        super().set_minimum_size(56)
diff --git a/tests/v0x04/test_controller2switch/test_flow_stats_request.py b/tests/v0x04/test_controller2switch/test_flow_stats_request.py
@@ -0,0 +1,15 @@
+"""Flow stats request message."""
+from pyof.v0x04.controller2switch.common import FlowStatsRequest
+from tests.test_struct import TestStruct
+
+
+class FlowStatsRequest(TestStruct):
+    """Flow stats request message."""
+
+    @classmethod
+    def setUpClass(cls):
+        """Configure raw file and its object in parent class (TestDump)."""
+        super().setUpClass()
+        super().set_raw_dump_file('v0x04', 'ofpt_flow_stats_request')
+        super().set_raw_dump_object(FlowStatsRequest)
+        super().set_minimum_size(40)