from django.test import Client, RequestFactory, TestCase
from django.utils import timezone  # 用于处理时间相关的测试

# 导入 WeRoBot 消息类，用于模拟微信消息
from werobot.messages.messages import TextMessage

# 导入项目中的模型和 API
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   # 微信机器人命令函数


class ServerManagerTest(TestCase):
    """
    Django 测试用例类，用于对 servermanager 应用的核心功能进行单元测试。
    测试内容包括：
    - ChatGPT 聊天功能
    - 微信机器人命令响应（搜索、分类、最新文章）
    - 服务器命令执行
    - 管理员会话流程
    """

    def setUp(self):
        """
        在每个测试方法执行前自动运行的初始化方法。
        设置测试所需的公共环境：
        - 创建测试客户端（用于模拟 HTTP 请求）
        - 创建请求工厂（用于构造请求对象）

        注意：虽然此处创建了 factory，但在当前测试中并未实际使用。
        """
        self.client = Client()           # Django 测试客户端
        self.factory = RequestFactory()  # 请求工厂，用于创建模拟请求

    def test_chat_gpt(self):
        """
        测试 ChatGPT 聊天功能是否正常工作。

        步骤：
        1. 调用 ChatGPT.chat 方法发送问候语 "你好"
        2. 断言返回内容不为 None，表示接口有响应

        目的：验证聊天接口的基本可用性。
        """
        content = ChatGPT.chat("你好")
        self.assertIsNotNone(content)

    def test_validate_comment(self):
        """
        综合测试方法，覆盖多个功能点。
        名称 'validate_comment' 不准确，实际测试的是 servermanager 的核心功能。

        测试流程：
        1. 创建管理员用户并登录
        2. 创建分类和文章用于测试搜索功能
        3. 测试微信机器人的 search、category、recents 命令
        4. 测试命令执行（CommandHandler）
        5. 模拟完整管理员会话流程（登录、执行命令、退出等）
        """

        # 1. 创建超级用户
        user = BlogUser.objects.create_superuser(
            email="liangliangyy1@gmail.com",
            username="liangliangyy1",
            password="liangliangyy1")

        # 使用测试客户端登录该用户
        self.client.login(username='liangliangyy1', password='liangliangyy1')

        # 2. 创建博客分类
        c = Category()
        c.name = "categoryccc"
        c.save()

        # 3. 创建一篇已发布的文章
        article = Article()
        article.title = "nicetitleccc"
        article.body = "nicecontentccc"
        article.author = user
        article.category = c
        article.type = 'a'  # 文章类型
        article.status = 'p'  # 发布状态
        article.save()

        # 4. 模拟微信文本消息
        s = TextMessage([])
        s.content = "nice"  # 搜索关键词

        # 5. 测试 search 命令
        rsp = search(s, None)  # 调用 search 过滤器
        # 断言有响应（即使未找到文章也应返回提示）

        # 6. 测试 category 命令
        rsp = category(None, None)
        self.assertIsNotNone(rsp)  # 确保返回了分类列表

        # 7. 测试 recents 命令
        rsp = recents(None, None)
        # 断言返回结果不是“暂时还没有文章”，说明能获取到文章
        self.assertTrue(rsp != '暂时还没有文章')

        # 8. 测试命令执行功能
        cmd = commands()
        cmd.title = "test"
        cmd.command = "ls"       # 测试命令（列出目录）
        cmd.describe = "test"
        cmd.save()               # 保存到数据库

        # 实例化命令处理器
        cmdhandler = CommandHandler()
        # 执行名为 'test' 的命令
        rsp = cmdhandler.run('test')
        # 断言命令执行有返回结果
        self.assertIsNotNone(rsp)

        # 9. 模拟管理员会话流程
        s.source = 'u'  # 设置用户标识（OpenID）
        s.content = 'test'  # 用户输入内容

        # 创建消息处理器实例，传入模拟消息和空会话
        msghandler = MessageHandler(s, {})

        # 模拟用户行为序列：
        msghandler.handler()  # 处理 "test" 消息（应提示确认）

        s.content = 'y'  # 用户确认执行
        msghandler.handler()  # 执行命令

        s.content = 'idcard:12321233'  # 尝试身份证查询（占位功能）
        msghandler.handler()

        s.content = 'weather:上海'  # 尝试天气查询（占位功能）
        msghandler.handler()

        s.content = 'admin'  # 请求进入管理员模式
        msghandler.handler()

        s.content = '123'  # 输入密码（测试环境下有效）
        msghandler.handler()

        s.content = 'exit'  # 退出管理员模式
        msghandler.handler()

        # 注意：此测试未包含 assert 断言来验证每个步骤的结果，
        # 主要是通过调用来检查是否抛出异常，确保代码路径可执行。