def test_name_format(self): # We format names as First name, father name, etc. citizen = Citizen(first_name="FIRST", father_name="FATHER", grandfather_name="GRANDFATHER", family_name="FAMILY", mother_name="MOTHER") name = unicode(citizen) self.assertEqual(u"FIRST FATHER GRANDFATHER FAMILY", name)
def test_upload_file_might_have_to_lookup_citizen(self): # Upload file can have a citizen we don't have a Citizen record for yet # (e.g. if we're blocking a citizen who hasn't tried to register) data = self.data data['how_to_select'] = str(Changeset.SELECT_UPLOADED_NIDS) nid1 = "199999999999" filetext = "{nid1}\n".format(nid1=nid1) upload_file = SimpleUploadedFile('my_filename', filetext) with patch('changesets.forms.get_citizen_by_national_id') as mock_get_citizen: # Make a Citizen but don't save it in the database, so the form validation # won't initially find it citizen = Citizen(national_id=nid1) mock_get_citizen.return_value = citizen form = ChangesetForm(data=data, files={'upload_file': upload_file}, request=MagicMock()) self.assertTrue(form.is_valid())
def prepare_loadtest(request): # pragma: no cover # Only run if this server is meant to run load tests if not settings.LOAD_TEST: return HttpResponseNotFound() # Delete Registrations, Citizens and Centers Citizen.objects.all().delete() Registration.objects.unfiltered().delete() RegistrationCenter.objects.all().delete() # create valid persons first_national_id = 100000000000 birth_date = datetime.date(1913, 2, 3) citizens = [] default_count = 20000 count = request.GET.get('n', default_count) try: count = int(count) except ValueError: count = default_count if count < 0: count = default_count for i in range(count): citizens.append(Citizen(national_id=first_national_id + i, birth_date=birth_date)) Citizen.objects.bulk_create(citizens) # create 1000 valid centers # we don't need more than 1000, b/c they can be reused over and over first_center_id = 10000 centers = [] for i in range(1000): centers.append(RegistrationCenter(center_id=first_center_id + i)) RegistrationCenter.objects.bulk_create(centers) return HttpResponse("Load test data created (n=%s)." % count)
def test_is_eligible_tomorrow(self): # Will be eligible tomorrow TODAY = datetime.datetime(2000, 6, 1) citizen = Citizen(birth_date=datetime.date(1982, 6, 2)) self.assertFalse(citizen.is_eligible(as_of=TODAY))
def test_is_eligible_yesterday(self): # Became eligible yesterday TODAY = datetime.datetime(2000, 6, 2) citizen = Citizen(birth_date=datetime.date(1982, 6, 1)) self.assertTrue(citizen.is_eligible(as_of=TODAY))
def test_is_eligible_today(self): # 18th birthday "today" TODAY = datetime.datetime(2000, 6, 1) citizen = Citizen(birth_date=datetime.date(1982, 6, 1)) self.assertTrue(citizen.is_eligible(as_of=TODAY))
def test_is_eligible_plenty_old(self): TODAY = datetime.datetime(2000, 1, 1) citizen = Citizen(birth_date=datetime.date(1970, 1, 1)) self.assertTrue(citizen.is_eligible(as_of=TODAY))
def test_is_eligible_way_too_young(self): TODAY = datetime.datetime(2000, 1, 1) citizen = Citizen(birth_date=datetime.date(1990, 1, 1)) self.assertFalse(citizen.is_eligible(as_of=TODAY))