def test_get_referenced_tables(self, tmp_path):
os.makedirs(tmp_path / "telmetry_derived")
query_file = tmp_path / "telmetry_derived" / "query.sql"
query_file.write_text(
"SELECT * FROM telemetry_derived.clients_daily_v6 "
"WHERE submission_date = '2020-01-01'"
)
dryrun = DryRun(str(query_file))
response = dryrun.get_referenced_tables()
assert len(response) == 1
assert response[0]["datasetId"] == "telemetry_derived"
assert response[0]["tableId"] == "clients_daily_v6"
def test_get_referenced_tables_empty(self, tmp_path):
query_file = tmp_path / "query.sql"
query_file.write_text("SELECT 123")
dryrun = DryRun(str(query_file))
assert dryrun.get_referenced_tables() == []
