def setUp(self):
self.bob_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'bob',
last_name = 'dylan')
self.bob_employee = Employee(user=self.bob_user, holidays_per_year=21)
self.bob_employee.save()
self.alice_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'alice',
last_name = 'wonderland')
self.alice_employee = Employee(user=self.alice_user, holidays_per_year=21)
self.alice_employee.save()
self.boss_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'boss',
last_name = 'boss')
self.boss_employee = Employee(user=self.boss_user, holidays_per_year=21)
self.boss_employee.save()
self.another_boss_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'another_boss',
last_name = 'another_boss')
self.another_boss_employee = Employee(user=self.another_boss_user, holidays_per_year=21)
self.another_boss_employee.save()
self.ceo_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'ceo',
last_name = 'spyhce')
self.ceo_employee = Employee(user=self.ceo_user, holidays_per_year=21)
self.ceo_employee.save()
self.project_alpha = Project(project_name = 'project_alpha')
self.project_alpha.save()
self.project_alpha.team_leaders.add(self.boss_employee, self.ceo_employee)
self.project_alpha.employees.add(self.boss_employee, self.bob_employee)
self.project_alpha.save()
self.project_beta = Project(project_name = 'project_beta')
self.project_beta.save()
self.project_beta.team_leaders.add(self.another_boss_employee, self.ceo_employee)
self.project_beta.employees.add(self.another_boss_employee, self.alice_employee)
self.project_beta.save()
self.factory = RequestFactory()
class SettingUpDB(TestCase):
"""
Only used to setup the DB.
Bob and Alice are employees of Spyhce.
Bob is working on project alpha whose the team leader are boss and ceo.
Alice is working on project beta whose the team leader are another_boss and ceo.
"""
def setUp(self):
self.bob_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'bob',
last_name = 'dylan')
self.bob_employee = Employee(user=self.bob_user, holidays_per_year=21)
self.bob_employee.save()
self.alice_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'alice',
last_name = 'wonderland')
self.alice_employee = Employee(user=self.alice_user, holidays_per_year=21)
self.alice_employee.save()
self.boss_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'boss',
last_name = 'boss')
self.boss_employee = Employee(user=self.boss_user, holidays_per_year=21)
self.boss_employee.save()
self.another_boss_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'another_boss',
last_name = 'another_boss')
self.another_boss_employee = Employee(user=self.another_boss_user, holidays_per_year=21)
self.another_boss_employee.save()
self.ceo_user = User.objects.create_user(username='******',
password='******',
email='*****@*****.**',
first_name = 'ceo',
last_name = 'spyhce')
self.ceo_employee = Employee(user=self.ceo_user, holidays_per_year=21)
self.ceo_employee.save()
self.project_alpha = Project(project_name = 'project_alpha')
self.project_alpha.save()
self.project_alpha.team_leaders.add(self.boss_employee, self.ceo_employee)
self.project_alpha.employees.add(self.boss_employee, self.bob_employee)
self.project_alpha.save()
self.project_beta = Project(project_name = 'project_beta')
self.project_beta.save()
self.project_beta.team_leaders.add(self.another_boss_employee, self.ceo_employee)
self.project_beta.employees.add(self.another_boss_employee, self.alice_employee)
self.project_beta.save()
self.factory = RequestFactory()
