|
|
# 导入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) |
