def test_citizen_created(self, get_records, open): citizen = CitizenFactory() data = model_to_dict(citizen) citizen.delete() get_records.return_value = [data] self.assertFalse(Citizen.objects.exists()) stats = import_citizen_dump(input_filename=None) self.assertEqual(1, stats.records_read) new_cit = Citizen.objects.get() self.assertEqual(data, model_to_dict(new_cit))
def test_temp_citizen_created(self, get_records, open): citizen = CitizenFactory() data = model_to_dict(citizen) citizen.delete() get_records.return_value = [data] self.assertFalse(Citizen.objects.exists()) self.assertFalse(TempCitizen.objects.exists()) with patch('civil_registry.utils.delete_all'): with patch('civil_registry.utils.mirror_database') as mirror_database: mirror_database.return_value = MirrorStats() stats = import_citizen_dump(input_filename=None) # We mocked mirror_database, so nothing actually got copied to Citizen # but it should have "read" the data into TempCitizen, and we # mocked delete_all, so it should still be there. self.assertFalse(Citizen.objects.exists()) self.assertEqual(1, stats.records_read) mirror_database.assert_called() temp_cit = TempCitizen.objects.get() self.assertEqual(data, model_to_dict(temp_cit))
def test_temp_citizen_created(self, get_records, open): citizen = CitizenFactory() data = model_to_dict(citizen) citizen.delete() get_records.return_value = [data] self.assertFalse(Citizen.objects.exists()) self.assertFalse(TempCitizen.objects.exists()) with patch('civil_registry.utils.delete_all'): with patch( 'civil_registry.utils.mirror_database') as mirror_database: mirror_database.return_value = MirrorStats() stats = import_citizen_dump(input_filename=None) # We mocked mirror_database, so nothing actually got copied to Citizen # but it should have "read" the data into TempCitizen, and we # mocked delete_all, so it should still be there. self.assertFalse(Citizen.objects.exists()) self.assertEqual(1, stats.records_read) mirror_database.assert_called() temp_cit = TempCitizen.objects.get() self.assertEqual(data, model_to_dict(temp_cit))