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