def test_one_create_and_print_employees(self):
user_input = [
3, self.employees[0].name, self.employees[0].age,
self.employees[0].salary, self.employees[0].employment_year, 1, 5
]
with redirect_stdout(self.response):
with patch('builtins.input', side_effect=user_input):
hr_pro.main()
self.assertTrue(
self.employees[0].name in self.response.getvalue())
def test_multiple_create_and_print_managers(self):
user_input = []
for manager in self.managers:
user_input.extend([
4, manager.name, manager.age, manager.salary,
manager.employment_year, manager.bonus_percentage
])
user_input.extend([2, 5])
with redirect_stdout(self.response):
with patch('builtins.input', side_effect=user_input):
hr_pro.main()
for manager in self.managers:
self.assertTrue(manager.name in self.response.getvalue())
def test_multiple_create_and_print_employees(self):
user_input = []
for employee in self.employees:
user_input.extend([
3, employee.name, employee.age, employee.salary,
employee.employment_year
])
user_input.extend([1, 5])
with redirect_stdout(self.response):
with patch('builtins.input', side_effect=user_input):
hr_pro.main()
for employee in self.employees:
self.assertTrue(employee.name in self.response.getvalue())
def testـonly_managers_printing(self):
user_input = []
for manager in self.managers:
user_input.extend([
4, manager.name, manager.age, manager.salary,
manager.employment_year, manager.bonus_percentage
])
for employee in self.employees:
user_input.extend([
3, employee.name, employee.age, employee.salary,
employee.employment_year
])
user_input.extend([2, 5])
with redirect_stdout(self.response):
with patch('builtins.input', side_effect=user_input):
hr_pro.main()
for manager in self.managers:
self.assertTrue(manager.name in self.response.getvalue())
for employee in self.employees:
self.assertFalse(employee.name in self.response.getvalue())