|
|
# 马莹：导入Django测试相关模块
|
|
|
from django.test import Client, RequestFactory, TestCase
|
|
|
from django.utils import timezone
|
|
|
# 马莹：导入微信机器人文本消息类
|
|
|
from werobot.messages.messages import TextMessage
|
|
|
|
|
|
# 马莹：导入项目内相关模型和工具类
|
|
|
from accounts.models import BlogUser
|
|
|
from blog.models import Category, Article
|
|
|
from servermanager.api.commonapi import ChatGPT
|
|
|
from .models import commands
|
|
|
from .robot import MessageHandler, CommandHandler
|
|
|
from .robot import search, category, recents
|
|
|
|
|
|
|
|
|
# Create your tests here.
|
|
|
# 马莹：定义服务器管理模块的测试类，继承自Django的TestCase
|
|
|
class ServerManagerTest(TestCase):
|
|
|
def setUp(self):
|
|
|
"""测试初始化方法，在每个测试方法执行前运行"""
|
|
|
# 马莹：创建测试客户端，用于模拟HTTP请求
|
|
|
self.client = Client()
|
|
|
# 马莹：创建请求工厂，用于构造测试用的请求对象
|
|
|
self.factory = RequestFactory()
|
|
|
|
|
|
def test_chat_gpt(self):
|
|
|
"""测试ChatGPT聊天功能"""
|
|
|
# 马莹：调用ChatGPT的chat方法发送"你好"消息
|
|
|
content = ChatGPT.chat("你好")
|
|
|
# 马莹：断言返回结果不为空
|
|
|
self.assertIsNotNone(content)
|
|
|
|
|
|
def test_validate_comment(self):
|
|
|
"""综合测试各类功能：文章搜索、分类、最新文章、命令执行、消息处理等"""
|
|
|
# 马莹：创建超级用户
|
|
|
user = BlogUser.objects.create_superuser(
|
|
|
email="liangliangyy1@gmail.com",
|
|
|
username="liangliangyy1",
|
|
|
password="liangliangyy1")
|
|
|
|
|
|
# 马莹：使用测试客户端登录该超级用户
|
|
|
self.client.login(username='liangliangyy1', password='liangliangyy1')
|
|
|
|
|
|
# 马莹：创建测试分类并保存到数据库
|
|
|
c = Category()
|
|
|
c.name = "categoryccc"
|
|
|
c.save()
|
|
|
|
|
|
# 马莹：创建测试文章并保存到数据库
|
|
|
article = Article()
|
|
|
article.title = "nicetitleccc"
|
|
|
article.body = "nicecontentccc"
|
|
|
article.author = user # 关联作者
|
|
|
article.category = c # 关联分类
|
|
|
article.type = 'a' # 文章类型
|
|
|
article.status = 'p' # 发布状态
|
|
|
article.save()
|
|
|
|
|
|
# 马莹：构造文本消息对象，内容为"nice"
|
|
|
s = TextMessage([])
|
|
|
s.content = "nice"
|
|
|
# 马莹：测试文章搜索功能
|
|
|
rsp = search(s, None)
|
|
|
# 马莹：测试获取分类功能
|
|
|
rsp = category(None, None)
|
|
|
# 马莹：断言分类功能返回结果不为空
|
|
|
self.assertIsNotNone(rsp)
|
|
|
# 马莹：测试获取最新文章功能
|
|
|
rsp = recents(None, None)
|
|
|
# 马莹：断言最新文章功能返回结果不是"暂时还没有文章"
|
|
|
self.assertTrue(rsp != '暂时还没有文章')
|
|
|
|
|
|
# 马莹：创建测试命令并保存到数据库
|
|
|
cmd = commands()
|
|
|
cmd.title = "test"
|
|
|
cmd.command = "ls"
|
|
|
cmd.describe = "test"
|
|
|
cmd.save()
|
|
|
|
|
|
# 马莹：实例化命令处理器
|
|
|
cmdhandler = CommandHandler()
|
|
|
# 马莹：测试执行命令功能
|
|
|
rsp = cmdhandler.run('test')
|
|
|
# 马莹：断言命令执行结果不为空
|
|
|
self.assertIsNotNone(rsp)
|
|
|
|
|
|
# 马莹：设置消息的发送者ID
|
|
|
s.source = 'u'
|
|
|
# 马莹：设置消息内容为'test'
|
|
|
s.content = 'test'
|
|
|
# 马莹：实例化消息处理器
|
|
|
msghandler = MessageHandler(s, {})
|
|
|
|
|
|
# 马莹：以下注释代码为管理员权限相关配置（当前测试未启用）
|
|
|
# msghandler.userinfo.isPasswordSet = True
|
|
|
# msghandler.userinfo.isAdmin = True
|
|
|
|
|
|
# 马莹：处理消息内容'test'
|
|
|
msghandler.handler()
|
|
|
# 马莹：修改消息内容为'y'（模拟确认执行命令）
|
|
|
s.content = 'y'
|
|
|
msghandler.handler()
|
|
|
# 马莹：修改消息内容为身份证查询指令（建设中功能）
|
|
|
s.content = 'idcard:12321233'
|
|
|
msghandler.handler()
|
|
|
# 马莹：修改消息内容为天气查询指令（建设中功能）
|
|
|
s.content = 'weather:上海'
|
|
|
msghandler.handler()
|
|
|
# 马莹：修改消息内容为'admin'（进入管理员模式）
|
|
|
s.content = 'admin'
|
|
|
msghandler.handler()
|
|
|
# 马莹：修改消息内容为'123'（输入管理员密码）
|
|
|
s.content = '123'
|
|
|
msghandler.handler()
|
|
|
# 马莹：修改消息内容为'exit'（退出管理员模式）
|
|
|
s.content = 'exit'
|
|
|
msghandler.handler() |
