def test_decode_local_key_instance(self):
@pymongo
@jsonclass
class SimpleDecodeLocalKeyInstanceAddress:
id: str = types.readonly.str.primary.mongoid.required
city: str
owner: SimpleDecodeLocalKeyInstance = types.objof(
'SimpleDecodeLocalKeyInstance').linkedby('address')
@pymongo
@jsonclass
class SimpleDecodeLocalKeyInstance:
id: str = types.readonly.str.primary.mongoid.required
address: SimpleDecodeLocalKeyInstanceAddress = (types
.linkto.objof(
SimpleDecodeLocalKeyInstanceAddress))
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'addressId': ObjectId()
}
instance = Decoder().decode_root(data, SimpleDecodeLocalKeyInstance)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(getattr(instance, 'address_id'),
str(data['addressId']))
def test_decode_embedded_instance(self):
@pymongo
@jsonclass
class SimpleDecodeInstanceAddress:
id: str = types.readonly.str.primary.mongoid.required
city: str
@pymongo
@jsonclass
class SimpleDecodeInstance:
id: str = types.readonly.str.primary.mongoid.required
address: SimpleDecodeInstanceAddress
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'address': {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'city': 'Shanghai'
}
}
instance = Decoder().decode_root(data, SimpleDecodeInstance)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(instance.address.id, str(data['address']['_id']))
self.assertEqual(instance.address.city, "Shanghai")
def test_decode_local_keys_list(self):
@pymongo
@jsonclass
class SimpleDecodeLocalKeyListAddress:
id: str = types.readonly.str.primary.mongoid.required
city: str
owner: SimpleDecodeLocalKeyList = types.objof(
'SimpleDecodeLocalKeyList').linkedby('address')
@pymongo
@jsonclass
class SimpleDecodeLocalKeyList:
id: str = types.readonly.str.primary.mongoid.required
addresses: List[SimpleDecodeLocalKeyListAddress] = (types.linkto
.listof(SimpleDecodeLocalKeyListAddress))
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'addressIds': [ObjectId(), ObjectId()]
}
instance = Decoder().decode_root(data, SimpleDecodeLocalKeyList)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(getattr(instance, 'address_ids')
[0], str(data['addressIds'][0]))
self.assertEqual(getattr(instance, 'address_ids')
[1], str(data['addressIds'][1]))
def test_decode_keep_dict_keys(self):
@pymongo
@jsonclass
class MediumDecodeCamelizeDictKeys:
id: str = types.readonly.str.primary.mongoid.required
val: Dict[str, str] = types.dictof(types.str)
data = {
'val': {
'keyOne': 'val_one',
'keyTwo': 'val_two'
}
}
instance = Decoder().decode_root(data, MediumDecodeCamelizeDictKeys)
self.assertEqual(
instance.val, {'keyOne': 'val_one', 'keyTwo': 'val_two'})
def test_decode_embedded_dict(self):
@pymongo
@jsonclass
class SimpleDecodeDict:
id: str = types.readonly.str.primary.mongoid.required
vals: Dict[str, int]
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'vals': {'one': 1, 'two': 2}
}
instance = Decoder().decode_root(data, SimpleDecodeDict)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(instance.vals, {'one': 1, 'two': 2})
def test_decode_embedded_list(self):
@pymongo
@jsonclass
class SimpleDecodeList:
id: str = types.readonly.str.primary.mongoid.required
vals: List[int]
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'vals': [1, 2, 3, 4, 5]
}
instance = Decoder().decode_root(data, SimpleDecodeList)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(instance.vals, [1, 2, 3, 4, 5])
def test_decode_datetime_into_date(self):
@pymongo
@jsonclass
class SimpleDecodeDate:
id: str = types.readonly.str.primary.mongoid.required
val1: date
val2: date
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'val1': datetime(2012, 9, 5, 0, 0, 0),
'val2': datetime(2020, 9, 5, 0, 0, 0),
}
instance = Decoder().decode_root(data, SimpleDecodeDate)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(instance.val1, date(2012, 9, 5))
self.assertEqual(instance.val2, date(2020, 9, 5))
def test_decode_bool_into_bool(self):
@pymongo
@jsonclass
class SimpleDecodeBool:
id: str = types.readonly.str.primary.mongoid.required
val1: bool
val2: bool
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'val1': True,
'val2': False
}
instance = Decoder().decode_root(data, SimpleDecodeBool)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(instance.val1, True)
self.assertEqual(instance.val2, False)
def test_decode_float_into_float(self):
@pymongo
@jsonclass
class SimpleDecodeFloat:
id: str = types.readonly.str.primary.mongoid.required
val1: float
val2: float
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'val1': 12345.6,
'val2': 67890.1
}
instance = Decoder().decode_root(data, SimpleDecodeFloat)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(instance.val1, 12345.6)
self.assertEqual(instance.val2, 67890.1)
def test_decode_str_into_str(self):
@pymongo
@jsonclass
class SimpleDecodeStr:
id: str = types.readonly.str.primary.mongoid.required
val1: str
val2: str
data = {
'_id': ObjectId(),
'createdAt': datetime.now(),
'updatedAt': datetime.now(),
'val1': '12345',
'val2': '67890'
}
instance = Decoder().decode_root(data, SimpleDecodeStr)
self.assertEqual(instance.id, str(data['_id']))
self.assertEqual(instance.val1, '12345')
self.assertEqual(instance.val2, '67890')
self.assertIsInstance(instance.val1, str)
self.assertIsInstance(instance.val2, str)