def test_create_recruitment_req(self):
        c = Controller(self.test_db)

        req_id = c.create_recruitment_req(type='part time', years_exp=321, title='title', description='reason1',
                                          req_dpt='service')
        req_data = c.get_recruitment_req('id', req_id)[0]

        self.assertEqual('part time', req_data['type'])
        self.assertEqual(321, req_data['years_exp'])
        self.assertEqual('title', req_data['title'])
        self.assertEqual('reason1', req_data['description'])
        self.assertEqual('service', req_data['req_dpt'])
    def test_update_recruitment_req(self):
        c = Controller(self.test_db)

        req_id = c.create_recruitment_req(type='part time', years_exp=321, title='title', description='reason1',
                                          req_dpt='service')

        updated_rec = {'id': req_id, 'type': 'full time', 'years_exp': 543, 'title': 'title2',
                       'description': 'description2', 'req_dpt': 'production'}
        c.update_recruitment_req(**updated_rec)

        req_data = c.get_recruitment_req('id', req_id)[0]

        self.assertEqual('full time', req_data['type'])
        self.assertEqual(543, req_data['years_exp'])
        self.assertEqual('title2', req_data['title'])
        self.assertEqual('description2', req_data['description'])
        self.assertEqual('production', req_data['req_dpt'])
class Mediator:
    base = None
    login_form = None

    roles = {'0': 'team_member',
             '1': 'customer_service',
             '2': 'senior_customer_service_officer',
             '3': 'human_resources',
             '4': 'administration',
             '5': 'financial',
             '6': 'production',
             '7': 'service',
             '8': 'vice_president'}

    def __init__(self, db_name='db.json'):
        self.c = Controller(db_name)

    def check_credentials(self, username, password):
        return self.c.login(username, password)

    def login(self, username):
        # Get that employee id, clients cant login...
        user_id = self.c.get_user_id(username)
        empl_data = self.get_employee('id', user_id, False)[0]
        name = empl_data['name']
        pos = empl_data['position']

        self.base = Base(name, self.roles[pos].title().replace('_', ' '), username)
        self.m = ManagerTabs(self.roles[pos], user_id, name)
        self.base.set_central_widget(self.m)

    def logout(self):
        self.login_form.show()

    def create_client(self, name, age, address, mail, phone):
        return self.c.create_client(name, name=name, age=age, address=address, mail=mail, phone=phone, events=[])

    def create_client_req(self, client_id, event_type, description, from_date, to_date, exp_no,
                          planned_budget, decorations, filming, poster,
                          food, music, computer, other, approved):
        return self.c.create_client_req(client_id=client_id, event_type=event_type, description=description,
                                        from_date=from_date,
                                        to_date=to_date, exp_no=exp_no, planned_budget=planned_budget,
                                        decorations=decorations,
                                        filming=filming, poster=poster, food=food, music=music, computer=computer,
                                        other=other, approved=approved)

    def create_employee(self, name, age, address, mail, position):
        return self.c.create_employee(name, name=name, age=age, address=address, mail=mail,
                                      position=position)

    def create_recruitment_req(self, type, years_exp, title, description, dpt_req):
        return self.c.create_recruitment_req(type=type, years_exp=years_exp, title=title, description=description,
                                             req_dpt=dpt_req)

    def update_recruitment_req(self, recruitment_id, type, years_exp, title, description, dpt_req):
        return self.c.update_recruitment_req(id=recruitment_id, type=type, years_exp=years_exp, title=title,
                                             description=description,
                                             req_dpt=dpt_req)

    def create_task(self, sub_team, event_id, description, staff_id, priority):
        return self.c.create_task(sub_team=sub_team, event_id=event_id, description=description, staff_id=staff_id,
                                  priority=priority)

    def update_task(self, task_id, sub_team, event_id, description, staff_id, priority):
        self.c.update_task(id=task_id, sub_team=sub_team, event_id=event_id, description=description,
                           staff_id=staff_id, priority=priority)

    def create_financial_req(self, event_id, req_amount, reason, req_dpt):
        return self.c.create_financial_req(event_id=event_id, req_amount=req_amount, reason=reason, req_dpt=req_dpt)

    def update_financial_req(self, fin_req_id, event_id, req_amount, reason, req_dpt):
        self.c.update_financial_req(id=fin_req_id, event_id=event_id, req_amount=req_amount, reason=reason,
                                    req_dpt=req_dpt)

    def get_financial_req(self, col_name='', criteria='', all_data=True):
        return self._get_data('financial_req', col_name, criteria, all_data)

    def get_recruitment_req(self, col_name='', criteria='', all_data=True):
        return self._get_data('recruitment_req', col_name, criteria, all_data)

    def get_client(self, col_name='', criteria='', all_data=True):
        return self._get_data('client', col_name, criteria, all_data)

    def get_employee(self, col_name='', criteria='', all_data=True):
        return self._get_data('employee', col_name, criteria, all_data)

    def get_task(self, col_name='', criteria='', all_data=True):
        return self._get_data('task', col_name, criteria, all_data)

    def get_event(self, col_name='', criteria='', all_data=True):
        return self._get_data('event', col_name, criteria, all_data)

    def update_event(self, ev_id, client_id, event_type, description, from_date, to_date, exp_no,
                     planned_budget, decorations, filming, poster,
                     food, music, computer, other, approved):
        return self.c.update_event(id=ev_id, client_id=client_id, event_type=event_type, description=description,
                                   from_date=from_date,
                                   to_date=to_date, exp_no=exp_no, planned_budget=planned_budget,
                                   decorations=decorations,
                                   filming=filming, poster=poster, food=food, music=music, computer=computer,
                                   other=other, approved=approved)

    def _get_data(self, table, col_name, criteria, all_data):
        return getattr(self.c, "get_%s" % table)(col_name, criteria, all_data)