def setUp(self):
create_test_data.create(num_registrations=0, num_registration_dates=0)
tasks.election_day()
tasks.registrations()
self.staff_user = UserFactory()
self.staff_user.is_staff = True
self.staff_user.save()
def setUp(self):
create_test_data.create(num_registrations=NUM_REGISTRATIONS)
self.unused_center = RegistrationCenterFactory()
self.unused_office = OfficeFactory()
tasks.election_day()
tasks.registrations()
self.staff_user = UserFactory()
self.staff_user.is_staff = True
self.staff_user.save()
def setUp(self):
self.election_dates = (now() - datetime.timedelta(days=20),
now() - datetime.timedelta(days=10), now())
create_test_data.create(num_registrations=10,
num_copy_centers=0,
election_dates=self.election_dates)
tasks.election_day()
# do it again, to ensure we don't try to add duplicate records to ElectionReport table
tasks.election_day()
def setUp(self):
create_test_data.create(num_registrations=NUM_REGISTRATIONS,
num_copy_centers=NUM_COPY_CENTERS,
num_no_reg_centers=NUM_NO_REG_CENTERS)
tasks.election_day()
tasks.registrations()
credentials = base64.b64encode(TEST_USERNAME + ':' + TEST_PASSWORD)
self.client.defaults['HTTP_AUTHORIZATION'] = 'Basic ' + credentials
views.REPORT_USER_DB[TEST_USERNAME] = TEST_PASSWORD
def setUp(self):
create_test_data.create(
num_registrations=NUM_REGISTRATIONS,
num_copy_centers=NUM_COPY_CENTERS,
num_no_reg_centers=NUM_NO_REG_CENTERS,
)
tasks.election_day()
tasks.registrations()
credentials = base64.b64encode(TEST_USERNAME + ":" + TEST_PASSWORD)
self.client.defaults["HTTP_AUTHORIZATION"] = "Basic " + credentials
views.REPORT_USER_DB[TEST_USERNAME] = TEST_PASSWORD
def setUp(self):
self.election_dates = (
now() - datetime.timedelta(days=20),
now() - datetime.timedelta(days=10),
now()
)
create_test_data.create(num_registrations=10,
num_copy_centers=0,
election_dates=self.election_dates)
tasks.election_day()
# do it again, to ensure we don't try to add duplicate records to ElectionReport table
tasks.election_day()
def handle(self, *args, **options):
if not options.pop(DELETE_EXISTING_DATA_OPT, False):
raise CommandError("%s is a required parameter" %
DELETE_EXISTING_DATA_ARG)
election_dates = []
if options['election_dates']:
tz = timezone(settings.TIME_ZONE)
for date_str in options['election_dates'].split(','):
election_dates.append(
tz.localize(datetime.strptime(date_str, '%Y-%m-%d')))
return create(center_without_office=options['center_without_office'],
num_registrations=options['num_registrations'],
num_registration_dates=options['num_registration_dates'],
num_daily_reports=options['num_daily_reports'],
num_registration_centers=options['num_centers'],
num_copy_centers=options['num_copy_centers'],
num_subconstituencies=options['num_subconstituencies'],
use_existing_infra=options['use_existing_infra'],
num_inactive_centers_per_election=options[
'num_inactive_per_election'],
num_no_reg_centers=options['num_no_reg_centers'],
election_dates=election_dates)
