From 550faa6dc733cd4db4119eed1a8ad8bf286f39e4 Mon Sep 17 00:00:00 2001 From: Lidi Zheng Date: Wed, 29 Sep 2021 14:23:31 -0700 Subject: [PATCH] Port compute_api_version to other TD implementations (#27518) --- .../framework/infrastructure/traffic_director.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py b/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py index 6e8bbfed919..5f39ff0a129 100644 --- a/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py +++ b/tools/run_tests/xds_k8s_test_driver/framework/infrastructure/traffic_director.py @@ -556,12 +556,14 @@ class TrafficDirectorAppNetManager(TrafficDirectorManager): *, resource_prefix: str, resource_suffix: Optional[str] = None, - network: str = 'default'): + network: str = 'default', + compute_api_version: str = 'v1'): super().__init__(gcp_api_manager, project, resource_prefix=resource_prefix, resource_suffix=resource_suffix, - network=network) + network=network, + compute_api_version=compute_api_version) # API self.netsvc = _NetworkServicesV1Alpha1(gcp_api_manager, project) @@ -656,12 +658,14 @@ class TrafficDirectorSecureManager(TrafficDirectorManager): resource_prefix: str, resource_suffix: Optional[str] = None, network: str = 'default', + compute_api_version: str = 'v1', ): super().__init__(gcp_api_manager, project, resource_prefix=resource_prefix, resource_suffix=resource_suffix, - network=network) + network=network, + compute_api_version=compute_api_version) # API self.netsec = _NetworkSecurityV1Beta1(gcp_api_manager, project)