def test_login(self): a = Authentication(self.test_db) c = Controller(self.test_db) a.create_user('employee', 'user1') self.assertTrue(c.login('user1', '12345'))
def gen_clients(): a = Authentication() a.create_user('client', 'c1', name='Ima Client', age='16', address='Clientville 2', mail='*****@*****.**', phone='01234567', events=[]) a.create_user('client', 'c2', name='YAC Client', age='106', address='Nexttohighway 42', mail='*****@*****.**', phone='76543210', events=[])
def __init__(self, db_name='db.json'): self.db = Database(name=db_name) self.a = Authentication(db_name=db_name)
class Controller(): def __init__(self, db_name='db.json'): self.db = Database(name=db_name) self.a = Authentication(db_name=db_name) def login(self, username, password): return self.a.login(username, password) def create_client(self, username, **kwargs): return self.a.create_user('client', username, **kwargs) def create_client_req(self, **kwargs): event_id = self.db.new_event(**kwargs) clients_events = self.db.get_client('id',kwargs['client_id'], all_data=False)[0]['events'] clients_events.append(event_id) self.db.update_client_events(kwargs['client_id'], clients_events) return event_id def create_employee(self, username, **kwargs): return self.a.create_user('employee', username, **kwargs) def update_event(self, **kwargs): self.db.update_event(kwargs) def create_task(self, **kwargs): return self.db.new_task(**kwargs) def update_task(self, **kwargs): self.db.update_task(kwargs) def create_financial_req(self, **kwargs): return self.db.new_financial_req(**kwargs) def create_recruitment_req(self, **kwargs): return self.db.new_recruitment_req(**kwargs) def update_recruitment_req(self, **kwargs): return self.db.update_recruitment_req(kwargs) def update_financial_req(self, **kwargs): self.db.update_financial_req(kwargs) def get_user_id(self, username): return self.db.get_login_data(username)['user_id'] def update_client_events(self, cl_id, events): self.db.update_client_events(cl_id, events) 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 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_data(self, table, col_name, criteria, all_data): return getattr(self.db, "get_%s" % table)(col_name, criteria, all_data)
def __init__(self, username, password): self.token = Authentication(username, password).logon() self.headers = { 'Content-Type': 'application/json', 'Authorization': 'Bearer {}'.format(self.token) }
def gen_employees(): a = Authentication() """ name=name, age=age, address=address, mail=mail, position=position """ # Vice presidnent a.create_user('employee', 'e8', name='Don Hartley', age='55', address='Beech Street 365', mail='*****@*****.**', position='8') # Service a.create_user('employee', 'e7', name='Shaw Stonebridge', age='52', address='Oak Street 582', mail='*****@*****.**', position='7') # Production a.create_user('employee', 'e6', name='Casey Netley', age='49', address='11th Street 749', mail='*****@*****.**', position='6') # Financial a.create_user('employee', 'e5', name='Raven Brown', age='50', address='Washington avenue 113', mail='*****@*****.**', position='5') # Administration a.create_user('employee', 'e4', name='Eddy Brady', age='35', address='East Street 572', mail='*****@*****.**', position='4') # Human Resources a.create_user('employee', 'e3', name='Neddy Hayley', age='41', address='Crescentt Street 271', mail='*****@*****.**' ,position='3') # Senior Customer service oficcer a.create_user('employee', 'e2', name='Elaina Kelsey', age='39', address='Hamilton Street 29', mail='*****@*****.**' ,position='2') # Customer service a.create_user('employee', 'e1', name='Kayla Bunce', age='28', address='School Street 455', mail='*****@*****.**', position='1') # Staff a.create_user('employee', 'e0', name='Peter Pana', age='22', address='Bro Street 42', mail='*****@*****.**', position='0')
def __init__(self, username, password): self.token = Authentication(username, password).logon() self.headers = {'Authorization': 'Bearer {}'.format(self.token)}
def test_wrong_password_login(self): a = Authentication(self.test_db) a.create_user('employee', 'user6') self.assertFalse(a.login('user6', '1'))
def test_wrong_username_login(self): a = Authentication(self.test_db) a.create_user('employee', 'user5') self.assertFalse(a.login('userWrong', '12345'))
def test_success_login(self): a = Authentication(self.test_db) a.create_user('employee', 'user4') self.assertTrue(a.login('user4', '12345'))
def test_no_such_kind(self): a = Authentication(self.test_db) with self.assertRaises(Exception): a.create_user('invalid', 'user3')
def test_username_exists(self): a = Authentication(self.test_db) a.create_user('employee', 'user2') with self.assertRaises(KeyError): a.create_user('employee', 'user2')
def test_create_user(self): a = Authentication(self.test_db) self.assertTrue(a.create_user('client', 'user1'))