Refactor test assertions to remove whitespace in JSON strings for consistency

Author: grongierisc

src/tests/test_business_host.py

@@ -52,7 +52,7 @@ def test_dispatch_serializer_valid(self, business_host):
         rsp = dispatch_serializer(message)
 
         assert rsp.classname == 'registerFilesIop.message.SimpleMessage'
-        assert rsp.GetObjectJson() == '{"integer": 1, "string": "test"}'
+        assert rsp.GetObjectJson() == '{"integer":1,"string":"test"}'
 
     def test_dispatch_serializer_none(self, business_host):
         assert dispatch_serializer(None) is None
@@ -62,27 +62,27 @@ def test_dispatch_serializer_none(self, business_host):
         (SimpleMessageNotDataclass(), TypeError),
         ("test", TypeError)
     ])
-    def test_dispatch_serializer_invalid(self, business_host, invalid_message, expected_error):
+    def test_dispatch_serializer_invalid(self, invalid_message, expected_error):
         with pytest.raises(expected_error):
             dispatch_serializer(invalid_message)
 
 class TestMessageDeserialization:
-    def test_serialize_deserialize_simple(self, business_host):
+    def test_serialize_deserialize_simple(self):
         original = SimpleMessage(integer=1, string='test')
         serialized = serialize_message(original)
         deserialized = deserialize_message(serialized)
 
         assert deserialized.integer == original.integer
         assert deserialized.string == original.string
 
-    def test_serialize_deserialize_japanese(self, business_host):
+    def test_serialize_deserialize_japanese(self):
         original = SimpleMessage(integer=1, string='あいうえお')
         serialized = serialize_message(original)
         deserialized = deserialize_message(serialized)
 
         assert deserialized.string == 'あいうえお'
 
-    def test_serialize_deserialize_large_string(self, business_host):
+    def test_serialize_deserialize_large_string(self):
         huge_string = 'test' * 1000000
         original = SimpleMessage(integer=1, string=huge_string)
         serialized = serialize_message(original)
@@ -91,7 +91,7 @@ def test_serialize_deserialize_large_string(self, business_host):
         assert deserialized.string == huge_string
 
 class TestPickledMessages:
-    def test_pickled_message_roundtrip(self, business_host):
+    def test_pickled_message_roundtrip(self):
         original = PickledMessage(integer=1, string='test')
         serialized = serialize_pickle_message(original)
         deserialized = deserialize_pickle_message(serialized)

src/tests/test_dispatch.py

@@ -64,6 +64,23 @@ def test_simple_message_serialization():
     assert result.text == msg.text
     assert result.number == msg.number
 
+def test_simple_message_serialization_with_extra():
+    msg = MessageTest(text="test", number=42)
+    msg.extra_field = "extra"
+    
+    # Test serialization
+    serial = serialize_message(msg)
+    assert type(serial).__module__.startswith('iris')
+    assert serial._IsA("IOP.Message")
+    assert serial.classname == f"{MessageTest.__module__}.{MessageTest.__name__}"
+    
+    # Test deserialization
+    result = deserialize_message(serial)
+    assert isinstance(result, MessageTest)
+    assert result.text == msg.text
+    assert result.number == msg.number
+    assert result.extra_field == msg.extra_field
+
 def test_message_serialization():
     msg = MessageTest(text="test", number=42)
 
@@ -79,6 +96,21 @@ def test_message_serialization():
     assert result.text == msg.text
     assert result.number == msg.number
 
+def test_message_serialization_wrong_type():
+    msg = MessageTest(text="test", number={"key": "value"})
+    
+    # Test serialization
+    serial = serialize_message(msg)
+    assert type(serial).__module__.startswith('iris')
+    assert serial._IsA("IOP.Message")
+    assert serial.classname == f"{MessageTest.__module__}.{MessageTest.__name__}"
+    
+    # Test deserialization
+    result = deserialize_message(serial)
+    assert isinstance(result, MessageTest)
+    assert result.text == msg.text
+    assert result.number == msg.number
+
 def test_pickle_message_serialization():
     msg = SimpleModel(text="test", number=42)
 

src/tests/test_private_session.py

@@ -54,7 +54,7 @@ def test_dispatch_on_private_session_stopped(self, process):
             result = process._dispatch_on_private_session_stopped(host_object, 'test_source', 'self_generated', test_request)
 
             mock_handler.assert_called_once()
-            assert result.json == '{"value": "test"}'
+            assert result.json == '{"value":"test"}'
 
     def test_dispatch_on_private_session_started(self, process):
         host_object = MagicMock()
@@ -64,7 +64,7 @@ def test_dispatch_on_private_session_started(self, process):
             result = process._dispatch_on_private_session_started(host_object, 'test_source', test_request)
 
             mock_handler.assert_called_once()
-            assert result.json == '{"value": "test"}'
+            assert result.json == '{"value":"test"}'
 
     def test_dispatch_on_document(self, process):
         host_object = MagicMock()
@@ -74,4 +74,4 @@ def test_dispatch_on_document(self, process):
             result = process._dispatch_on_document(host_object, 'test_source', test_request)
 
             mock_handler.assert_called_once()
-            assert result.json == '{"value": "test"}'
+            assert result.json == '{"value":"test"}'

src/tests/test_serialization.py

@@ -37,7 +37,7 @@ def test_json_serialization():
     # Create test data
     test_items = [{'col1': 1, 'col2': 'a'}, {'col1': 2, 'col2': 'b'}]  # Simple list of dicts instead of DataFrame
     test_uuid = uuid.uuid4()
-    test_bytes = b'hello world'
+    test_bytes = b'hello world\x04'
 
     msg = FullMessge(
         text="test",