# -------------------------------------------------------------------------
# This file is part of the MindStudio project.
# Copyright (c) 2025-2026 Huawei Technologies Co.,Ltd.
#
# MindStudio is licensed under Mulan PSL v2.
# You can use this software according to the terms and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#
#          `http://license.coscl.org.cn/MulanPSL2`
#
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
# MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
# See the Mulan PSL v2 for more details.
# -------------------------------------------------------------------------

import os
import sys
import tempfile
import shutil
import unittest
from unittest.mock import patch, MagicMock, mock_open, call, ANY
from pathlib import Path

# 更新导入路径以匹配您的实际模块位置
from msserviceprofiler.modelevalstate.patch.patch_manager import check_flag, add_patch, Patch2rc1


class TestPatchManager(unittest.TestCase):

    def setUp(self):
        # 创建临时目录
        self.temp_dir = Path(tempfile.mkdtemp())

        # 创建模拟的 mindie_llm 模块
        self.mock_mindie_llm = MagicMock()
        sys.modules['mindie_llm'] = self.mock_mindie_llm

    def tearDown(self):
        # 安全地删除整个临时目录
        shutil.rmtree(self.temp_dir, ignore_errors=True)
        # 清理模拟模块
        if 'mindie_llm' in sys.modules:
            del sys.modules['mindie_llm']

    # 重构文件检查测试 - 使用真实文件
    def test_check_flag_no_match(self):
        """测试完全不匹配的情况"""
        # 创建真实文件
        target_file = self.temp_dir / "target.txt"
        patch_file = self.temp_dir / "patch.txt"

        with open(target_file, "w", encoding="utf-8") as f:
            f.write("line1\nline2\n")

        with open(patch_file, "w", encoding="utf-8") as f:
            f.write("patch1\npatch2\n")

        result = check_flag(str(target_file), str(patch_file))
        self.assertTrue(result, "内容完全不匹配时应返回True")

    def test_check_flag_partial_match(self):
        """测试部分匹配的情况"""
        # 创建真实文件
        target_file = self.temp_dir / "target.txt"
        patch_file = self.temp_dir / "patch.txt"

        # 目标文件包含部分补丁内容
        with open(target_file, "w", encoding="utf-8") as f:
            f.write("start\n")
            f.write("patch_line1: some code 123\n")  # 补丁的第一行
            f.write("middle\n")
            # 缺少第二行补丁内容
            f.write("end\n")

        with open(patch_file, "w", encoding="utf-8") as f:
            f.write("patch_line1: some code 123\n")
            f.write("patch_line2: more code 456\n")

        result = check_flag(str(target_file), str(patch_file))
        self.assertTrue(result, "部分匹配时应返回True")

    def test_check_flag_full_match(self):
        """测试完全匹配的情况"""
        # 创建真实文件
        target_file = self.temp_dir / "target.txt"
        patch_file = self.temp_dir / "patch.txt"

        with open(target_file, "w", encoding="utf-8") as f:
            f.write("patch1\npatch2\nfooter\n")

        with open(patch_file, "w", encoding="utf-8") as f:
            f.write("patch1\npatch2\n")

        result = check_flag(str(target_file), str(patch_file))
        self.assertFalse(result, "补丁内容在目标文件中完全匹配时，预期返回False(表示已存在)但实际返回了True")

    # 测试版本检查
    def test_check_version_low_warning(self):
        """测试版本过低的情况"""
        with patch('msserviceprofiler.modelevalstate.patch.patch_manager.logger.warning') as mock_warning:
            Patch2rc1.check_version("1.0")
            mock_warning.assert_called_once_with("The version may not match.")

    def test_check_version_high_warning(self):
        """测试版本过高的情况"""
        with patch('msserviceprofiler.modelevalstate.patch.patch_manager.logger.warning') as mock_warning:
            Patch2rc1.check_version("3.0")
            mock_warning.assert_called_once_with("The version may not match.")

    def test_check_version_exactly_low(self):
        """测试精确匹配下限版本的情况"""
        with patch('msserviceprofiler.modelevalstate.patch.patch_manager.logger.warning') as mock_warning:
            Patch2rc1.check_version(Patch2rc1.mindie_llm_low)
            mock_warning.assert_called_once_with("The version may not match.")

    def test_check_version_exactly_up(self):
        """测试精确匹配上限版本的情况"""
        with patch('msserviceprofiler.modelevalstate.patch.patch_manager.logger.warning') as mock_warning:
            Patch2rc1.check_version(Patch2rc1.mindie_llm)
            mock_warning.assert_not_called()

    def test_check_version_patch_level(self):
        """测试补丁版本比较"""
        with patch('msserviceprofiler.modelevalstate.patch.patch_manager.logger.warning') as mock_warning:
            # 注意：2.0.1 > 2.0，所以应该在范围外
            Patch2rc1.check_version("2.0.1")
            mock_warning.assert_called_once_with("The version may not match.")


