diff --git a/framework/xds_k8s_testcase.py b/framework/xds_k8s_testcase.py index a5044476..57389a91 100644 --- a/framework/xds_k8s_testcase.py +++ b/framework/xds_k8s_testcase.py @@ -906,7 +906,7 @@ def initTrafficDirectorManager(self) -> TrafficDirectorSecureManager: compute_api_version=self.compute_api_version, ) - def initKubernetesServerRunner(self) -> KubernetesServerRunner: + def initKubernetesServerRunner(self, **kwargs) -> KubernetesServerRunner: return KubernetesServerRunner( k8s.KubernetesNamespace( self.k8s_api_manager, self.server_namespace @@ -921,9 +921,10 @@ def initKubernetesServerRunner(self) -> KubernetesServerRunner: xds_server_uri=self.xds_server_uri, deployment_template="server-secure.deployment.yaml", debug_use_port_forwarding=self.debug_use_port_forwarding, + **kwargs, ) - def initKubernetesClientRunner(self) -> KubernetesClientRunner: + def initKubernetesClientRunner(self, **kwargs) -> KubernetesClientRunner: return KubernetesClientRunner( k8s.KubernetesNamespace( self.k8s_api_manager, self.client_namespace @@ -940,6 +941,7 @@ def initKubernetesClientRunner(self) -> KubernetesClientRunner: stats_port=self.client_port, reuse_namespace=self.server_namespace == self.client_namespace, debug_use_port_forwarding=self.debug_use_port_forwarding, + **kwargs, ) def startSecureTestServer(self, replica_count=1, **kwargs) -> XdsTestServer: