def test_remove_supervisor(self, mock_db):
mock_db().check_supervisor_accident.return_value = False
admin = Admin()
admin.remove_supervisor(1, 1)
mock_db().remove_supervisor_accident.assert_called_once()
mock_db().check_supervisor_accident.assert_called_once()
mock_db().remove_supervisor_role.assert_called_once()
def test_remove_supervisor_exception(self, mock_db):
mock_db().remove_supervisor_accident.side_effect = Exception
admin = Admin()
admin.remove_supervisor(1, 1)
mock_db().remove_supervisor_accident.assert_called_once()
mock_db().rollback_data.assert_called_once()
