def test_department_with_single_employee_pays_correctly(self):
department = Department()
department.add_employee(Developer())
amount_paid = department.pay()
self.assertEqual(1000, amount_paid)
def test_department_with_all_kinds_of_employees(self):
department = Department()
department.add_employee(Developer())
department.add_employee(QA())
manager = Manager()
manager.add_employee(QA())
manager.add_employee(Developer())
manager.add_employee(Manager())
manager2 = Manager()
manager2.add_employee(Developer())
manager.add_employee(manager2)
department.add_employee(manager)
amount_paid = department.pay()
self.assertEqual(4900, amount_paid)
def test_empty_department_pays_nothing(self):
department = Department()
amount_paid = department.pay()
self.assertEqual(0, amount_paid)
