Bladeren bron

Import fix try

Denis K 10 jaren geleden
bovenliggende
commit
407a5f373c
1 gewijzigde bestanden met toevoegingen van 6 en 2 verwijderingen
  1. 6 2
      jet/tests/test_dashboard.py

+ 6 - 2
jet/tests/test_dashboard.py

@@ -1,4 +1,3 @@
-from django.contrib.auth.models import User
 from django.test import TestCase, Client
 from django.test import TestCase, Client
 from jet.dashboard.modules import LinkList, RecentActions
 from jet.dashboard.modules import LinkList, RecentActions
 from jet.dashboard.models import UserDashboardModule
 from jet.dashboard.models import UserDashboardModule
@@ -14,12 +13,17 @@ class DashboardTestCase(TestCase):
         self._login()
         self._login()
         self._init_dashboard()
         self._init_dashboard()
 
 
+    @property
+    def User(self):
+        from django.contrib.auth.models import User
+        return User
+
     def _login(self):
     def _login(self):
         username = 'admin'
         username = 'admin'
         email = 'admin@example.com'
         email = 'admin@example.com'
         password = 'admin'
         password = 'admin'
         self.admin = Client()
         self.admin = Client()
-        self.admin_user = User.objects.create_superuser(username, email, password)
+        self.admin_user = self.User.objects.create_superuser(username, email, password)
         return self.admin.login(username=username, password=password)
         return self.admin.login(username=username, password=password)
 
 
     def _init_dashboard(self):
     def _init_dashboard(self):