def test_create_report(self, input, mock_show_complaint, mock_conn):
input.side_effect = [1, 'cause', 'details', 10, 10]
test_class = SupervisorService.Supervisor()
result = test_class.create_report()
assert result is True
def test_show_reports_empty(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchall.return_value = []
test_class = SupervisorService.Supervisor()
result = test_class.show_reports()
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 = SupervisorService.Supervisor()
result = test_class.supervisor_login()
assert result is False
def test_show_complaint(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchall.return_value = [
(1, 'gas leakout', 'test comments')
]
test_class = SupervisorService.Supervisor()
result = test_class.show_complaint()
assert result is True
def test_login_password_match(self, inputs, mock_conn, getpass):
inputs.return_value = '*****@*****.**'
getpass.return_value = 'pass'
mock_conn.return_value.cursor.return_value.fetchone.return_value = (
'*****@*****.**', 'pass', 1, 'TeamName')
test_class = SupervisorService.Supervisor()
result = test_class.supervisor_login()
assert result is True
def test_show_reports(self, mock_conn):
mock_conn.return_value.cursor.return_value.fetchall.return_value = [
(1, 1, 'soul', 'short circuit', 'test details', 10, 5, 'approved',
'2020-05-14 12:41:24')
]
test_class = SupervisorService.Supervisor()
result = test_class.show_reports()
assert result is True
def test_worker_tasks(self, inputs, mock_show_complaint,
mock_create_report, mock_show_reports):
inputs.side_effect = ['1', '2', '3', '6', '4']
test_class = SupervisorService.Supervisor()
test_class.supervisor_tasks()
mock_show_complaint.assert_called_once_with()
mock_create_report.assert_called_once_with()
mock_show_reports.assert_called_once_with()
def test_create_report_failure(self):
test_class = SupervisorService.Supervisor()
result = test_class.create_report()
assert result is False
num = 0
conn = create_connection()
while ch != '4':
print("MAIN MENU")
print("1. Admin Login")
print("2. Supervisor Login")
print("3. Worker Login")
print("4. Exit")
ch = input("Select Your Option ")
if ch == '1':
admin = AdminService.Admin()
if admin.admin_login():
admin.admin_tasks()
elif ch == '2':
supervisor = SupervisorService.Supervisor()
if supervisor.supervisor_login():
supervisor.supervisor_tasks()
elif ch == '3':
worker = WorkerService.Worker()
if worker.worker_login():
worker.worker_tasks()
elif ch == '4':
print("Thank You.")
else:
print("Invalid choice")