ios-unittest-generator 4.19.3__tar.gz

ios_unittest_generator-4.19.3/.gitignore

@@ -0,0 +1,23 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+venv/
+env/
+ENV/
+*.egg-info/
+dist/
+build/
+
+# IDEs
+.vscode/
+.idea/
+*.swp
+*.swo
+*~
+
+# OS
+.DS_Store
+Thumbs.db

ios_unittest_generator-4.19.3/LICENSE

@@ -0,0 +1,28 @@
+BSD 3-Clause License
+
+Copyright (c) Microsoft Corporation. All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+1. Redistributions of source code must retain the above copyright notice, this
+   list of conditions and the following disclaimer.
+
+2. Redistributions in binary form must reproduce the above copyright notice,
+   this list of conditions and the following disclaimer in the documentation
+   and/or other materials provided with the distribution.
+
+3. Neither the name of the copyright holder nor the names of its
+   contributors may be used to endorse or promote products derived from
+   this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

ios_unittest_generator-4.19.3/PKG-INFO

@@ -0,0 +1,115 @@
+Metadata-Version: 2.4
+Name: ios-unittest-generator
+Version: 4.19.3
+Summary: MCP Server for generating iOS unit tests for Chromium/Edge projects
+Author: Microsoft Edge iOS Team
+License-Expression: BSD-3-Clause
+License-File: LICENSE
+Keywords: chromium,edge,ios,mcp,test-generation,unittest
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Testing
+Requires-Python: >=3.11
+Requires-Dist: mcp<2.0.0,>=1.9.4
+Description-Content-Type: text/markdown
+
+# iOS Unit Test Generator
+
+智能 MCP 服务器，自动生成、编译、运行和修复 iOS 单元测试。
+
+## 🚀 快速开始
+
+**无需拉取代码，直接在 VS Code 中配置即可使用！**
+
+### 方式一：VS Code 一键安装（最简单 ⭐）
+
+1. 打开 VS Code
+2. `Cmd+Shift+P` → 输入 `MCP: Add Server`
+3. 选择 **"Pip Package"**
+4. 输入包名：`ios-unittest-generator`
+5. 设置环境变量 `CHROMIUM_SRC` 为你的 Chromium 源码路径
+6. 完成！
+
+### 方式二：手动编辑配置文件
+
+**macOS/Linux** - 编辑 `~/.vscode/mcp.json`:
+```json
+{
+  "servers": {
+    "ios-unittest-generator": {
+      "command": "uvx",
+      "args": ["ios-unittest-generator"],
+      "env": {
+        "CHROMIUM_SRC": "/path/to/chromium/src"
+      }
+    }
+  }
+}
+```
+
+**Windows** - 编辑 `%APPDATA%\Code\User\mcp.json`:
+```json
+{
+  "servers": {
+    "ios-unittest-generator": {
+      "command": "uvx",
+      "args": ["ios-unittest-generator"],
+      "env": {
+        "CHROMIUM_SRC": "C:\\path\\to\\chromium\\src"
+      }
+    }
+  }
+}
+```
+
+## 📋 核心功能
+
+- ✅ 自动分析源文件，识别可测试元素
+- ✅ 生成完整的测试文件（包含 fixture、SetUp、测试用例）
+- ✅ 智能检测测试目标（支持 `ios/chrome/*`、`components/*/ios/*` 等所有路径）
+- ✅ 自动更新 BUILD.gn 文件（按字母顺序）
+- ✅ 自动编译测试，智能分析编译错误
+- ✅ 自动运行测试，智能分析运行时错误
+
+## 📖 11 个 MCP 工具
+
+| 工具 | 功能 |
+|------|------|
+| `full_test_workflow` | 完整工作流（分析→生成→增强→编译→运行） |
+| `analyze_ios_code_for_testing` | 分析源文件，提取可测试元素 |
+| `generate_ios_unittest_file` | 生成测试文件 |
+| `check_ios_test_coverage` | 检查测试覆盖率 |
+| `verify_test_enhancement_complete` | 验证测试增强完成（质量门控） |
+| `compile_ios_unittest` | 编译测试（自动错误分析） |
+| `run_ios_unittest` | 运行测试（自动错误分析） |
+| `analyze_runtime_errors` | 分析运行时错误 |
+| `analyze_compilation_errors` | 分析编译错误 |
+| `update_existing_tests` | 增量更新测试 |
+| `update_build_file_for_test` | 自动更新 BUILD 文件 |
+
+## 💡 使用示例
+
+```bash
+# 一键生成完整测试
+Use full_test_workflow for ios/chrome/browser/ui/foo.mm
+
+# 单独编译
+Use compile_ios_unittest for ios/chrome/browser/ui/foo.mm
+
+# 单独运行
+Use run_ios_unittest with filter FooTest.*
+```
+
+## 🔗 相关链接
+
+- **PyPI**: https://pypi.org/project/ios-unittest-generator/
+- **快速入门**: [QUICKSTART.md](QUICKSTART.md)
+
+---
+
+**版本**: v4.19.1  
+**更新日期**: 2026-01-22

