def show_accident_complain(self):
"""
this method is used to show supervisor accident complain.
"""
cursor = self.conn.show_supervisor_accidents(self.supervisor_id)
admin = Admin()
temp_dict = dict()
print('choose accident to send report')
temp_dict['accident_id'] = admin.show_table_menu(cursor)
if temp_dict['accident_id'] is None:
return
cursor = self.conn.show_members()
print('choose eye witness')
temp_dict['eye_witness'] = admin.show_table_menu(cursor)
print('choose victims')
victims = admin.choose_multiple_employee()
print('choose culprit')
temp_dict['victims'] = ''
temp_dict['culprit'] = ''
culprit = admin.choose_multiple_employee()
if culprit != [] and culprit != None:
temp_dict['culprit'] = ','.join(culprit)
if victims != [] and victims != None:
temp_dict['victims'] = ','.join(victims)
temp_dict['submitted_by'] = self.supervisor_id
temp_dict['reason'] = input('enter reason ')
temp_dict['timing'] = input('enter accident timing ')
temp_dict['submission_time'] = d.datetime.now()
temp_dict['location'] = input('enter location of accident ')
temp_dict['action_to_resolve'] = input('enter action to resolve ' )
try:
self.conn.register_supervisor_report(temp_dict)
self.conn.commit_data()
print('report submitted successfully.')
except Exception as e:
print('some error occured')
print(e)
self.conn.rollback_data()
def test_choose_multiple_employee_exception(self, mock_input, mock_raw_table, mock_db):
mock_db().show_members().fetchall.return_value = [[1, 2, 3]]
mock_input.return_value = 'asd,1,2'
admin = Admin()
result = admin.choose_multiple_employee()
self.assertEqual(result, None)
def test_choose_multiple_employee(self, mock_input, mock_raw_table, mock_db):
mock_db().show_members().fetchall.return_value = [[1, 2, 3]]
mock_input.return_value = '0,1,2'
admin = Admin()
result = admin.choose_multiple_employee()
assert result, [1]