浏览代码

Import fix try

Denis K 8 年之前
父节点
当前提交
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):