def handle(self, *args, **options):
num_editors = options['num'][0]
fake = Faker()
existing_users = User.objects.all()
# Superuser the only user, per twlight_vagrant README instructions.
if existing_users.count() == 0:
raise CommandError('No users present to Superuser. '
'Please login first.')
elif existing_users.count() > 1:
raise CommandError('More than one user present. '
'Please ensure that only one user is present.')
else:
user = existing_users[0]
user.is_superuser = True
user.is_staff = True
user.save()
for _ in range(num_editors):
user = UserFactory(email=fake.word() + "@example.com")
editor = EditorFactory(
user=user,
real_name=fake.name(),
country_of_residence=fake.country(),
occupation=fake.job(),
affiliation=fake.company(),
wp_editcount=random.randint(50, 2000),
wp_registered=fake.date_time_between(start_date="-10y",
end_date="now",
tzinfo=None),
contributions=fake.paragraph(nb_sentences=4))
# All users who aren't the superuser
all_users = User.objects.exclude(is_superuser=True)
# Flag wp_valid correctly if user is valid
for user in all_users:
date_valid = datetime.today().date() - timedelta(
days=182) >= user.editor.wp_registered
if user.editor.wp_editcount > 500 and date_valid:
user.editor.wp_valid = True
user.editor.save()
# Make 5 random users coordinators
coordinators = get_coordinators()
for user in random.sample(all_users, 5):
user.groups.add(coordinators)
user.save()
# Set 5 random non-coordinator users to have restricted data processing
restricted = get_restricted()
non_coordinators = all_users.exclude(groups__name='coordinators')
for user in random.sample(non_coordinators, 5):
user.groups.add(restricted)
user.save()
def test_authorization_authorizer_validation_staff(self):
"""
The authorizer can be a staff member but not a coordinator.
"""
user = UserFactory()
user2 = UserFactory()
user2.is_staff = True
user2.save()
auth = Authorization(user=user, authorizer=user2)
try:
auth.save()
except ValidationError:
self.fail("Authorization authorizer validation failed.")