Function __auto_closed_cursor

tests/conftest.py:572–584

Returns a cursor created from conn. This is intended to be used in a "with" block. The cursor will be closed upon exiting the block.

(conn)

@contextlib.contextmanager
def __auto_closed_cursor(conn):
  """Returns a cursor created from conn. This is intended to be used in a "with" block.
     The cursor will be closed upon exiting the block.
  """
  cursor = conn.cursor()
  cursor.conn = conn
  try:
    yield cursor
  finally:
    try:
      cursor.close()
    except Exception as e:
      LOG.warn("Error closing Impala cursor: %s", e)


@pytest.fixture