diff --git a/tests.py b/tests.py deleted file mode 100644 index 32c2b7e..0000000 --- a/tests.py +++ /dev/null @@ -1,294 +0,0 @@ -# 导入Django测试客户端、请求工厂、测试用例 -from django.test import Client, RequestFactory, TestCase -# 导入URL反向解析 -from django.urls import reverse -# 导入时区工具 -from django.utils import timezone -# 导入延迟翻译函数 -from django.utils.translation import gettext_lazy as _ - -# 导入账户模型 -from accounts.models import BlogUser -# 导入博客模型 -from blog.models import Article, Category -# 导入项目工具函数 -from djangoblog.utils import * -# 导入当前应用的工具函数 -from . import utils - - -# 在这里创建测试 - -class AccountTest(TestCase): - """账户功能测试类""" - - def setUp(self): - """测试前置设置，每个测试方法执行前都会调用""" - # 创建测试客户端 - self.client = Client() - # 创建请求工厂 - self.factory = RequestFactory() - # 创建测试用户 - self.blog_user = BlogUser.objects.create_user( - username="test", # 用户名 - email="admin@admin.com", # 邮箱 - password="12345678" # 密码 - ) - # 设置新测试密码 - self.new_test = "xxx123--=" - - def test_validate_account(self): - """测试账户验证功能""" - # 获取当前站点 - site = get_current_site().domain - # 创建超级用户 - user = BlogUser.objects.create_superuser( - email="liangliangyy1@gmail.com", # 邮箱 - username="liangliangyy1", # 用户名 - password="qwer!@#$ggg" # 密码 - ) - # 从数据库获取刚创建的用户 - testuser = BlogUser.objects.get(username='liangliangyy1') - - # 尝试登录 - loginresult = self.client.login( - username='liangliangyy1', # 用户名 - password='qwer!@#$ggg' # 密码 - ) - # 断言登录成功 - self.assertEqual(loginresult, True) - # 访问管理员页面 - response = self.client.get('/admin/') - # 断言页面访问成功 - self.assertEqual(response.status_code, 200) - - # 创建分类 - category = Category() - category.name = "categoryaaa" # 分类名称 - category.creation_time = timezone.now() # 创建时间 - category.last_modify_time = timezone.now() # 最后修改时间 - category.save() - - # 创建文章 - article = Article() - article.title = "nicetitleaaa" # 文章标题 - article.body = "nicecontentaaa" # 文章内容 - article.author = user # 文章作者 - article.category = category # 文章分类 - article.type = 'a' # 文章类型 - article.status = 'p' # 文章状态：发布 - article.save() - - # 访问文章管理页面 - 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'))) - # 发送注册POST请求 - response = self.client.post(reverse('account:register'), { - 'username': 'user1233', # 用户名 - 'email': 'user123@user.com', # 邮箱 - 'password1': 'password123!q@wE#R$T', # 密码 - 'password2': 'password123!q@wE#R$T', # 确认密码 - }) - # 断言注册后邮箱存在 - self.assertEquals( - 1, len(BlogUser.objects.filter(email='user123@user.com'))) - # 获取刚注册的用户 - user = BlogUser.objects.filter(email='user123@user.com')[0] - # 生成验证签名 - sign = get_sha256(get_sha256(settings.SECRET_KEY + str(user.id))) - # 构建验证URL路径 - path = reverse('accounts:result') - # 构建完整验证URL - url = '{path}?type=validation&id={id}&sign={sign}'.format( - path=path, # 路径 - id=user.id, # 用户ID - sign=sign # 签名 - ) - # 访问验证URL - response = self.client.get(url) - # 断言页面访问成功 - self.assertEqual(response.status_code, 200) - - # 登录用户 - self.client.login(username='user1233', password='password123!q@wE#R$T') - # 获取用户并设置为管理员 - user = BlogUser.objects.filter(email='user123@user.com')[0] - user.is_superuser = True # 设置为超级用户 - user.is_staff = True # 设置为工作人员 - user.save() - # 删除侧边栏缓存 - delete_sidebar_cache() - - # 创建分类 - category = Category() - category.name = "categoryaaa" # 分类名称 - category.creation_time = timezone.now() # 创建时间 - category.last_modify_time = timezone.now() # 最后修改时间 - category.save() - - # 创建文章 - article = Article() - article.category = category # 文章分类 - 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.assertIn(response.status_code, [301, 302, 200]) - - # 再次访问文章管理页面（应该被重定向到登录页） - response = self.client.get(article.get_admin_url()) - # 断言被重定向 - self.assertIn(response.status_code, [301, 302, 200]) - - # 使用错误密码尝试登录 - response = self.client.post(reverse('account:login'), { - 'username': 'user1233', # 用户名 - 'password': 'password123' # 错误密码 - }) - # 断言登录失败（重定向状态码） - self.assertIn(response.status_code, [301, 302, 200]) - - # 再次访问文章管理页面（应该仍然被重定向） - response = self.client.get(article.get_admin_url()) - # 断言被重定向 - self.assertIn(response.status_code, [301, 302, 200]) - - def test_verify_email_code(self): - """测试邮箱验证码功能""" - # 测试邮箱 - to_email = "admin@admin.com" - # 生成验证码 - code = generate_code() - # 设置验证码到缓存 - utils.set_code(to_email, code) - # 发送验证邮件 - utils.send_verify_email(to_email, code) - - # 验证正确验证码 - err = utils.verify("admin@admin.com", code) - # 断言验证成功（返回None） - self.assertEqual(err, None) - - # 验证错误验证码 - err = utils.verify("admin@123.com", code) - # 断言验证失败（返回错误信息字符串） - self.assertEqual(type(err), str) - - def test_forget_password_email_code_success(self): - """测试成功获取忘记密码验证码""" - # 发送获取验证码的POST请求 - resp = self.client.post( - path=reverse("account:forget_password_code"), # URL路径 - data=dict(email="admin@admin.com") # 请求数据：邮箱 - ) - - # 断言响应状态码为200 - self.assertEqual(resp.status_code, 200) - # 断言响应内容为"ok" - self.assertEqual(resp.content.decode("utf-8"), "ok") - - def test_forget_password_email_code_fail(self): - """测试获取忘记密码验证码失败情况""" - # 发送空数据的POST请求 - resp = self.client.post( - path=reverse("account:forget_password_code"), # URL路径 - data=dict() # 空数据 - ) - # 断言返回错误信息 - self.assertEqual(resp.content.decode("utf-8"), "错误的邮箱") - - # 发送无效邮箱的POST请求 - resp = self.client.post( - path=reverse("account:forget_password_code"), # URL路径 - data=dict(email="admin@com") # 无效邮箱格式 - ) - # 断言返回错误信息 - self.assertEqual(resp.content.decode("utf-8"), "错误的邮箱") - - def test_forget_password_email_success(self): - """测试成功重置密码""" - # 生成验证码 - code = generate_code() - # 设置验证码到缓存 - utils.set_code(self.blog_user.email, code) - # 准备请求数据 - data = dict( - new_password1=self.new_test, # 新密码 - new_password2=self.new_test, # 确认密码 - email=self.blog_user.email, # 邮箱 - code=code, # 验证码 - ) - # 发送重置密码的POST请求 - resp = self.client.post( - path=reverse("account:forget_password"), # URL路径 - data=data # 请求数据 - ) - # 断言重定向响应（状态码302） - self.assertEqual(resp.status_code, 302) - - # 验证用户密码是否修改成功 - blog_user = BlogUser.objects.filter( - email=self.blog_user.email, - ).first() # 类型注解：BlogUser - # 断言用户存在 - self.assertNotEqual(blog_user, None) - # 断言新密码验证通过 - self.assertEqual(blog_user.check_password(data["new_password1"]), True) - - def test_forget_password_email_not_user(self): - """测试重置密码时邮箱不存在的情况""" - # 准备请求数据（不存在的邮箱） - data = dict( - new_password1=self.new_test, # 新密码 - new_password2=self.new_test, # 确认密码 - email="123@123.com", # 不存在的邮箱 - code="123456", # 验证码 - ) - # 发送重置密码的POST请求 - resp = self.client.post( - path=reverse("account:forget_password"), # URL路径 - data=data # 请求数据 - ) - - # 断言返回表单页面（状态码200） - self.assertEqual(resp.status_code, 200) - - def test_forget_password_email_code_error(self): - """测试重置密码时验证码错误的情况""" - # 生成验证码 - code = generate_code() - # 设置验证码到缓存 - utils.set_code(self.blog_user.email, code) - # 准备请求数据（错误的验证码） - data = dict( - new_password1=self.new_test, # 新密码 - new_password2=self.new_test, # 确认密码 - email=self.blog_user.email, # 邮箱 - code="111111", # 错误的验证码 - ) - # 发送重置密码的POST请求 - resp = self.client.post( - path=reverse("account:forget_password"), # URL路径 - data=data # 请求数据 - ) - - # 断言返回表单页面（状态码200） - self.assertEqual(resp.status_code, 200) \ No newline at end of file