def test_cancel_query_failed(self, engine_mock): query = Query() cursor_mock = engine_mock.raiseError.side_effect = Exception() assert PostgresEngineSpec.cancel_query(cursor_mock, query, 123) is False
def test_cancel_query(self, engine_mock): query = Query() cursor_mock = engine_mock.return_value.__enter__.return_value assert PostgresEngineSpec.cancel_query(cursor_mock, query, 123) is True