From 315727de9183d08b50e38cee32b328d8ed6c2968 Mon Sep 17 00:00:00 2001 From: Kevin Smith Date: Wed, 21 Aug 2024 13:48:58 -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 2dec25a4..0f11fbb2 100644 --- a/singlestoredb/tests/test_fusion.py +++ b/singlestoredb/tests/test_fusion.py @@ -141,6 +141,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() @@ -150,6 +152,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()