test: Fix grant test env var (#89)

firebolt-db · Dec 19, 2022 · e347c3b · e347c3b
1 parent 2aeddc1
commit e347c3b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/functional/adapter/test_grants.py b/tests/functional/adapter/test_grants.py
@@ -1,3 +1,6 @@
+import os
+from unittest import mock
+
 from dbt.tests.adapter.grants.base_grants import BaseGrants
 from dbt.tests.adapter.grants.test_incremental_grants import (
     BaseIncrementalGrants,
@@ -25,6 +28,18 @@ class TestGrantsFailWithException(BaseGrants):
     grant functionality in Firebolt.
     """
 
+    @fixture(autouse=True)
+    def env_setup(monkeypatch):
+        with mock.patch.dict(
+            os.environ,
+            {
+                'DBT_TEST_USER_1': 'dummy-user1',
+                'DBT_TEST_USER_2': 'dummy-user2',
+                'DBT_TEST_USER_3': 'dummy-user3',
+            },
+        ):
+            yield
+
     @fixture(scope='class')
     def models(self):
         updated_schema = self.interpolate_name_overrides(model_schema_yml)