def test_manage_supervisor_reports_resolved(self, mock_choose_employee, mock_show_table, mock_menu, mock_raw_table, mock_db,
mock_print, mock_input):
mock_menu().draw_menu.return_value = 'RESOLVED'
admin = Admin()
admin.manage_supervisor_reports()
mock_menu().draw_menu.assert_called_once()
mock_raw_table.assert_called_once()
def test_manage_supervisor_reports_wip_no_accident(self, mock_choose_employee, mock_show_table, mock_menu, mock_raw_table, mock_db,
mock_print, mock_input):
mock_menu().draw_menu.return_value = 'WIP'
mock_show_table.return_value = None
admin = Admin()
admin.manage_supervisor_reports()
mock_menu().draw_menu.assert_called_once()
mock_db().show_status_accident_report.assert_called_once()
def test_manage_supervisor_reports_wip(self, mock_choose_employee, mock_show_table, mock_menu, mock_raw_table, mock_db,
mock_print, mock_input):
mock_menu().draw_menu.return_value = 'WIP'
mock_choose_employee.return_value = ['1', '2']
admin = Admin()
admin.manage_supervisor_reports()
mock_menu().draw_menu.assert_called_once()
mock_db().commit_data.assert_called_once()
mock_db().insert_finalize_report.assert_called_once()
mock_db().close_accident_cases.assert_called_once()