From 351bfad1f75c35488b59e581c3270311d06acd9c Mon Sep 17 00:00:00 2001 From: Sergii Tkachenko Date: Fri, 5 Aug 2022 14:45:01 -0700 Subject: [PATCH] xDS interop: log the subTest start and beginning (#30517) To improve debugging of the tests with steps that look similar, f.e. failover. Makes the end of one subtest, and the beginning of the next one much clearer. Note: URL map test suite does not use subtests, so I didn't add the logging there. --- .../xds_k8s_test_driver/framework/xds_k8s_testcase.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tools/run_tests/xds_k8s_test_driver/framework/xds_k8s_testcase.py b/tools/run_tests/xds_k8s_test_driver/framework/xds_k8s_testcase.py index dbd7b5a20a3..ff568614570 100644 --- a/tools/run_tests/xds_k8s_test_driver/framework/xds_k8s_testcase.py +++ b/tools/run_tests/xds_k8s_test_driver/framework/xds_k8s_testcase.py @@ -12,6 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. import abc +import contextlib import datetime import enum import hashlib @@ -169,6 +170,14 @@ class XdsKubernetesBaseTestCase(absltest.TestCase): cls.secondary_k8s_api_manager.close() cls.gcp_api_manager.close() + @contextlib.contextmanager + def subTest(self, msg, **params): # noqa pylint: disable=signature-differs + logger.info('--- Starting subTest %s.%s ---', self.id(), msg) + try: + yield super().subTest(msg, **params) + finally: + logger.info('--- Finished subTest %s.%s ---', self.id(), msg) + def setupTrafficDirectorGrpc(self): self.td.setup_for_grpc(self.server_xds_host, self.server_xds_port,