Fixing error with district user and cache

openimis · Nov 8, 2024 · 772b81a · 772b81a
1 parent e54270d
commit 772b81a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/location/models.py b/location/models.py
@@ -263,7 +263,7 @@ def cache_location_graph(location_id=None):
 
 
 def cache_location_graph_if_empty():
-    if cache.has_key("location_types") is None:
+    if not cache.has_key("location_types"):
         cache_location_graph()
 
 

diff --git a/location/tests/test.py b/location/tests/test.py
@@ -9,7 +9,7 @@
 from claim.test_helpers import create_test_claim_admin
 from django.core.cache import caches
 
-from location.models import LocationManager
+from location.models import LocationManager, UserDistrict
 from core.services import (
     create_or_update_interactive_user,
     create_or_update_core_user,
@@ -123,6 +123,9 @@ def test_allowed_location(self):
         )
         cached = caches["location"].get(f"user_locations_{self.test_user._u.id}")
         self.assertIsNotNone(cached)
+        districts = UserDistrict.get_user_districts(self.test_user)
+        self.assertIsNotNone(cached)
+
 
     def test_cache_invalidation(self):
         LocationManager().is_allowed(self.test_user, [])