Port compute_api_version to other TD implementations (#27518)

3 years ago · 550faa6dc7
parent 251274022c
commit 550faa6dc7
1 changed files with 7 additions and 3 deletions
--- 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)