test(s) added

Author: dwsutherland

tests/integration/test_publisher.py

@@ -32,9 +32,13 @@ async def test_publisher(flow, scheduler, run, one_conf, port_range):
             schd.workflow,
             host=schd.host,
             port=schd.server.pub_port,
-            topics=[b'workflow']
+            topics=[b'shutdown']
         )
 
+        subscriber.unsubscribe_topic(b'shutdown')
+        subscriber.subscribe_topic(b'workflow')
+        assert subscriber.topics == {b'workflow'}
+
         async with timeout(2):
             # wait for the first delta from the workflow
             btopic, msg = await subscriber.socket.recv_multipart()

tests/integration/test_server.py

@@ -79,6 +79,17 @@ def test_pb_entire_workflow(myflow):
     assert data.workflow.id == myflow.id
 
 
+def test_pb_workflow_only(myflow):
+    """Test Protobuf workflow only endpoint method."""
+    data = PB_METHOD_MAP['pb_workflow_only']()
+    data.ParseFromString(
+        call_server_method(
+            myflow.server.pb_workflow_only
+        )
+    )
+    assert data.workflow.id == myflow.id
+
+
 async def test_stop(one: Scheduler, start):
     """Test stop."""
     async with start(one):