From babff9484f4304342a2f15db69ec57e138f5be5f Mon Sep 17 00:00:00 2001 From: ahmedabu98 Date: Fri, 22 Sep 2023 16:09:22 -0400 Subject: [PATCH] re-enable expansion service --- sdks/python/apache_beam/io/gcp/bigtableio_it_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/sdks/python/apache_beam/io/gcp/bigtableio_it_test.py b/sdks/python/apache_beam/io/gcp/bigtableio_it_test.py index c9506e2a8a66..9a54c50fcf19 100644 --- a/sdks/python/apache_beam/io/gcp/bigtableio_it_test.py +++ b/sdks/python/apache_beam/io/gcp/bigtableio_it_test.py @@ -144,9 +144,9 @@ def test_read_xlang(self): @pytest.mark.uses_gcp_java_expansion_service @pytest.mark.uses_transform_service -# @unittest.skipUnless( -# os.environ.get('EXPANSION_PORT'), -# "EXPANSION_PORT environment var is not provided.") +@unittest.skipUnless( + os.environ.get('EXPANSION_PORT'), + "EXPANSION_PORT environment var is not provided.") @unittest.skipIf(client is None, 'Bigtable dependencies are not installed') class TestWriteToBigtableXlangIT(unittest.TestCase): # These are integration tests for the cross-language write transform. @@ -158,7 +158,7 @@ def setUpClass(cls): cls.test_pipeline = TestPipeline(is_integration_test=True) cls.project = cls.test_pipeline.get_option('project') cls.args = cls.test_pipeline.get_full_options_as_args() - cls.expansion_service = None #('localhost:%s' % os.environ.get('EXPANSION_PORT')) + cls.expansion_service = ('localhost:%s' % os.environ.get('EXPANSION_PORT')) instance_id = '%s-%s-%s' % ( cls.INSTANCE, str(int(time.time())), secrets.token_hex(3))