Function test_get_table_names

duckdb_engine/tests/test_basic.py:215–238

(inspector: Inspector, session: Session)

    reason="ATTACH is not supported for DuckDB version < 0.7.0",
)
def test_get_table_names(inspector: Inspector, session: Session) -> None:
    # Using multi-line strings because of all the single and double quotes flying around...
    cmds = [
        """CREATE TABLE "daffy duck"."quack quack"."t2" (i INTEGER, j INTEGER);""",
        """CREATE TABLE "t3" (i INTEGER, j INTEGER);""",
        """CREATE SCHEMA "porky" """,
        """CREATE TABLE "porky"."t4" (i INTEGER, j INTEGER);""",
    ]
    for cmd in cmds:
        session.execute(text(cmd))
        session.commit()

    for schema, table_names in zip(
        ['"daffy duck"."quack quack"', "main", "porky"], [["t1", "t2"], ["t3"], ["t4"]]
    ):
        _table_names = inspector.get_table_names(schema=schema)
        assert set(_table_names).issuperset(set(table_names))
        for _table_name in _table_names:
            assert inspector.has_table(_table_name, schema)

    table_names_all = inspector.get_table_names()
    assert set(table_names_all).issuperset({"t1", "t2", "t3", "t4"})
    for table_name in table_names_all:
        assert inspector.has_table(table_name)


def test_get_views(conn: Connection, dialect: Dialect) -> None: