v1.1.0b1

packages/bthlabs-jsonrpc-core/tests/codecs/test_JSONCodec.py

@@ -0,0 +1,121 @@
+# -*- coding: utf-8 -*-
+# type: ignore
+from __future__ import annotations
+
+import json
+from unittest import mock
+
+from bthlabs_jsonrpc_core import codecs
+
+
+def test_decode():
+    # Given
+    json_codec = codecs.JSONCodec()
+
+    # When
+    result = json_codec.decode('{"spam": true, "eggs": false}')
+
+    # Then
+    expected_result = {
+        'spam': True,
+        'eggs': False,
+    }
+    assert result == expected_result
+
+
+@mock.patch.object(codecs.json, 'loads')
+def test_decode_decoder_kwargs(mock_json_loads: mock.Mock):
+    # Given
+    mock_json_loads.return_value = 'spam'
+
+    json_codec = codecs.JSONCodec()
+
+    payload = '{"spam": true, "eggs": false}'
+    fake_json_decoder = mock.Mock(spec=json.JSONDecoder)
+    fake_object_hook = mock.Mock()
+    fake_parse_float = mock.Mock()
+    fake_parse_int = mock.Mock()
+    fake_parse_constant = mock.Mock()
+    fake_object_pairs_hook = mock.Mock()
+
+    # When
+    _ = json_codec.decode(
+        payload,
+        cls=fake_json_decoder,
+        object_hook=fake_object_hook,
+        parse_float=fake_parse_float,
+        parse_int=fake_parse_int,
+        parse_constant=fake_parse_constant,
+        object_pairs_hook=fake_object_pairs_hook,
+    )
+
+    # Then
+    mock_json_loads.assert_called_once_with(
+        payload,
+        cls=fake_json_decoder,
+        object_hook=fake_object_hook,
+        parse_float=fake_parse_float,
+        parse_int=fake_parse_int,
+        parse_constant=fake_parse_constant,
+        object_pairs_hook=fake_object_pairs_hook,
+    )
+
+
+def test_encode():
+    # Given
+    json_codec = codecs.JSONCodec()
+
+    # When
+    result = json_codec.encode({'spam': True, 'eggs': False})
+
+    # Then
+    expected_result = '{"spam": true, "eggs": false}'
+    assert result == expected_result
+
+
+@mock.patch.object(codecs.json, 'dumps')
+def test_encode_encoder_kwargs(mock_json_dumps: mock.Mock):
+    # Given
+    mock_json_dumps.return_value = 'spam'
+
+    json_codec = codecs.JSONCodec()
+
+    payload = {"spam": True, "eggs": False}
+    fake_json_encoder = mock.Mock(spec=json.JSONEncoder)
+
+    # When
+    _ = json_codec.encode(
+        payload,
+        skipkeys=False,
+        ensure_ascii=True,
+        check_circular=True,
+        allow_nan=True,
+        cls=fake_json_encoder,
+        indent=2, separators=(':', ','),
+        default='DEFAULT',
+        sort_keys=False,
+    )
+
+    # Then
+    mock_json_dumps.assert_called_once_with(
+        payload,
+        skipkeys=False,
+        ensure_ascii=True,
+        check_circular=True,
+        allow_nan=True,
+        cls=fake_json_encoder,
+        indent=2, separators=(':', ','),
+        default='DEFAULT',
+        sort_keys=False,
+    )
+
+
+def test_get_content_type():
+    # Given
+    json_codec = codecs.JSONCodec()
+
+    # When
+    result = json_codec.get_content_type()
+
+    # Then
+    assert result == 'application/json'