Example #1
0
 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)
Example #2
0
 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())
Example #3
0
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)
Example #4
0
 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))
Example #5
0
 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))
Example #6
0
 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))
Example #7
0
 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))
Example #8
0
 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))