Function test_multiple_cte_extraction

tests/parseutils/test_ctes.py:105–137

()

def test_multiple_cte_extraction():
    sql = """WITH
            x AS (SELECT abc, def FROM x),
            y AS (SELECT ghi, jkl FROM y)
            SELECT * FROM a, b"""

    start1 = len(
        """WITH
            x AS """
    )

    stop1 = len(
        """WITH
            x AS (SELECT abc, def FROM x)"""
    )

    start2 = len(
        """WITH
            x AS (SELECT abc, def FROM x),
            y AS """
    )

    stop2 = len(
        """WITH
            x AS (SELECT abc, def FROM x),
            y AS (SELECT ghi, jkl FROM y)"""
    )

    ctes, remainder = extract_ctes(sql)
    assert tuple(ctes) == (
        ("x", ("abc", "def"), start1, stop1),
        ("y", ("ghi", "jkl"), start2, stop2),
    )