def test_copy_goal_true(self): employee = EmployeeFactory.create() goal = IndividualGoalFactory.create() goal_pk, created = goal.copy(employee) self.assertTrue(created) goal_db = IndividualGoal.objects.get(pk=goal_pk) self.assertIsNotNone(goal_db)
def test_assigned_to_project(self): employees = EmployeeFactory.create_batch(10) member = ProjectMemberFactory.create() for employee in employees[:5]: ProjectMemberFactory.create(project=member.project, employee=employee) assigned = ProjectMember.objects.assigned_to_project(member.project) self.assertEqual(6, len(assigned))
def setUp(self): self.employee = EmployeeFactory( first_name='Александр', middle_name='Сергеевич', last_name='Зубов', email='*****@*****.**', user_position=None, date_of_birth=datetime.datetime.today(), phone_number='9633609225', extra_phone_number='9633609225', other_contacts='Telegram - trelop', timezone='2018-08-29T20:43:18.869351+03:00', is_active=True, is_staff=True, is_superuser=False, status='1', login_skype='trelop', confirmed_email=True, group=None, password='******') self.task = TaskFactory( name='Task', creation_date='2018-08-29T20:43:18.869351+03:00', expiration_date='2018-08-29T20:43:18.869351+03:00', status='1', priority='1', task_description='Description', task_creator=self.employee, task_executor=self.employee)
def setUp(self): self.position = PositionFactory() self.group = EmployeeGroupFactory( name='Name', description='This is description for this group', creation_date=datetime.datetime.now(tz=TZ), creator=None ) self.employee = EmployeeFactory( first_name='Александр', middle_name='Сергеевич', last_name='Зубов', email='*****@*****.**', user_position=self.position, date_of_birth=datetime.datetime.today(), phone_number='9633609225', extra_phone_number='9633609225', other_contacts='Telegram - trelop', timezone='2018-08-29T20:43:18.869351+03:00', is_active=True, is_staff=True, is_superuser=False, status='1', login_skype='trelop', confirmed_email=True, group=self.group ) self.company = ClientFactory( name='РАО «Харитонова»', other_names='РАО Аристотель', zip_code='184682', email='*****@*****.**', address='Россия, Ивановская область, проспект Ленина, дом 2', region='Ивановская область', city='Иваново', website='https://google.com', timezone='2018-08-29T20:43:18.869351+03:00', additional_info='Компания занимается добычей угля', note='Клиент также работает в странах СНГ', employee_manager=self.employee, client_status=None, employee_creator=None, creation_date=datetime.date.today(), date_last_editing=datetime.date.today(), is_active=True, client_group=[] ) self.task = TaskFactory( name='Task', creation_date='2018-08-29T20:43:18.869351+03:00', expiration_date='2018-08-29T20:43:18.869351+03:00', status='1', priority='1', task_description='Description', task_creator=self.employee, task_executor=self.employee )
def test_assign_goals(self): out = StringIO() company_group = CompanyGroupFactory.create(name='TINO-NS') employees = EmployeeFactory.create_batch(10) goal = IndividualGoalFactory.create() self.assertEqual(1, IndividualGoal.objects.count()) for employee in employees: company_group.assign(employee, date.today()) call_command('create_goals', assign_goal=True, goal_pk=goal.pk, companygroup_name=company_group.slug, stdout=out) self.assertEqual(11, IndividualGoal.objects.count()) self.assertTrue(out.getvalue().startswith('1 Copied '))