import tempfile
import shutil
from pathlib import Path
from unittest.mock import patch, MagicMock
import pytest

from msserviceprofiler.modelevalstate.patch.patch_manager import check_flag, add_patch, Patch2rc1


@pytest.fixture
def temp_dir():
    """创建临时目录的fixture"""
    temp_dir = Path(tempfile.mkdtemp())
    yield temp_dir
    shutil.rmtree(temp_dir, ignore_errors=True)


class TestCheckFlagPytest:
    """使用pytest测试check_flag函数"""

    def test_check_flag_same_content(self, temp_dir):
        """测试相同内容的情况"""
        target_file = temp_dir / "target.txt"
        patch_file = temp_dir / "patch.txt"

        content = "test content\n"
        target_file.write_text(content, encoding="utf-8")
        patch_file.write_text(content, encoding="utf-8")

        result = check_flag(str(target_file), str(patch_file))
        assert result is False

    def test_check_flag_different_content(self, temp_dir):
        """测试不同内容的情况"""
        target_file = temp_dir / "target.txt"
        patch_file = temp_dir / "patch.txt"

        target_file.write_text("target content\n", encoding="utf-8")
        patch_file.write_text("patch content\n", encoding="utf-8")

        result = check_flag(str(target_file), str(patch_file))
        assert result is True

    def test_check_flag_partial_content_match(self, temp_dir):
        """测试部分内容匹配的情况"""
        target_file = temp_dir / "target.txt"
        patch_file = temp_dir / "patch.txt"

        target_file.write_text("line1\npatch1\nline3\n", encoding="utf-8")
        patch_file.write_text("patch1\npatch2\n", encoding="utf-8")

        result = check_flag(str(target_file), str(patch_file))
        assert result is True


class TestAddPatchPytest:
    """使用pytest测试add_patch函数"""

    def test_add_patch_normal_case(self, temp_dir):
        """测试正常添加补丁的情况"""
        target_file = temp_dir / "target.txt"
        patch_file = temp_dir / "patch.txt"

        target_file.write_text("original\n", encoding="utf-8")
        patch_file.write_text("patch\n", encoding="utf-8")

        add_patch(str(target_file), str(patch_file))

        result_content = target_file.read_text(encoding="utf-8")
        assert "original" in result_content
        assert "patch" in result_content

    def test_add_patch_empty_file(self, temp_dir):
        """测试空文件的情况"""
        target_file = temp_dir / "target.txt"
        patch_file = temp_dir / "patch.txt"

        # 创建空文件
        target_file.touch()
        patch_file.touch()

        add_patch(str(target_file), str(patch_file))

        # 文件应该仍然为空
        assert target_file.read_text(encoding="utf-8") == ""


class TestPatch2rc1Pytest:
    """使用pytest测试Patch2rc1类"""

    @pytest.mark.parametrize("version_str,should_warn", [
        ("2.0", True),      # 低于下限
        ("2.1rc1", True),   # 等于下限（应该警告）
        ("2.1.5", False),   # 在范围内
        ("2.2", False),     # 等于上限
        ("2.3", True),      # 高于上限
        ("2.1.5.dev1", False),  # 开发版本
    ])
    def test_check_version_warnings(self, version_str, should_warn):
        """测试版本检查的警告行为"""
        with patch('msserviceprofiler.modelevalstate.patch.patch_manager.logger.warning') as mock_warning:
            result = Patch2rc1.check_version(version_str)
            assert result is True
            
            if should_warn:
                mock_warning.assert_called_once_with("The version may not match.")
            else:
                mock_warning.assert_not_called()


# 性能测试
@pytest.mark.performance
class TestPerformance:
    """性能测试类"""

    def test_check_flag_performance(self, temp_dir):
        """测试check_flag的性能"""
        target_file = temp_dir / "target.txt"
        patch_file = temp_dir / "patch.txt"
        
        # 创建大文件
        large_content = "line\n" * 10000
        target_file.write_text(large_content, encoding="utf-8")
        patch_file.write_text("line\n", encoding="utf-8")
        
        # 测试性能（这里只是基本测试，实际应该使用timeit）
        result = check_flag(str(target_file), str(patch_file))
        assert isinstance(result, bool)