def assert_db(db_dict):
def assert_model(model, model_dict):
def raise_assertion_error(key, actual_value, expected_value):
raise AssertionError(
f'Value of `{key}` is `{actual_value}` but should be `{expected_value}`',
)
for key, expected_value in model_dict.items():
actual_value = getattr(model, key)
if isinstance(expected_value, datetime.datetime):
if abs(actual_value - expected_value) > TIME_TOLERANCE:
raise_assertion_error(key, actual_value, expected_value)
elif hasattr(expected_value,
'match'): # Regular expression pattern.
if expected_value.match(actual_value) is None:
raise AssertionError(
f'Value of `{key}` is `{actual_value}` and doesn\'t match'
f' `{expected_value}`', )
elif actual_value != expected_value:
raise_assertion_error(key, actual_value, expected_value)
for model_name, models_dicts in db_dict.items():
if model_name == 'strangers':
for stranger_dict in models_dicts:
stranger_instance = Stranger.get(id=stranger_dict['id'])
assert_model(stranger_instance, stranger_dict)
elif model_name == 'talks':
for talk_dict, talk_instance in zip(models_dicts, Talk.select()):
assert_model(talk_instance, talk_dict)
else:
raise AssertionError(f'Unknown model name: `{model_name}`')
