def test_update_employee_email(self, mock_validation, mock_db, mock_print, mock_input, mock_table_menu, mock_menu):
mock_db().worker_table_fields.return_value = []
mock_menu().draw_menu.return_value = 'email'
admin = Admin()
admin.update_employee()
mock_db().update_member.assert_called_once()
mock_db().commit_data.assert_called_once()
def test_update_employee_back_button(self, mock_db, mock_print, mock_input, mock_table_menu, mock_menu):
mock_db().worker_table_fields.return_value = []
mock_menu().draw_menu.return_value = 'BACK'
admin = Admin()
admin.update_employee()
mock_print.assert_called_once()
mock_input.assert_called_once()
def test_update_employee_sqlite_exception(self, mock_validation, mock_db, mock_print, mock_input, mock_table_menu, mock_menu):
mock_db().worker_table_fields.return_value = []
mock_menu().draw_menu.return_value = 'email'
mock_db().commit_data.side_effect = sqlite3.IntegrityError
admin = Admin()
admin.update_employee()
mock_db().update_member.assert_called_once()
def test_update_employee_no_emp(self, mock_db, mock_table_menu):
mock_table_menu.return_value = None
admin = Admin()
admin.update_employee()
mock_table_menu.assert_called_once()
