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()