Added unit tests for QueueStats and QueueStatsRequest - Issue #158

kytos · Sep 14, 2016 · 7f38b4e · 7f38b4e
1 parent 63dd1f8
commit 7f38b4e
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/tests/v0x04/test_controller2switch/test_queue_stats.py b/tests/v0x04/test_controller2switch/test_queue_stats.py
@@ -0,0 +1,15 @@
+"""Stats queue message."""
+from pyof.v0x04.controller2switch.common import QueueStats
+from tests.test_struct import TestStruct
+
+
+class TestQueueStats(TestStruct):
+    """Stats queue message."""
+
+    @classmethod
+    def setUpClass(cls):
+        """Configure raw file and its object in parent class (TestDump)."""
+        super().setUpClass()
+        super().set_raw_dump_file('v0x04', 'ofpt_queue_stats')
+        super().set_raw_dump_object(QueueStats)
+        super().set_minimum_size(40)
diff --git a/tests/v0x04/test_controller2switch/test_queue_stats_request.py b/tests/v0x04/test_controller2switch/test_queue_stats_request.py
@@ -0,0 +1,15 @@
+"""Queue Stat Request message."""
+from pyof.v0x04.controller2switch.common import QueueStatsRequest
+from tests.test_struct import TestStruct
+
+
+class TestQueueStatsRequest(TestStruct):
+    """Queue Stat 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_queue_stats_request')
+        super().set_raw_dump_object(QueueStatsRequest)
+        super().set_minimum_size(8)