@@ -1216,6 +1216,10 @@ def test_templating(self, create_task_instance_of_operator, session):
1216
1216
region = "{{ 'region' }}" ,
1217
1217
project_id = "{{ 'project_id' }}" ,
1218
1218
impersonation_chain = "{{ 'impersonation-chain' }}" ,
1219
+ template_path = "{{ 'template-path' }}" ,
1220
+ pipeline_root = "{{ 'pipeline-root' }}" ,
1221
+ service_account = "{{ 'service-account' }}" ,
1222
+ parameter_values = {"{{ 'parameter-key' }}" : "{{ 'parameter-value' }}" },
1219
1223
# Other parameters
1220
1224
dag_id = "test_template_body_templating_dag" ,
1221
1225
task_id = "test_template_body_templating_task" ,
@@ -1229,6 +1233,10 @@ def test_templating(self, create_task_instance_of_operator, session):
1229
1233
assert task .region == "region"
1230
1234
assert task .project_id == "project_id"
1231
1235
assert task .impersonation_chain == "impersonation-chain"
1236
+ assert task .template_path == "template-path"
1237
+ assert task .pipeline_root == "pipeline-root"
1238
+ assert task .service_account == "service-account"
1239
+ assert task .parameter_values == {"parameter-key" : "parameter-value" }
1232
1240
assert task .training_pipeline_id == "training-pipeline-id"
1233
1241
assert task .custom_job_id == "custom_job_id"
1234
1242
0 commit comments