def test_roundtrip(self): field = JSONDictField(use_header=False) result_to_mongo = field.to_mongo(MOCK_DATA_DICT) result_to_python = field.to_python(result_to_mongo) self.assertEqual(result_to_python, MOCK_DATA_DICT) # sets get serialized to a list input_dict = {"a": 1, "set": {1, 2, 3, 4, 4, 4, 5, 5}} result = {"a": 1, "set": [1, 2, 3, 4, 5]} field = JSONDictField(use_header=False) result_to_mongo = field.to_mongo(input_dict) result_to_python = field.to_python(result_to_mongo) self.assertEqual(result_to_python, result)
def test_to_python_no_compression(self): field = JSONDictField(use_header=True) serialized_data = field.to_mongo(MOCK_DATA_DICT) self.assertTrue(isinstance(serialized_data, bytes)) split = serialized_data.split(b":", 2) self.assertEqual(split[0], JSONDictFieldCompressionAlgorithmEnum.NONE.value) self.assertEqual(split[1], JSONDictFieldSerializationFormatEnum.ORJSON.value) desserialized_data = field.to_python(serialized_data) self.assertEqual(desserialized_data, MOCK_DATA_DICT)
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_roundtrip(self): field = JSONDictField(use_header=False) result_to_mongo = field.to_mongo(MOCK_DATA_DICT) result_to_python = field.to_python(result_to_mongo) self.assertEqual(result_to_python, MOCK_DATA_DICT)
def test_to_mongo(self): field = JSONDictField(use_header=False) result = field.to_mongo(MOCK_DATA_DICT) self.assertTrue(isinstance(result, bytes)) self.assertEqual(result, orjson.dumps(MOCK_DATA_DICT))
def test_header_set_to_mongo(self): field = JSONDictField(use_header=True) result = field.to_mongo({"test": {1, 2}}) self.assertTrue(isinstance(result, bytes))