fix: Leverage actual database for rendering Jinjarized SQL (#27646)

This commit is contained in:
John Bodley
2024-03-27 08:12:25 +13:00
committed by GitHub
parent ed9e542781
commit 28cbedb82f
4 changed files with 15 additions and 12 deletions

View File

@@ -17,6 +17,7 @@
# pylint: disable=invalid-name, redefined-outer-name, too-many-lines
from typing import Optional
from unittest.mock import Mock
import pytest
import sqlparse
@@ -1959,7 +1960,10 @@ def test_extract_tables_from_jinja_sql(
expected: set[Table],
) -> None:
assert (
extract_tables_from_jinja_sql(sql.format(engine=engine, macro=macro), engine)
extract_tables_from_jinja_sql(
sql=sql.format(engine=engine, macro=macro),
database=Mock(),
)
== expected
)