# 导入Python内置os模块，用于文件路径操作（如图片文件的保存与删除） import os # 导入Django项目配置模块，用于获取项目设置（如分页数量、文件上传路径） from django.conf import settings # 导入Django文件上传相关类，用于模拟文件上传请求（如图片上传测试） from django.core.files.uploadedfile import SimpleUploadedFile # 导入Django管理命令调用函数，用于在测试中执行自定义管理命令（如重建搜索索引） from django.core.management import call_command # 导入Django分页类，用于测试分页功能 from django.core.paginator import Paginator # 导入Django静态文件模板标签，用于生成静态文件URL（如测试用户头像） from django.templatetags.static import static # 导入Django测试核心模块： # 1. Client：模拟HTTP客户端，用于发送GET/POST请求并获取响应 # 2. RequestFactory：生成请求对象，用于测试视图函数/模板标签 # 3. TestCase：Django测试基类，提供测试环境初始化、断言方法等 from django.test import Client, RequestFactory, TestCase # 导入Django URL反向解析模块，用于生成测试用的URL（避免硬编码） from django.urls import reverse # 导入Django时区工具，用于处理时间字段（如模型创建时间） from django.utils import timezone # 从accounts应用导入用户模型BlogUser（自定义用户模型，替代Django默认用户模型） from accounts.models import BlogUser # 从当前应用（blog）导入搜索表单BlogSearchForm，用于测试表单功能 from blog.forms import BlogSearchForm # 从当前应用导入核心模型：文章、分类、标签、侧边栏、友情链接 from blog.models import Article, Category, Tag, SideBar, Links # 从当前应用导入自定义模板标签函数，用于测试模板标签的逻辑正确性 from blog.templatetags.blog_tags import load_pagination_info, load_articletags # 从自定义工具模块导入工具函数： # 1. get_current_site：获取当前站点信息（用于生成完整URL） # 2. get_sha256：生成SHA256加密字符串（用于测试文件上传签名验证） from djangoblog.utils import get_current_site, get_sha256 # 从oauth应用导入第三方登录相关模型（OAuth用户、OAuth配置），用于测试第三方登录数据 from oauth.models import OAuthUser, OAuthConfig # 测试类创建标识注释（Django测试框架约定：测试类需继承TestCase） # Create your tests here. # 定义文章相关测试类ArticleTest，继承自Django的TestCase（测试基类） # 作用：集中测试博客核心功能，包括模型操作、URL访问、搜索、分页、文件上传等 class ArticleTest(TestCase): # 测试初始化方法：在每个测试方法执行前自动调用，用于准备测试环境 def setUp(self): # 初始化HTTP客户端（模拟用户发送请求） self.client = Client() # 初始化请求工厂（用于生成原始请求对象，测试视图/模板标签时使用） self.factory = RequestFactory() # 核心测试方法：测试文章相关完整流程（模型创建、关联、URL访问、搜索等） def test_validate_article(self): # 获取当前站点域名（用于验证完整URL生成） site = get_current_site().domain # 创建/获取测试用户：邮箱、用户名固定，若已存在则直接获取（get_or_create返回元组，取第0个元素） user = BlogUser.objects.get_or_create( email="liangliangyy@gmail.com", username="liangliangyy")[0] # 设置用户密码（set_password会自动加密，避免明文存储） user.set_password("liangliangyy") # 设置用户为 staff（后台管理权限）和 superuser（超级管理员权限） user.is_staff = True user.is_superuser = True # 保存用户信息到数据库 user.save() # 模拟访问用户个人主页，验证响应状态码为200（正常访问） response = self.client.get(user.get_absolute_url()) self.assertEqual(response.status_code, 200) # 模拟访问后台邮件发送日志页面（测试后台URL可达性） response = self.client.get('/admin/servermanager/emailsendlog/') # 模拟访问后台操作日志列表页面（测试后台URL可达性） response = self.client.get('admin/admin/logentry/') # 创建测试侧边栏数据 s = SideBar() s.sequence = 1 # 排序序号（控制显示顺序） s.name = 'test' # 侧边栏标题 s.content = 'test content' # 侧边栏内容（HTML文本） s.is_enable = True # 启用侧边栏 s.save() # 保存到数据库 # 创建测试分类数据 category = Category() category.name = "category" # 分类名称 category.creation_time = timezone.now() # 创建时间 category.last_mod_time = timezone.now() # 最后修改时间 category.save() # 保存到数据库 # 创建测试标签数据 tag = Tag() tag.name = "nicetag" # 标签名称 tag.save() # 保存到数据库 # 创建测试文章数据 article = Article() article.title = "nicetitle" # 文章标题 article.body = "nicecontent" # 文章内容 article.author = user # 关联作者（测试用户） article.category = category # 关联分类 article.type = 'a' # 文章类型：普通文章（'a'=Article） article.status = 'p' # 文章状态：已发布（'p'=Published） article.save() # 保存到数据库 # 验证文章初始标签数量为0（未添加标签） self.assertEqual(0, article.tags.count()) # 为文章添加标签（多对多关联） article.tags.add(tag) article.save() # 再次保存，更新关联关系 # 验证文章标签数量为1（添加成功） self.assertEqual(1, article.tags.count()) # 批量创建20篇测试文章（用于测试分页功能） for i in range(20): article = Article() article.title = "nicetitle" + str(i) # 标题带序号，避免重复 article.body = "nicetitle" + str(i) # 内容与标题一致 article.author = user # 关联同一测试用户 article.category = category # 关联同一分类 article.type = 'a' # 普通文章类型 article.status = 'p' # 已发布状态 article.save() # 保存文章 article.tags.add(tag) # 关联同一标签 article.save() # 更新关联关系 # 导入Elasticsearch启用标识（判断是否启用搜索引擎） from blog.documents import ELASTICSEARCH_ENABLED # 若启用Elasticsearch，测试搜索功能 if ELASTICSEARCH_ENABLED: call_command("build_index") # 执行自定义管理命令，重建搜索索引 # 模拟发送搜索请求，关键词为'nicetitle' response = self.client.get('/search', {'q': 'nicetitle'}) self.assertEqual(response.status_code, 200) # 验证搜索页面正常响应 # 模拟访问最后一篇测试文章的详情页，验证响应状态码为200 response = self.client.get(article.get_absolute_url()) self.assertEqual(response.status_code, 200) # 导入爬虫通知工具类，测试文章发布后通知搜索引擎（如百度） from djangoblog.spider_notify import SpiderNotify SpiderNotify.notify(article.get_absolute_url()) # 通知搜索引擎该文章URL # 模拟访问标签详情页，验证响应状态码为200 response = self.client.get(tag.get_absolute_url()) self.assertEqual(response.status_code, 200) # 模拟访问分类详情页，验证响应状态码为200 response = self.client.get(category.get_absolute_url()) self.assertEqual(response.status_code, 200) # 模拟搜索不存在的关键词'django'，验证搜索页面正常响应（状态码200） response = self.client.get('/search', {'q': 'django'}) self.assertEqual(response.status_code, 200) # 测试自定义模板标签load_articletags，验证返回结果不为空 s = load_articletags(article) self.assertIsNotNone(s) # 模拟用户登录（使用测试用户的用户名和密码） self.client.login(username='liangliangyy', password='liangliangyy') # 模拟访问文章归档页面，验证响应状态码为200 response = self.client.get(reverse('blog:archives')) self.assertEqual(response.status_code, 200) # 测试所有文章的分页功能：使用项目配置的分页数量（settings.PAGINATE_BY） p = Paginator(Article.objects.all(), settings.PAGINATE_BY) self.check_pagination(p, '', '') # 调用自定义分页测试方法 # 测试标签归档的分页功能：筛选指定标签的文章 p = Paginator(Article.objects.filter(tags=tag), settings.PAGINATE_BY) self.check_pagination(p, '分类标签归档', tag.slug) # 传递分页类型和标签slug # 测试作者归档的分页功能：筛选指定作者的文章 p = Paginator( Article.objects.filter( author__username='liangliangyy'), settings.PAGINATE_BY) self.check_pagination(p, '作者文章归档', 'liangliangyy') # 传递分页类型和作者名 # 测试分类归档的分页功能：筛选指定分类的文章 p = Paginator(Article.objects.filter(category=category), settings.PAGINATE_BY) self.check_pagination(p, '分类目录归档', category.slug) # 传递分页类型和分类slug # 测试搜索表单：初始化表单并调用search方法（验证表单逻辑无报错） f = BlogSearchForm() f.search() # 测试百度爬虫通知功能：通知单篇文章的完整URL from djangoblog.spider_notify import SpiderNotify SpiderNotify.baidu_notify([article.get_full_url()]) # 测试头像相关模板标签：验证gravatar_url和gravatar函数正常返回结果 from blog.templatetags.blog_tags import gravatar_url, gravatar u = gravatar_url('liangliangyy@gmail.com') # 生成Gravatar头像URL u = gravatar('liangliangyy@gmail.com') # 生成Gravatar头像HTML标签 # 创建测试友情链接数据 link = Links( sequence=1, # 排序序号 name="lylinux", # 链接名称 link='https://wwww.lylinux.net') # 链接URL link.save() # 保存到数据库 # 模拟访问友情链接页面，验证响应状态码为200 response = self.client.get('/links.html') self.assertEqual(response.status_code, 200) # 模拟访问RSS订阅 feed 页面，验证响应状态码为200 response = self.client.get('/feed/') self.assertEqual(response.status_code, 200) # 模拟访问站点地图页面，验证响应状态码为200 response = self.client.get('/sitemap.xml') self.assertEqual(response.status_code, 200) # 模拟访问后台文章删除页面（测试后台操作可达性） self.client.get("/admin/blog/article/1/delete/") # 模拟访问后台邮件发送日志页面（重复访问，验证稳定性） self.client.get('/admin/servermanager/emailsendlog/') # 模拟访问后台操作日志列表页面（重复访问，验证稳定性） self.client.get('/admin/admin/logentry/') # 模拟访问后台操作日志编辑页面（测试后台详情页可达性） self.client.get('/admin/admin/logentry/1/change/') # 自定义分页测试方法：验证分页逻辑和分页URL的可达性 def check_pagination(self, p, type, value): # 遍历所有分页页面（从第1页到最后一页） for page in range(1, p.num_pages + 1): # 调用模板标签load_pagination_info，获取分页信息（上一页URL、下一页URL等） s = load_pagination_info(p.page(page), type, value) self.assertIsNotNone(s) # 验证分页信息不为空 # 若存在上一页URL，模拟访问并验证响应状态码为200 if s['previous_url']: response = self.client.get(s['previous_url']) self.assertEqual(response.status_code, 200) # 若存在下一页URL，模拟访问并验证响应状态码为200 if s['next_url']: response = self.client.get(s['next_url']) self.assertEqual(response.status_code, 200) # 测试方法：测试图片上传功能（含未授权上传、授权上传验证） def test_image(self): # 导入requests库（需提前安装），用于下载测试图片 import requests # 下载Python官网logo图片，作为测试上传文件 rsp = requests.get( 'https://www.python.org/static/img/python-logo@2x.png') # 定义图片保存路径：项目根目录下的'python.png' imagepath = os.path.join(settings.BASE_DIR, 'python.png') # 将下载的图片内容写入本地文件 with open(imagepath, 'wb') as file: file.write(rsp.content) # 模拟未授权的图片上传请求（未带签名），验证响应状态码为403（禁止访问） rsp = self.client.post('/upload') self.assertEqual(rsp.status_code, 403) # 生成上传授权签名：双重SHA256加密项目SECRET_KEY（与后端上传接口的签名验证逻辑一致） sign = get_sha256(get_sha256(settings.SECRET_KEY)) # 读取本地测试图片文件，创建SimpleUploadedFile对象（模拟文件上传数据） with open(imagepath, 'rb') as file: imgfile = SimpleUploadedFile( 'python.png', file.read(), content_type='image/jpg') # 文件名、文件内容、MIME类型 form_data = {'python.png': imgfile} # 构造表单数据（键为文件名，值为文件对象） # 模拟带签名的图片上传请求，跟随重定向（follow=True） rsp = self.client.post( '/upload?sign=' + sign, form_data, follow=True) self.assertEqual(rsp.status_code, 200) # 验证授权上传成功，响应状态码为200 # 删除本地测试图片文件（清理测试残留） os.remove(imagepath) # 测试工具函数：保存用户头像、发送邮件 from djangoblog.utils import save_user_avatar, send_email send_email(['qq@qq.com'], 'testTitle', 'testContent') # 发送测试邮件（收件人、标题、内容） # 从URL保存用户头像（测试图片下载与保存逻辑） save_user_avatar( 'https://www.python.org/static/img/python-logo@2x.png') # 测试方法：测试错误页面（404页面） def test_errorpage(self): # 模拟访问不存在的URL（'/eee'），验证响应状态码为404（页面未找到） rsp = self.client.get('/eee') self.assertEqual(rsp.status_code, 404) # 测试方法：测试自定义管理命令（如重建索引、清理缓存等） def test_commands(self): # 创建/获取测试用户（与test_validate_article方法一致） user = BlogUser.objects.get_or_create( email="liangliangyy@gmail.com", username="liangliangyy")[0] user.set_password("liangliangyy") user.is_staff = True user.is_superuser = True user.save() # 创建第三方登录配置（QQ登录） c = OAuthConfig() c.type = 'qq' # 登录类型：QQ c.appkey = 'appkey' # 测试用appkey c.appsecret = 'appsecret' # 测试用appsecret c.save() # 保存到数据库 # 创建第三方登录用户（关联测试用户） u = OAuthUser() u.type = 'qq' # 登录类型：QQ u.openid = 'openid' # 测试用openid（第三方平台用户唯一标识） u.user = user # 关联本地测试用户 u.picture = static("/blog/img/avatar.png") # 静态文件中的默认头像 u.metadata = ''' { "figureurl": "https://qzapp.qlogo.cn/qzapp/101513904/C740E30B4113EAA80E0D9918ABC78E82/30" }''' # 第三方用户元数据（如QQ头像URL） u.save() # 保存到数据库 # 创建另一个第三方登录用户（不关联本地用户，测试头像URL直接赋值） u = OAuthUser() u.type = 'qq' # 登录类型：QQ u.openid = 'openid1' # 不同的openid # 直接赋值头像URL（而非静态文件） u.picture = 'https://qzapp.qlogo.cn/qzapp/101513904/C740E30B4113EAA80E0D9918ABC78E82/30' u.metadata = ''' { "figureurl": "https://qzapp.qlogo.cn/qzapp/101513904/C740E30B4113EAA80E0D9918ABC78E82/30" }''' # 第三方用户元数据 u.save() # 保存到数据库 # 导入Elasticsearch启用标识 from blog.documents import ELASTICSEARCH_ENABLED # 若启用Elasticsearch，执行重建搜索索引命令 if ELASTICSEARCH_ENABLED: call_command("build_index") # 执行自定义管理命令：通知百度爬虫（参数"all"表示通知所有文章） call_command("ping_baidu", "all") # 执行自定义管理命令：创建测试数据 call_command("create_testdata") # 执行自定义管理命令：清理缓存 call_command("clear_cache") # 执行自定义管理命令：同步用户头像（从第三方平台同步头像到本地） call_command("sync_user_avatar") # 执行自定义管理命令：构建搜索关键词（优化搜索体验） call_command("build_search_words")