def setUp(self):
super().setUp()
self.creds = [{'username': '******'.format(i), 'password': '******'} for i in range(3)]
self.users = [
SystemUser.objects.create_user(c['username'], '{0}@{0}.upc.edu'.format(c['username']), c['password']) for c
in self.creds]
# 0 es un estudiante, 1 es un profesor, 2 es una empresa, y 3 es una empresa premium
self.est = crear_estudiante(self.users[0])
self.prof = crear_profesor(self.users[1])
self.empr = crear_empresa(self.users[2])
from apps.cuentas.models import SystemUser
from apps.usuarios.factory import crear_estudiante, crear_profesor, crear_empresa, crear_administrador
# Obtenemos todos los usuarios y los agrupamos de 3 en 3 (para tener tres de cada tipo de perfiles, y el resto administradores)
usuarios = SystemUser.objects.all()
chunks = lambda l, n: [l[x: x + n] for x in range(0, len(l), n)]
grupo = chunks(usuarios, 3)
# CREAMOS ESTUDIANTES
for u in grupo[0]:
crear_estudiante(u)
# CREAMOS PROFESORES
for u in grupo[1]:
crear_profesor(u)
# CREAMOS EMPRESAS
for u in grupo[2]:
crear_empresa(u)
# CREAMOS ADMINISTRADORES
for u in grupo[3]:
crear_administrador(u)
