Unittest - controller.py - get_preview_ports

hyades · Aug 1, 2013 · 13455e4 · 13455e4
1 parent 5cd2849
commit 13455e4
Showing 1 changed file with 15 additions and 12 deletions.
diff --git a/python-api/gstswitch/test_controller.py b/python-api/gstswitch/test_controller.py
@@ -160,15 +160,18 @@ def test_normal_unpack(self):
         assert controller.get_audio_port() == 4000
 
 
-# class TestGetPreviewPorts(object):
-
-#     def test_unpack(self):
-#         controller = Controller(address='unix:abstract=abcdefghijk')
-#         controller.connection = MockConnection(True)
-#         with pytest.raises(ConnectionError):
-#             controller.get_preview_ports()
-
-#     def test_normal_unpack(self):
-#         controller =  Controller(address='unix:abstract=abcdef')
-#         controller.connection = MockConnection(False)
-#         assert controller.get_preview_ports() == '[(3002, 1, 7), (3003, 1, 8)]'
+class TestGetPreviewPorts(object):
+
+    def test_unpack(self):
+        controller = Controller(address='unix:abstract=abcdefghijk')
+        controller.connection = MockConnection(True)
+        with pytest.raises(ConnectionError):
+            controller.get_preview_ports()
+
+    def test_normal_unpack(self):
+        controller =  Controller(address='unix:abstract=abcdef')
+        controller.connection = MockConnection(False)
+        controller._parse_preview_ports = Mock(return_value=[3001, 3002])
+        assert controller.get_preview_ports() == [3001, 3002]
+
+