From 306603aba5429de9af8333c56095a10e0d9ce1f8 Mon Sep 17 00:00:00 2001 From: Kevin Smith Date: Wed, 21 Aug 2024 13:10:12 -0500 Subject: [PATCH] Add env vars for unit tests --- singlestoredb/tests/test_fusion.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/singlestoredb/tests/test_fusion.py b/singlestoredb/tests/test_fusion.py index 70166cbb..2dec25a4 100644 --- a/singlestoredb/tests/test_fusion.py +++ b/singlestoredb/tests/test_fusion.py @@ -519,6 +519,8 @@ def tearDownClass(cls): def setUp(self): self.enabled = os.environ.get('SINGLESTOREDB_FUSION_ENABLED') os.environ['SINGLESTOREDB_FUSION_ENABLED'] = '1' + self.hidden_enabled = os.environ.get('SINGLESTOREDB_FUSION_ENABLE_HIDDEN') + os.environ['SINGLESTOREDB_FUSION_ENABLE_HIDDEN'] = '1' self.conn = s2.connect(database=type(self).dbname, local_infile=True) self.cur = self.conn.cursor() @@ -528,6 +530,11 @@ def tearDown(self): else: del os.environ['SINGLESTOREDB_FUSION_ENABLED'] + if self.hidden_enabled: + os.environ['SINGLESTOREDB_FUSION_ENABLE_HIDDEN'] = self.hidden_enabled + else: + del os.environ['SINGLESTOREDB_FUSION_ENABLE_HIDDEN'] + try: if self.cur is not None: self.cur.close()