完善测试

liangliangyy · liangliangyy · commit d2df0107a2ca · 2017-04-09T16:26:36.000+08:00
diff --git a/DjangoBlog/spider_notify.py b/DjangoBlog/spider_notify.py
@@ -36,7 +36,7 @@ def __google_notify():
             print(e)
 
     @staticmethod
-    def notify(self, url):
+    def notify(url):
 
         SpiderNotify.baidu_notify(url)
         SpiderNotify.__google_notify()
diff --git a/accounts/tests.py b/accounts/tests.py
@@ -4,6 +4,7 @@
 from django.contrib.sites.models import Site
 import datetime
 from accounts.models import BlogUser
+from django.core.urlresolvers import reverse
 
 
 # Create your tests here.
@@ -39,3 +40,45 @@ def test_validate_account(self):
 
         response = self.client.get(article.get_admin_url())
         self.assertEqual(response.status_code, 200)
+
+    def test_validate_register(self):
+        self.assertEquals(0, len(BlogUser.objects.filter(email='user123@user.com')))
+        response = self.client.post(reverse('account:register'), {
+            'username': 'user1233',
+            'email': 'user123@user.com',
+            'password1': 'password123',
+            'password2': 'password123',
+        })
+        self.assertEquals(1, len(BlogUser.objects.filter(email='user123@user.com')))
+
+        self.client.login(username='user1233', password='password123')
+        user = BlogUser.objects.filter(email='user123@user.com')[0]
+        user.is_superuser = True
+        user.is_staff = True
+        user.save()
+        article = Article()
+        article.title = "nicetitle333"
+        article.body = "nicecontentttt"
+        article.author = user
+
+        article.type = 'a'
+        article.status = 'p'
+        article.save()
+
+        response = self.client.get(article.get_admin_url())
+        self.assertEqual(response.status_code, 200)
+
+        response = self.client.get(reverse('account:logout'))
+        self.assertEqual(response.status_code, 302)
+
+        response = self.client.get(article.get_admin_url())
+        self.assertEqual(response.status_code, 302)
+
+        response = self.client.post(reverse('account:login'), {
+            'username': 'user1233',
+            'password': 'password123'
+        })
+        self.assertEqual(response.status_code, 302)
+
+        response = self.client.get(article.get_admin_url())
+        self.assertEqual(response.status_code, 200)
diff --git a/blog/tests.py b/blog/tests.py
@@ -27,8 +27,6 @@ def test_validate_article(self):
         category.last_mod_time = datetime.datetime.now()
         category.save()
 
-        response = self.client.get(category.get_absolute_url())
-        self.assertEqual(response.status_code, 200)
         tag = Tag()
         tag.name = "nicetag"
         tag.save()
@@ -53,6 +51,12 @@ def test_validate_article(self):
         response = self.client.get(tag.get_absolute_url())
         self.assertEqual(response.status_code, 200)
 
+        response = self.client.get(category.get_absolute_url())
+        self.assertEqual(response.status_code, 200)
+
+        from DjangoBlog.spider_notify import SpiderNotify
+        SpiderNotify.baidu_notify([article.get_full_url()])
+
     def test_validate_feed(self):
         user = BlogUser.objects.get_or_create(email="liangliangyy12@gmail.com", username="liangliangyy")[0]
         user.set_password("liangliangyy")
