re-enable expansion service

apache · Sep 22, 2023 · babff94 · babff94
1 parent 491c1ac
commit babff94
Showing 1 changed file with 4 additions and 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
@@ -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))