def test_show_all_routes_empty(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchall.return_value = []
test_class = AdminService.Admin()
result = test_class.show_all_routes()
assert result is False
def test_create_employee(self, inputs, mock_conn):
inputs.side_effect = ['test', '*****@*****.**']
test_class = AdminService.Admin()
result = test_class.create_employee()
assert result is True
def test_create_employee_invalid_email(self, inputs, mock_conn):
inputs.side_effect = ['test', 'testgmail']
test_class = AdminService.Admin()
result = test_class.create_employee()
assert result is False
def test_create_cab_unknown_error(self, inputs):
inputs.side_effect = ['cab123', 'b']
test_class = AdminService.Admin()
result = test_class.create_cab()
assert result is False
def test_show_total_bookings_week_empty(self, inputs, mock_conn):
inputs.side_effect = [1]
mock_conn.return_value.cursor.return_value.fetchone.return_value = []
test_class = AdminService.Admin()
result = test_class.show_total_bookings_week()
assert result is False
def test_login_invalid_user(self, inputs, mock_conn):
inputs.return_value = '*****@*****.**'
mock_conn.return_value.cursor.return_value.fetchone.return_value = None
test_class = AdminService.Admin()
result = test_class.admin_login()
assert result is False
def test_show_emp_bookings_invalid_id(self, inputs, mock_conn):
inputs.side_effect = [1]
mock_conn.return_value.cursor.return_value.fetchall.return_value = []
test_class = AdminService.Admin()
result = test_class.show_emp_bookings()
assert result is False
def test_create_route(self, inputs, mock_conn):
inputs.side_effect = ['cab123']
test_class = AdminService.Admin()
result = test_class.create_route()
assert result is True
def test_show_total_bookings_month_unknown_error(self, inputs, mock_conn):
inputs.side_effect = ['b']
mock_conn.return_value.cursor.return_value.fetchone.return_value = []
test_class = AdminService.Admin()
result = test_class.show_total_bookings_month()
assert result is False
def test_get_route_by_id_invalid_id(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchone.return_value = []
test_class = AdminService.Admin()
result = test_class.get_route_by_id(1)
assert result is False
def test_update_cab(self, inputs, mock_conn, mock_get_cab_by_id):
inputs.side_effect = ['cab123', 1]
mock_get_cab_by_id.return_value = ['cab123', 2]
test_class = AdminService.Admin()
result = test_class.update_cab()
assert result is True
def test_get_cab_by_id(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchone.return_value = [
'cab123', 2
]
test_class = AdminService.Admin()
result = test_class.get_cab_by_id('test')
assert result == ['cab123', 2]
def test_show_all_routes(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchall.return_value = [
(1, 'r123')
]
test_class = AdminService.Admin()
result = test_class.show_all_routes()
assert result is True
def test_show_allemployees(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchall.return_value = [
(1, 'test', '*****@*****.**')
]
test_class = AdminService.Admin()
result = test_class.show_all_employees()
assert result is True
def test_delete_employee_unknown_error(self, inputs, mock_conn,
mock_show_allemployees):
inputs.side_effect = ['a']
test_class = AdminService.Admin()
result = test_class.delete_employee()
mock_show_allemployees.assert_called_once_with()
assert result is False
def test_login_password_mismatch(self, inputs, mock_conn, getpass):
inputs.return_value = '*****@*****.**'
getpass.return_value = 'password'
mock_conn.return_value.cursor.return_value.fetchone.return_value = (
'*****@*****.**', 'pass', 1)
test_class = AdminService.Admin()
result = test_class.admin_login()
assert result is False
def test_update_employee(self, inputs, mock_conn, mock_get_employee_by_id):
inputs.side_effect = [1, 'test', '*****@*****.**']
mock_get_employee_by_id.return_value = ['Name', '*****@*****.**']
test_class = AdminService.Admin()
result = test_class.update_employee()
assert result is True
def test_get_employee_by_id(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchone.return_value = [
'Name', '*****@*****.**'
]
test_class = AdminService.Admin()
result = test_class.get_employee_by_id(1)
assert result == ['Name', '*****@*****.**']
def test_delete_cab_invalid_id(self, inputs, mock_conn,
mock_get_route_by_id, mock_show_all_routes):
inputs.side_effect = ['1']
mock_get_route_by_id.return_value = []
test_class = AdminService.Admin()
result = test_class.delete_route()
mock_show_all_routes.assert_called_once_with()
assert result is False
def test_delete_cab(self, inputs, mock_conn, mock_get_cab_by_id,
mock_show_all_cabs):
inputs.side_effect = ['1']
mock_get_cab_by_id.return_value = ['cab123', 2]
test_class = AdminService.Admin()
result = test_class.delete_cab()
mock_show_all_cabs.assert_called_once_with()
assert result is True
def test_show_total_bookings_month(self, inputs, mock_conn):
inputs.side_effect = [1]
mock_conn.return_value.cursor.return_value.fetchone.return_value = [
'01', 2
]
test_class = AdminService.Admin()
result = test_class.show_total_bookings_month()
assert result is True
def test_admin_cab_mgmnt(self, inputs, mock_create_cab, mock_update_cab,
mock_delete_cab, mock_show_all_cabs):
inputs.side_effect = ['1', '2', '3', '4', '6', '5']
test_class = AdminService.Admin()
test_class.admin_cab_management()
mock_create_cab.assert_called_once_with()
mock_update_cab.assert_called_once_with()
mock_delete_cab.assert_called_once_with()
mock_show_all_cabs.assert_called_once_with()
def test_show_emp_bookings(self, inputs, mock_conn):
inputs.side_effect = [1]
mock_conn.return_value.cursor.return_value.fetchall.return_value = [
(1, '2020-01-01', '12:00', 'cab123', 's', 'd')
]
test_class = AdminService.Admin()
result = test_class.show_emp_bookings()
assert result is True
def test_delete_employee(self, inputs, mock_conn, mock_get_employee_by_id,
mock_show_allemployees):
inputs.side_effect = [1]
mock_get_employee_by_id.return_value = ['Name', '*****@*****.**']
test_class = AdminService.Admin()
result = test_class.delete_employee()
mock_show_allemployees.assert_called_once_with()
assert result is True
def test_admin_tasks(self, inputs, mock_admin_employee_mgmnt,
mock_admin_cab_mgmnt, mock_admin_route_management,
mock_admin_booking_management):
inputs.side_effect = ['1', '2', '3', '4', '6', '5']
test_class = AdminService.Admin()
test_class.admin_tasks()
mock_admin_employee_mgmnt.assert_called_once_with()
mock_admin_cab_mgmnt.assert_called_once_with()
mock_admin_route_management.assert_called_once_with()
mock_admin_booking_management.assert_called_once_with()
def test_admin_booking_management(self, inputs, mock_show_emp_bookings,
mock_show_total_bookings_day,
mock_show_total_bookings_week,
show_total_bookings_month):
inputs.side_effect = ['1', '2', '3', '4', '6', '5']
test_class = AdminService.Admin()
test_class.admin_booking_management()
mock_show_emp_bookings.assert_called_once_with()
mock_show_total_bookings_day.assert_called_once_with()
mock_show_total_bookings_week.assert_called_once_with()
show_total_bookings_month.assert_called_once_with()
def test_admin_route_management(self, inputs, mock_create_route,
mock_delete_route, mock_show_all_routes,
mock_create_cab_route,
mock_update_cab_route, mock_get_all_routes,
mock_delete_cab_route):
inputs.side_effect = ['1', '2', '3', '4', '5', '6', '7', '9', '8']
test_class = AdminService.Admin()
test_class.admin_route_management()
mock_create_route.assert_called_once_with()
mock_delete_route.assert_called_once_with()
mock_show_all_routes.assert_called_once_with()
mock_create_cab_route.assert_called_once_with()
mock_update_cab_route.assert_called_once_with()
mock_get_all_routes.assert_called_once_with()
mock_delete_cab_route.assert_called_once_with()
from src import EmployeeService, AdminService
if __name__ == '__main__':
option = ''
while option != '3':
print("MAIN MENU")
print("1. Admin Login")
print("2. Employee Login")
print("3. Exit")
option = input("Select Your Option ")
if option == '1':
admin = AdminService.Admin()
if admin.admin_login():
admin.admin_tasks()
elif option == '2':
employee = EmployeeService.Employee()
if employee.employee_login():
employee.employee_tasks()
elif option == '3':
print("Thank You.")
else:
print("Invalid choice.")
def test_update_cab_unknown_error(self):
test_class = AdminService.Admin()
result = test_class.update_cab()
assert result is False
def test_delete_route_unknown_error(self):
test_class = AdminService.Admin()
result = test_class.delete_route()
assert result is False
