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

# -*- 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'