ios_unittest_generator-4.19.3/README.md

@@ -0,0 +1,96 @@
+# iOS Unit Test Generator
+
+智能 MCP 服务器，自动生成、编译、运行和修复 iOS 单元测试。
+
+## 🚀 快速开始
+
+**无需拉取代码，直接在 VS Code 中配置即可使用！**
+
+### 方式一：VS Code 一键安装（最简单 ⭐）
+
+1. 打开 VS Code
+2. `Cmd+Shift+P` → 输入 `MCP: Add Server`
+3. 选择 **"Pip Package"**
+4. 输入包名：`ios-unittest-generator`
+5. 设置环境变量 `CHROMIUM_SRC` 为你的 Chromium 源码路径
+6. 完成！
+
+### 方式二：手动编辑配置文件
+
+**macOS/Linux** - 编辑 `~/.vscode/mcp.json`:
+```json
+{
+  "servers": {
+    "ios-unittest-generator": {
+      "command": "uvx",
+      "args": ["ios-unittest-generator"],
+      "env": {
+        "CHROMIUM_SRC": "/path/to/chromium/src"
+      }
+    }
+  }
+}
+```
+
+**Windows** - 编辑 `%APPDATA%\Code\User\mcp.json`:
+```json
+{
+  "servers": {
+    "ios-unittest-generator": {
+      "command": "uvx",
+      "args": ["ios-unittest-generator"],
+      "env": {
+        "CHROMIUM_SRC": "C:\\path\\to\\chromium\\src"
+      }
+    }
+  }
+}
+```
+
+## 📋 核心功能
+
+- ✅ 自动分析源文件，识别可测试元素
+- ✅ 生成完整的测试文件（包含 fixture、SetUp、测试用例）
+- ✅ 智能检测测试目标（支持 `ios/chrome/*`、`components/*/ios/*` 等所有路径）
+- ✅ 自动更新 BUILD.gn 文件（按字母顺序）
+- ✅ 自动编译测试，智能分析编译错误
+- ✅ 自动运行测试，智能分析运行时错误
+
+## 📖 11 个 MCP 工具
+
+| 工具 | 功能 |
+|------|------|
+| `full_test_workflow` | 完整工作流（分析→生成→增强→编译→运行） |
+| `analyze_ios_code_for_testing` | 分析源文件，提取可测试元素 |
+| `generate_ios_unittest_file` | 生成测试文件 |
+| `check_ios_test_coverage` | 检查测试覆盖率 |
+| `verify_test_enhancement_complete` | 验证测试增强完成（质量门控） |
+| `compile_ios_unittest` | 编译测试（自动错误分析） |
+| `run_ios_unittest` | 运行测试（自动错误分析） |
+| `analyze_runtime_errors` | 分析运行时错误 |
+| `analyze_compilation_errors` | 分析编译错误 |
+| `update_existing_tests` | 增量更新测试 |
+| `update_build_file_for_test` | 自动更新 BUILD 文件 |
+
+## 💡 使用示例
+
+```bash
+# 一键生成完整测试
+Use full_test_workflow for ios/chrome/browser/ui/foo.mm
+
+# 单独编译
+Use compile_ios_unittest for ios/chrome/browser/ui/foo.mm
+
+# 单独运行
+Use run_ios_unittest with filter FooTest.*
+```
+
+## 🔗 相关链接
+
+- **PyPI**: https://pypi.org/project/ios-unittest-generator/
+- **快速入门**: [QUICKSTART.md](QUICKSTART.md)
+
+---
+
+**版本**: v4.19.1  
+**更新日期**: 2026-01-22

