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