def test_list_all_users_may_list_them(self):
UserFactory.create_batch(5)
url = reverse('user-list')
response = self.client.get(url)
expected_users_number = User.objects.count()
self.assertEqual(status.HTTP_200_OK, response.status_code)
self.assertEqual(expected_users_number, len(response.data['results']))
def handle(self, *args, **options):
"""Automatically called when the sampledata command is given."""
if settings.DEPLOYMENT_TYPE == 'prod' and not settings.DEBUG:
raise management.base.CommandError(
'This command can only be executed in DEBUG mode on non-production website.'
)
# Clear all data
print(LOG_HEADER.format('Wipe database'))
management.call_command('flush', interactive=False)
print('Database wiped.')
management.call_command('load_user_types')
print(LOG_HEADER.format('Create sample users'))
User = get_user_model()
# Create admin account
admin = User.objects.create_superuser(
'admin',
'*****@*****.**',
password=settings.SAMPLE_DATA_ADMIN_PASSWORD,
first_name='Admin',
last_name='Account',
user_type=UserType.objects.get(slug='teacher'))
EmailAddress.objects.create(user=admin,
email=admin.email,
primary=True,
verified=True)
print('Admin created.')
# Create user account
user = User.objects.create_user(
'user',
'*****@*****.**',
password=settings.SAMPLE_DATA_USER_PASSWORD,
first_name='Alex',
last_name='Doe',
user_type=UserType.objects.get(slug='student'))
EmailAddress.objects.create(user=user,
email=user.email,
primary=True,
verified=True)
UserFactory.create_batch(size=100)
print('Users created.')
# Codewof
management.call_command('load_questions')
print('Programming questions loaded.')
# Research
StudyFactory.create_batch(size=5)
StudyGroupFactory.create_batch(size=15)
print('Research studies loaded.')
def handle(self, *args, **options):
"""Automatically called when the sampledata command is given."""
if settings.PRODUCTION_ENVIRONMENT:
raise management.base.CommandError(
'This command can only be executed on a non-production website.'
)
skip = options['skip_backdate']
# Clear all data
print(LOG_HEADER.format('Wipe database'))
management.call_command('flush', interactive=False)
print('Database wiped.')
# Update data
management.call_command('update_data')
print(LOG_HEADER.format('Create sample users'))
# Create admin account
management.call_command('create_admin')
# Create user account
User = get_user_model() # noqa N806
user = User.objects.create_user(
'user',
'*****@*****.**',
password=settings.SAMPLE_DATA_USER_PASSWORD,
first_name='Alex',
last_name='Doe',
user_type=UserType.objects.get(slug='student'))
EmailAddress.objects.create(user=user,
email=user.email,
primary=True,
verified=True)
UserFactory.create_batch(size=100)
print('Users created.\n')
# Attempts
AttemptFactory.create_batch(size=50)
print('Attempts loaded.\n')
# Award points and achievements
if not skip:
management.call_command('backdate_points_and_achievements')
else:
print('Ignoring backdate step as requested.\n')
