def test_save_data_to_db(self, downloader_obj, modifier_obj):
persons = len(Person.select())
contacts = len(Contact.select())
logins = len(Login.select())
localizations = len(Location.select())
modifier_obj.execute_modifications()
save_obj = ApiDataSave(downloader_obj, 'results')
save_obj.save_data_to_db()
error = 'Incorrect number of objects saved in the database'
assert len(Person.select()) == persons + API_PERSONS, error
assert len(Contact.select()) == contacts + API_PERSONS, error
assert len(Login.select()) == logins + API_PERSONS, error
assert len(Location.select()) == localizations + API_PERSONS, error
def test_save_login(self, downloader_obj, modifier_obj):
modifier_obj.execute_modifications()
dict_obj = modifier_obj._data[0]
save_obj = ApiDataSave(downloader_obj, 'results')
p = save_obj.save_person(dict_obj)
save_obj.save_login(dict_obj, p)
assert len(Login.select()) == 1, 'The Login object has not been saved'
login = Login.select()[0]
error = 'Invalid data has been written'
assert login.uuid == dict_obj['login']['uuid'], error
assert login.username == dict_obj['login']['username'], error
assert login.password == dict_obj['login']['password'], error
assert login.salt == dict_obj['login']['salt'], error
assert login.md5 == dict_obj['login']['md5'], error
assert login.sha1 == dict_obj['login']['sha1'], error
assert login.sha256 == dict_obj['login']['sha256'], error
assert login.registration_date == dict_obj['registered']['date'], error
assert login.years_since_registration == dict_obj['registered'][
'age'], error
assert login.person == p, error
