def test_to_mongo_no_compression(self): field = JSONDictField(use_header=True) result = field.to_mongo(MOCK_DATA_DICT) self.assertTrue(isinstance(result, bytes)) split = result.split(b":", 2) self.assertEqual(split[0], JSONDictFieldCompressionAlgorithmEnum.NONE.value) self.assertEqual(split[1], JSONDictFieldSerializationFormatEnum.ORJSON.value) self.assertEqual(orjson.loads(split[2]), MOCK_DATA_DICT) parsed_value = field.parse_field_value(result) self.assertEqual(parsed_value, MOCK_DATA_DICT)
def test_to_mongo_zstandard_compression(self): field = JSONDictField(use_header=True, compression_algorithm="zstandard") result = field.to_mongo(MOCK_DATA_DICT) self.assertTrue(isinstance(result, bytes)) split = result.split(b":", 2) self.assertEqual( split[0], JSONDictFieldCompressionAlgorithmEnum.ZSTANDARD.value ) self.assertEqual(split[1], JSONDictFieldSerializationFormatEnum.ORJSON.value) self.assertEqual( orjson.loads(zstandard.ZstdDecompressor().decompress(split[2])), MOCK_DATA_DICT, ) parsed_value = field.parse_field_value(result) self.assertEqual(parsed_value, MOCK_DATA_DICT)
def test_parse_field_value(self): # 1. Value not provided, should use default one field = JSONDictField(use_header=False, default={}) result = field.parse_field_value(b"") self.assertEqual(result, {}) result = field.parse_field_value(None) self.assertEqual(result, {}) field = JSONDictField(use_header=False, default={"foo": "bar"}) result = field.parse_field_value(b"") self.assertEqual(result, {"foo": "bar"}) result = field.parse_field_value(None) self.assertEqual(result, {"foo": "bar"}) # Value should be deserialized result = field.parse_field_value(b'{"a": "b"}') self.assertEqual(result, {"a": "b"}) # Already a dict result = field.parse_field_value({"c": "d"}) self.assertEqual(result, {"c": "d"})