bemaniutils/bemani/tests/test_BaseData.py

# vim: set fileencoding=utf-8
import unittest
from unittest.mock import Mock

from bemani.data.mysql.base import BaseData


class TestBaseData(unittest.TestCase):
    def test_basic_serialize(self) -> None:
        data = BaseData(Mock(), None)

        testdict = {
            "test1": 1,
            "test2": "2",
            "test3": 3.3,
            "test4": [1, 2, 3, 4],
            "test5": {
                "a": "b",
            },
            "testempty": [],
        }

        self.assertEqual(data.deserialize(data.serialize(testdict)), testdict)

    def test_basic_byte_serialize(self) -> None:
        data = BaseData(Mock(), None)

        testdict = {
            "bytes": b"\x01\x02\x03\x04\x05",
        }

        serialized = data.serialize(testdict)
        self.assertEqual(serialized, '{"bytes": ["__bytes__", 1, 2, 3, 4, 5]}')
        self.assertEqual(data.deserialize(serialized), testdict)

    def test_deep_byte_serialize(self) -> None:
        data = BaseData(Mock(), None)

        testdict = {
            "sentinal": True,
            "test": {
                "sentinal": False,
                "bytes": b"\x01\x02\x03\x04\x05",
                "bytes2": b"",
            },
        }

        self.assertEqual(data.deserialize(data.serialize(testdict)), testdict)