Update tests for output processor.

Author: ellisonbg

jupyter_rtc_core/tests/test_output_processor.py

@@ -9,20 +9,21 @@ class TestOutputProcessor(OutputProcessor):
 
     settings = {}
 
-
-def test_instantiation():
-    op = OutputProcessor()
-
 def create_incoming_message(cell_id):
     msg_id = str(uuid4())
-    header = {"msg_id": msg_id, "msg_type": "shell"}
+    header = {"msg_id": msg_id, "msg_type": "execute_request"}
     parent_header = {}
     metadata = {"cellId": cell_id}
     msg = [json.dumps(item) for item in [header, parent_header, metadata]]
     return msg_id, msg
 
-def test_incoming_message():
+def test_instantiation():
+    """Test instantiation of the output processor."""
+    op = OutputProcessor()
+    assert isinstance(op, OutputProcessor)
 
+def test_incoming_message():
+    """Test incoming message processing."""
     with TemporaryDirectory() as td:
         op = TestOutputProcessor()
         om = OutputsManager()
@@ -34,18 +35,19 @@ def test_incoming_message():
         op.process_incoming_message('shell', msg)
         assert op.get_cell_id(msg_id) == cell_id
         assert op.get_msg_id(cell_id) == msg_id
-        # # Clear the cell_id
-        # op.clear(cell_id)
-        # assert op.get_cell_id(msg_id) is None
-        # assert op.get_msg_id(cell_id) is None
-        # # Simulate the running of a cell
-        # cell_id = str(uuid4())
-        # msg_id, msg = create_incoming_message(cell_id)
-        # op.process_incoming_message('shell', msg)
-        # assert op.get_cell_id(msg_id) == cell_id
-        # assert op.get_msg_id(cell_id) == msg_id
+        # Clear the cell_id
+        op.clear(cell_id)
+        assert op.get_cell_id(msg_id) is None
+        assert op.get_msg_id(cell_id) is None
+        # Simulate the running of a cell
+        cell_id = str(uuid4())
+        msg_id, msg = create_incoming_message(cell_id)
+        op.process_incoming_message('shell', msg)
+        assert op.get_cell_id(msg_id) == cell_id
+        assert op.get_msg_id(cell_id) == msg_id
         # # Run it again without clearing to ensure it self clears
-        # cell_id = str(uuid4())
-        # msg_id, msg = create_incoming_message(cell_id)
-        # assert op.get_cell_id(msg_id) == cell_id
-        # assert op.get_msg_id(cell_id) == msg_id
+        cell_id = str(uuid4())
+        msg_id, msg = create_incoming_message(cell_id)
+        op.process_incoming_message('shell', msg)
+        assert op.get_cell_id(msg_id) == cell_id
+        assert op.get_msg_id(cell_id) == msg_id