Add env vars for unit tests

singlestore-labs · Aug 21, 2024 · 315727d · 315727d
1 parent 306603a
commit 315727d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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()