ios_unittest_generator-4.19.3/pyproject.toml

@@ -0,0 +1,49 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "ios-unittest-generator"
+version = "4.19.3"
+description = "MCP Server for generating iOS unit tests for Chromium/Edge projects"
+readme = "README.md"
+license = "BSD-3-Clause"
+authors = [
+    { name = "Microsoft Edge iOS Team" }
+]
+keywords = ["mcp", "ios", "unittest", "chromium", "edge", "test-generation"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: BSD License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Software Development :: Testing",
+]
+requires-python = ">=3.11"
+dependencies = [
+    "mcp>=1.9.4,<2.0.0",
+]
+
+[project.scripts]
+ios-unittest-generator = "ios_unittest_generator:main"
+ios_unittest_generator = "ios_unittest_generator:main"
+
+[project.urls]
+# Update these URLs when publishing to your actual repository
+# Homepage = "https://github.com/your-org/ios-unittest-generator"
+# Repository = "https://github.com/your-org/ios-unittest-generator"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/ios_unittest_generator"]
+
+[tool.hatch.build.targets.wheel.force-include]
+"src/ios_unittest_generator/ios_test_patterns.json" = "ios_unittest_generator/ios_test_patterns.json"
+
+[tool.hatch.build.targets.sdist]
+include = [
+    "src/",
+    "README.md",
+    "LICENSE",
+]

ios_unittest_generator-4.19.3/src/ios_unittest_generator/__init__.py

@@ -0,0 +1,15 @@
+#!/usr/bin/env python3
+# Copyright (C) Microsoft Corporation. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""iOS Unit Test Generator MCP Server.
+
+This package provides MCP tools to generate unit tests for iOS Edge/Chromium code.
+"""
+
+__version__ = "4.19.3"
+
+from .server import mcp, main
+
+__all__ = ["mcp", "main", "__version__"]

ios_unittest_generator-4.19.3/src/ios_unittest_generator/__main__.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python3
+# Copyright (C) Microsoft Corporation. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Entry point for running the MCP server directly with `python -m ios_unittest_generator`."""
+
+from .server import main
+
+if __name__ == "__main__":
+    main()

ios_unittest_generator-4.19.3/src/ios_unittest_generator/build_file_utils.py

