Function inserted

tests/_test_transaction.py:37–51

Return the values inserted in the test table.

(conn)

def inserted(conn):
    """Return the values inserted in the test table."""
    sql = "SELECT * FROM test_table"
    if isinstance(conn, psycopg.Connection):
        rows = conn.cursor().execute(sql).fetchall()
        return {v for (v,) in rows}
    else:

        async def f():
            cur = conn.cursor()
            await cur.execute(sql)
            rows = await cur.fetchall()
            return {v for (v,) in rows}

        return f()


def in_transaction(conn):