فهرست منبع

Import fix try

Denis K 9 سال پیش
والد
کامیت
407a5f373c
1فایلهای تغییر یافته به همراه6 افزوده شده و 2 حذف شده
  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):