def test_matching_record_filtering_self(self):
hasher = HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user)
hasher.save()
self.assertListEqual(HasherDataModel.matching_records(hasher), list())
def test_save_same_record_twice(self):
start = HasherDataModel('hasher1',
mother_kennel=self.kennel,
hash_name=self.name)
start.save()
start.save()
self.assertTrue(HasherDataModel.record_exists(start))
def test_record_exists(self):
hasher = HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user)
hasher.save()
self.assertTrue(HasherDataModel.record_exists(hasher))
def test_nullable_fields(self):
hasher = HasherDataModel(self.hasher_id,
hash_name=self.name,
mother_kennel=self.kennel)
hasher.save()
self.assertIsNone(hasher.contact_info)
self.assertIsNone(hasher.real_name)
self.assertIsNone(hasher.user)
def test_save_with_existing_record(self):
HasherDataModel('hasher1',
mother_kennel=self.kennel,
hash_name=self.name).save()
with self.assertRaises(AlreadyExists):
HasherDataModel('hasher2',
mother_kennel=self.kennel,
hash_name=self.name).save()
def test_is_not_ref(self):
mdl = HasherDataModel('hasher1',
mother_kennel=self.kennel,
hash_name=self.name)
mdl2 = HasherDataModel('diff',
mother_kennel=self.kennel,
hash_name=self.name)
ref = mdl2.to_ref()
self.assertFalse(ref.is_ref(mdl))
def test_matching_record_with_different_users(self):
HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user).save()
hasher = HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user='******')
self.assertListEqual(HasherDataModel.matching_records(hasher), list())
def test_timestamps(self):
time = datetime.now(timezone.utc)
with freeze_time(time):
hasher = HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user)
hasher.save()
self.assertEqual(hasher.modified_at, time)
self.assertEqual(hasher.created_at, time)
def test_save_with_no_user_existing_record(self):
start = HasherDataModel('hasher1',
mother_kennel=self.kennel,
hash_name=self.name)
start.save()
hasher = HasherDataModel('hasher1',
mother_kennel=self.kennel,
hash_name=self.name,
user='******')
hasher.save()
self.assertTrue(HasherDataModel.record_exists(start))
self.assertTrue(HasherDataModel.record_exists(hasher))
def setUp(self):
if KennelDataModel.exists():
KennelDataModel.delete_table()
KennelDataModel.create_table(read_capacity_units=1,
write_capacity_units=1,
wait=True)
self.hasher_id = 'test_id'
self.hash_name = 'Testy Cream'
self.mother_kennel = KennelLogicModel.create('Test Kennel 1', 'TK1H3')
if HasherDataModel.exists():
HasherDataModel.delete_table()
HasherDataModel.create_table(read_capacity_units=1,
write_capacity_units=1,
wait=True)
def setUp(self):
self.hasher_id = 'test_id'
self.name = 'test_hasher'
self.kennel = {
'kennel_id': 'test_kennel_1',
'name': 'Test Kennel 1',
'acronym': 'TK1H3'
}
self.user = '******'
if HasherDataModel.exists():
HasherDataModel.delete_table()
HasherDataModel.create_table(read_capacity_units=1,
write_capacity_units=1,
wait=True)
def test_kennel_required(self):
with self.assertRaises(ValueError):
HasherDataModel(
self.hasher_id,
lower_hash_name=self.name.lower(),
hash_name=self.name,
mother_kennel_name_lower=self.kennel['name'].lower()).save()
def test_hash_name_update(self):
hasher = HasherDataModel(self.hasher_id,
hash_name=self.name,
mother_kennel=self.kennel)
hasher.save()
hasher.add_update_action('hash_name', 'set', 'Different Hash Name')
hasher.update()
self.assertEqual(hasher.searchable_hash_name,
'differenthashname'.lower())
def test_mother_kennel_update(self):
kennel2 = {
'kennel_id': 'test_kennel_2',
'name': 'Test Kennel 2',
'acronym': 'TK2H3'
}
hasher = HasherDataModel(self.hasher_id,
hash_name=self.name,
mother_kennel=self.kennel)
hasher.save()
hasher.add_update_action('mother_kennel', 'set', kennel2)
hasher.update()
self.assertEqual(hasher.searchable_mother_kennel_name, 'testkennel2')
def test_timestamps_on_update(self):
new_kennel_name = 'New Kennel 1'
start_time = datetime.now(timezone.utc)
with freeze_time(start_time):
hasher = HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user)
hasher.save()
mid_time = datetime.now(timezone.utc)
with freeze_time(mid_time):
hasher.add_update_action('mother_kennel.name', 'set',
new_kennel_name)
hasher.update()
self.assertEqual(hasher.modified_at, mid_time)
self.assertEqual(hasher.created_at, start_time)
self.assertEqual(hasher.mother_kennel.name, new_kennel_name)
def test_init_from_lookup(self):
HasherDataModel(
self.hasher_id,
hash_name=self.hash_name,
lower_hash_name=self.hash_name.lower(),
mother_kennel=HasherLogicModel.map_mother_kennel(
self.mother_kennel),
mother_kennel_name_lower=self.mother_kennel.name.lower()).save()
actual = HasherLogicModel.lookup_by_id(self.hasher_id)
self.assertEqual(self.hasher_id, actual.hasher_id)
self.assertEqual(self.hash_name, actual.hash_name)
self.assertEqual(self.mother_kennel.kennel_id,
actual.mother_kennel.kennel_id)
self.assertEqual(self.mother_kennel.name, actual.mother_kennel.name)
self.assertEqual(self.mother_kennel.acronym,
actual.mother_kennel.acronym)
def test_matching_records_multiple_close_matches(self):
HasherDataModel(
'hasher1',
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user,
).save()
two = HasherDataModel('hasher2',
mother_kennel=self.kennel,
hash_name=self.name,
user='******')
two.save()
HasherDataModel('hasher3',
mother_kennel=self.kennel,
hash_name=self.name,
user='******').save()
hasher = HasherDataModel('hasher4',
mother_kennel=self.kennel,
hash_name=self.name,
user='******')
self.assertListEqual(HasherDataModel.matching_records(hasher), [two])
def __init__(self,
hash_name,
mother_kennel,
hasher_id=None,
contact_info=None,
real_name=None,
user=None,
persistence_object=None):
super().__init__()
self.hasher_id = ulid() if hasher_id is None else hasher_id
self.hash_name = hash_name
if hasattr(mother_kennel, 'create'):
self.mother_kennel = self.map_mother_kennel(mother_kennel)
else:
self.mother_kennel = mother_kennel
self.contact_info = contact_info
self.real_name = real_name
self.user = user
if persistence_object is None:
self.persistence_object = HasherDataModel(
**self.persistable_attributes())
else:
self.persistence_object = persistence_object
def lookup_by_id(cls, hasher_id):
result = HasherDataModel.get(hasher_id)
attribute_dict = result.attributes()
attribute_dict['persistence_object'] = result
attribute_dict['hasher_id'] = result.hasher_id
return HasherLogicModel(**attribute_dict)
def test_record_does_not_exist(self):
hasher = HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user)
self.assertFalse(HasherDataModel.record_exists(hasher))
def test_matching_records_does_not_exist(self):
hasher = HasherDataModel(self.hasher_id,
mother_kennel=self.kennel,
hash_name=self.name,
user=self.user)
self.assertListEqual(HasherDataModel.matching_records(hasher), list())
def test_is_ref(self):
mdl = HasherDataModel('hasher1',
mother_kennel=self.kennel,
hash_name=self.name)
ref = mdl.to_ref()
self.assertTrue(ref.is_ref(mdl))
def tearDown(self):
if HasherDataModel.exists():
HasherDataModel.delete_table()
