ソースを参照

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):