test(integration): pass typed x_database_id to execute_saved_query

zfarrell · zfarrell · commit ea00a1b58b64 · 2026-06-02T09:34:29.000-07:00
diff --git a/tests/integration/test_saved_query_versioning.py b/tests/integration/test_saved_query_versioning.py
@@ -50,11 +50,10 @@ def test_saved_query_versioning(
             f"expected versions 1,2,3 in {sorted(version_numbers)}"
         )
 
-        # execute_saved_query runs SQL, so it also needs the database scope.
-        # The endpoint has no typed x_database_id param yet, so set the header
-        # directly via the _headers override.
+        # execute_saved_query runs SQL, so it also needs the database scope,
+        # passed via the typed x_database_id parameter.
         executed = saved_queries_api.execute_saved_query(
-            created.id, _headers={"X-Database-Id": database_id}
+            created.id, x_database_id=database_id
         )
         assert executed.row_count == 1
         assert executed.rows == [[3]]

Original file line number	Diff line number	Diff line change
`@@ -50,11 +50,10 @@ def test_saved_query_versioning(`
`50`	`50`	`f"expected versions 1,2,3 in {sorted(version_numbers)}"`
`51`	`51`	`)`
`52`	`52`
`53`		`- # execute_saved_query runs SQL, so it also needs the database scope.`
`54`		`- # The endpoint has no typed x_database_id param yet, so set the header`
`55`		`- # directly via the _headers override.`
	`53`	`+ # execute_saved_query runs SQL, so it also needs the database scope,`
	`54`	`+ # passed via the typed x_database_id parameter.`
`56`	`55`	`executed = saved_queries_api.execute_saved_query(`
`57`		`- created.id, _headers={"X-Database-Id": database_id}`
	`56`	`+ created.id, x_database_id=database_id`
`58`	`57`	`)`
`59`	`58`	`assert executed.row_count == 1`
`60`	`59`	`assert executed.rows == [[3]]`