@@ -0,0 +1,293 @@
+#!/usr/bin/env python3
+# Copyright (C) Microsoft Corporation. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""BUILD file manipulation utilities.
+
+This module provides functions for updating BUILD.gn and BUILD_edge.gni files.
+Refactored from the monolithic _add_component_to_test_target function.
+"""
+
+import re
+import sys
+from typing import List, Optional, Tuple
+
+
+class BuildFileUpdater:
+    """Handles BUILD file updates with strategy pattern."""
+    
+    def __init__(self, build_content: str, component_target: str, target_name: str):
+        """Initialize the updater.
+        
+        Args:
+            build_content: Content of the BUILD file
+            component_target: Target path like "//components/xxx/ios:unit_tests"
+            target_name: Test target name ('ios_chrome_unittests', etc.)
+        """
+        self.build_content = build_content
+        self.component_target = component_target
+        self.target_name = target_name
+    
+    def update(self) -> Tuple[bool, str]:
+        """Update BUILD file using appropriate strategy.
+        
+        Returns:
+            (success, modified_content) tuple
+        """
+        # Try strategies in order
+        strategies = [
+            self._strategy_edge_components_template,
+            self._strategy_edge_ios_template,
+            self._strategy_direct_target,
+        ]
+        
+        for strategy in strategies:
+            success, content = strategy()
+            if success:
+                return True, content
+        
+        return False, self.build_content
+    
+    def _strategy_edge_components_template(self) -> Tuple[bool, str]:
+        """Strategy 0: edge_overlay_test_components_unittests template."""
+        if self.target_name != 'components_unittests':
+            return False, self.build_content
+        
+        if 'edge_overlay_test_components_unittests' not in self.build_content:
+            return False, self.build_content
+        
+        sys.stderr.write(f"\n[STRATEGY 0] Edge components template\n")
+        sys.stderr.flush()
+        
+        # Find template definition
+        template_match = self._find_template_definition('edge_overlay_test_components_unittests')
+        if not template_match:
+            sys.stderr.write(f"[FAIL] Template not found\n")
+            sys.stderr.flush()
+            return False, self.build_content
+        
+        template_content = template_match.group(1)
+        template_start = template_match.start(1)
+        
+        # Find if (is_ios || is_android) block
+        ios_block = self._find_ios_conditional_block(template_content)
+        if not ios_block:
+            sys.stderr.write(f"[FAIL] iOS conditional block not found\n")
+            sys.stderr.flush()
+            return False, self.build_content
+        
+        ios_content, ios_start, ios_end = ios_block
+        
+        # Find or create deps += array
+        deps_match = re.search(r'(deps\s*\+=\s*\[)(.*?)(\])', ios_content, re.DOTALL)
+        
+        if deps_match:
+            # Insert into existing deps
+            return self._insert_into_deps_array(
+                template_start + ios_start + deps_match.start(2),
+                template_start + ios_start + deps_match.end(2),
+                deps_match.group(2),
+                indent='        '
+            )
+        else:
+            # Create new deps += array
+            insert_pos = template_start + ios_start
+            new_deps = f'\n      deps += [\n        "{self.component_target}",\n      ]'
+            new_content = self.build_content[:insert_pos] + new_deps + self.build_content[insert_pos:]
+            sys.stderr.write(f"[OK] Created new deps += section\n")
+            sys.stderr.flush()
+            return True, new_content
+    
+    def _strategy_edge_ios_template(self) -> Tuple[bool, str]:
+        """Strategy 1: edge_overlay_test_ios_chrome_unittests template."""
+        if self.target_name != 'ios_chrome_unittests':
+            return False, self.build_content
+        
+        if 'edge_overlay_test_ios_chrome_unittests' not in self.build_content:
+            return False, self.build_content
+        
+        sys.stderr.write(f"\n[STRATEGY 1] Edge iOS template\n")
+        sys.stderr.flush()
+        
+        pattern = r'(edge_overlay_test_ios_chrome_unittests\s*\([^)]+\)\s*\{[^}]*?deps\s*(?:\+)?=\s*\[)(.*?)(\])'
+        match = re.search(pattern, self.build_content, re.DOTALL)
+        
+        if not match:
+            return False, self.build_content
+        
+        return self._insert_into_deps_array(
+            match.start(2),
+            match.end(2),
+            match.group(2),
+            indent='      '
+        )
+    
+    def _strategy_direct_target(self) -> Tuple[bool, str]:
+        """Strategy 2: Find target directly by name."""
+        sys.stderr.write(f"\n[STRATEGY 2] Direct target search\n")
+        sys.stderr.flush()
+        
+        # Try to find test(...) or source_set(...)
+        pattern = rf'((?:source_set|test)\s*\(\s*"{re.escape(self.target_name)}"\s*\)\s*\{{.*?deps\s*(?:\+)?=\s*\[)(.*?)(\])'
+        match = re.search(pattern, self.build_content, re.DOTALL)
+        
+        if not match:
+            sys.stderr.write(f"[FAIL] Target not found\n")
+            sys.stderr.flush()
+            return False, self.build_content
+        
+        return self._insert_into_deps_array(
+            match.start(2),
+            match.end(2),
+            match.group(2),
+            indent='    '
+        )
+    
+    def _find_template_definition(self, template_name: str) -> Optional[re.Match]:
+        """Find template definition in BUILD file."""
+        template_pattern = rf'template\s*\(\s*"{template_name}"\s*\)\s*\{{'
+        match = re.search(template_pattern, self.build_content)
+        
+        if not match:
+            return None
+        
+        # Find matching closing brace
+        start_pos = match.end()
+        brace_count = 1
+        pos = start_pos
+        
+        while pos < len(self.build_content) and brace_count > 0:
+            if self.build_content[pos] == '{':
+                brace_count += 1
+            elif self.build_content[pos] == '}':
+                brace_count -= 1
+            pos += 1
+        
+        if brace_count == 0:
+            # Create a match object for the template content
+            class TemplateMatch:
+                def __init__(self, content, start, end):
+                    self._content = content
+                    self._start = start
+                    self._end = end
+                
+                def group(self, n):
+                    if n == 1:
+                        return self._content[self._start:self._end]
+                    return None
+                
+                def start(self, n):
+                    return self._start
+                
+                def end(self, n):
+                    return self._end
+            
+            return TemplateMatch(self.build_content, start_pos, pos - 1)
+        
+        return None
+    
+    def _find_ios_conditional_block(self, content: str) -> Optional[Tuple[str, int, int]]:
+        """Find if (is_ios || is_android) or if (is_ios) block."""
+        # Try is_ios || is_android first
+        markers = [
+            'if (is_ios || is_android) {',
+            'if (is_ios) {',
+        ]
+        
+        for marker in markers:
+            pos = content.find(marker)
+            if pos != -1:
+                block_start = pos + len(marker)
+                brace_count = 1
+                close_pos = block_start
+                
+                while close_pos < len(content) and brace_count > 0:
+                    if content[close_pos] == '{':
+                        brace_count += 1
+                    elif content[close_pos] == '}':
+                        brace_count -= 1
+                    close_pos += 1
+                
+                if brace_count == 0:
+                    block_content = content[block_start:close_pos - 1]
+                    return (block_content, block_start, close_pos - 1)
+        
+        return None
+    
+    def _insert_into_deps_array(
+        self,
+        deps_start: int,
+        deps_end: int,
+        deps_content: str,
+        indent: str
+    ) -> Tuple[bool, str]:
+        """Insert component target into deps array alphabetically.
+        
+        Args:
+            deps_start: Start position of deps content
+            deps_end: End position of deps content
+            deps_content: Current deps content
+            indent: Indentation to use for entries
+            
+        Returns:
+            (success, modified_content) tuple
+        """
+        # Parse existing deps
+        deps_lines = deps_content.split('\n')
+        deps_entries = []
+        for line in deps_lines:
+            stripped = line.strip()
+            if stripped and (stripped.startswith('"//') or stripped.startswith('"-=')):
+                deps_entries.append(stripped)
+        
+        # Check if already exists
+        new_entry = f'"{self.component_target}",'
+        if new_entry.strip() in ' '.join(deps_entries):
+            sys.stderr.write(f"[INFO] Component already in deps\n")
+            sys.stderr.flush()
+            return False, self.build_content
+        
+        # Find insertion point (alphabetical)
+        insertion_idx = len(deps_entries)
+        for i, entry in enumerate(deps_entries):
+            if '-=' in entry:
+                continue
+            if entry > new_entry:
+                insertion_idx = i
+                break
+            insertion_idx = i + 1
+        
+        # Insert new entry
+        deps_entries.insert(insertion_idx, new_entry)
+        
+        # Reconstruct deps
+        new_deps_lines = [f'{indent}{entry}' for entry in deps_entries]
+        new_deps_content = '\n' + '\n'.join(new_deps_lines) + f'\n{indent[:-2]}'
+        
+        # Replace in content
+        new_content = self.build_content[:deps_start] + new_deps_content + self.build_content[deps_end:]
+        sys.stderr.write(f"[OK] Component added to deps\n")
+        sys.stderr.flush()
+        return True, new_content
+
+
+def add_component_to_test_target(
+    build_content: str,
+    component_target: str,
+    target_name: str
+) -> Tuple[bool, str]:
+    """Add component target to the appropriate test target's deps.
+    
+    Refactored version using strategy pattern.
+    
+    Args:
+        build_content: Content of BUILD file
+        component_target: Target path like "//components/xxx/ios:unit_tests"
+        target_name: Test target name ('ios_chrome_unittests', etc.)
+        
+    Returns:
+        (success, modified_content) tuple
+    """
+    updater = BuildFileUpdater(build_content, component_target, target_name)
+    return updater.update()