File tree Expand file tree Collapse file tree 2 files changed +10
-4
lines changed
sqlmesh/core/engine_adapter Expand file tree Collapse file tree 2 files changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -1108,7 +1108,7 @@ def _execute(
11081108
11091109 # Create job config with reservation support
11101110 job_config = QueryJobConfig (** self ._job_params , connection_properties = connection_properties )
1111-
1111+
11121112 # Set reservation directly on the job_config object if specified
11131113 reservation_id = self ._extra_config .get ("reservation_id" )
11141114 if reservation_id :
Original file line number Diff line number Diff line change @@ -1050,12 +1050,18 @@ def test_bigquery(make_config):
10501050 check_import = False ,
10511051 )
10521052 assert isinstance (config_with_reservation , BigQueryConnectionConfig )
1053- assert config_with_reservation .reservation_id == "projects/my-project/locations/us-central1/reservations/my-reservation"
1054-
1053+ assert (
1054+ config_with_reservation .reservation_id
1055+ == "projects/my-project/locations/us-central1/reservations/my-reservation"
1056+ )
1057+
10551058 # Test that reservation_id is included in _extra_engine_config
10561059 extra_config = config_with_reservation ._extra_engine_config
10571060 assert "reservation_id" in extra_config
1058- assert extra_config ["reservation_id" ] == "projects/my-project/locations/us-central1/reservations/my-reservation"
1061+ assert (
1062+ extra_config ["reservation_id" ]
1063+ == "projects/my-project/locations/us-central1/reservations/my-reservation"
1064+ )
10591065
10601066 with pytest .raises (ConfigError , match = "you must also specify the `project` field" ):
10611067 make_config (type = "bigquery" , execution_project = "execution_project" , check_import = False )
You can’t perform that action at this time.
0 commit comments