class App(object):
def __init__(self):
self._accountant = Accountant()
def execute(self, change_contract):
staff = self._create_staff_fixtures()
if change_contract:
self._change_contract_for_hourly_to_fixed(staff)
print(self._accountant.calculate_staff_salary(staff))
def _create_staff_fixtures(self):
return Staff([
self._new_employee(150, 'hourly', [5, 5]),
self._new_employee(200, 'hourly', [10, 5, 5]),
self._new_employee(3000, 'fixed', [40, 40, 40, 38]),
self._new_employee(5000, 'fixed', [40, 40, 40, 43]),
self._new_employee(0, 'voluntary', [50, 50, 60, 50]),
])
def _new_employee(self, salary, contract, tracked_hours):
if contract == 'hourly':
cards = self._from_hours_to_cards(tracked_hours)
return Employee(Hourly(salary), HourlyCalculator(), cards)
if contract == 'voluntary':
cards = self._from_hours_to_cards(tracked_hours)
return Employee(Voluntary(), VoluntaryCalculator(), cards)
if contract == 'fixed':
cards = self._from_hours_to_cards(tracked_hours)
return Employee(Fixed(salary), FixedCalculator(), cards)
pass
def _from_hours_to_cards(self, tracked_hours):
cards = []
for hours in tracked_hours:
cards.append(Card(hours))
return cards
def _change_contract_for_hourly_to_fixed(self, staff):
employee = staff.employees()[1] # Getting an hourly
employee.change_contract(Fixed(2500), FixedCalculator())
employee.add_card(Card(10))
def test_calculates_staff_total_salary(self):
accountant = Accountant()
staff = Mock()
staff.obtain_salary = Mock(return_value=6000)
result = accountant.calculate_staff_salary(staff)
expected = 6000