Fix utils to not require TENANT_USE_ASGIREF to be defined in the ho…

…st django project (#206) * Fix if statement * try fix tests
citusdata · Dec 18, 2023 · 44b4e86 · 44b4e86
1 parent d119cc5
commit 44b4e86
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/django_multitenant/tests/test_utils.py b/django_multitenant/tests/test_utils.py
@@ -58,6 +58,7 @@ def test_tenant_persists_from_async_task_to_thread(self):
         account = projects[0].account
 
         with self.settings(TENANT_USE_ASGIREF=True):
+            importlib.reload(sys.modules["django_multitenant.settings"])
             importlib.reload(sys.modules["django_multitenant.utils"])
 
             # Set the tenant in task
@@ -66,6 +67,7 @@ def test_tenant_persists_from_async_task_to_thread(self):
             unset_current_tenant()
 
         with self.settings(TENANT_USE_ASGIREF=False):
+            importlib.reload(sys.modules["django_multitenant.settings"])
             importlib.reload(sys.modules["django_multitenant.utils"])
 
             # Set the tenant in task

diff --git a/django_multitenant/utils.py b/django_multitenant/utils.py
@@ -1,10 +1,10 @@
 import inspect
 
 from django.apps import apps
-from django.conf import settings
+from .settings import TENANT_USE_ASGIREF
 
 
-if settings.TENANT_USE_ASGIREF:
+if TENANT_USE_ASGIREF:
     # asgiref must be installed, its included with Django >= 3.0
     from asgiref.local import Local as local
 else: