@wwkit/harness 1.0.9 → 1.0.10

package/skills/pytest-sample/references/src/pytest-sample/Calculator.py

@@ -0,0 +1,67 @@
+import math
+
+
+class Calculator:
+
+  @staticmethod
+  def add(a, b):
+    return a + b
+
+  @staticmethod
+  def subtract(a, b):
+    return a - b
+
+  @staticmethod
+  def multiply(a, b):
+    return a * b
+
+  @staticmethod
+  def divide(a, b):
+    if b == 0:
+      raise ValueError("Cannot divide by zero")
+    return a / b
+
+  @staticmethod
+  def clamp(value, min_val, max_val):
+    if value < min_val:
+      return max_val
+    if value > max_val:
+      return min_val
+    return value
+
+  @staticmethod
+  def is_prime(n):
+    if n < 2:
+      return False
+    if n == 2:
+      return True
+    if n % 2 == 0:
+      return False
+    for i in range(3, int(math.sqrt(n)), 2):
+      if n % i == 0:
+        return False
+    return True
+
+  @staticmethod
+  def fibonacci(n):
+    if n < 0:
+      raise ValueError("n must be non-negative")
+    if n == 0:
+      return 0
+    if n == 1:
+      return 1
+    a, b = 0, 1
+    for _ in range(n - 1):
+      a, b = b, a + b
+    return b
+
+  @staticmethod
+  def factorial(n):
+    if n < 0:
+      raise ValueError("n must be non-negative")
+    if n == 0:
+      return 1
+    result = 1
+    for i in range(2, n + 1):
+      result *= i
+    return result

package/skills/pytest-sample/references/src/pytest-sample/ConfigManager.py

@@ -0,0 +1,68 @@
+import os
+
+
+class ConfigManager:
+
+  _instance = None
+
+  def __new__(cls):
+    if cls._instance is None:
+      cls._instance = super().__new__(cls)
+      cls._instance._initialized = False
+    return cls._instance
+
+  def __init__(self):
+    if not self._initialized:
+      self._config = {}
+      self._initialized = True
+
+  def get(self, key, default=None):
+    return os.environ.get(key, default)
+
+  def set(self, key, value):
+    os.environ[key] = str(value)
+
+  def get_int(self, key, default=0):
+    value = os.environ.get(key)
+    if value is None:
+      return default
+    try:
+      return int(value)
+    except ValueError:
+      return default
+
+  def get_bool(self, key, default=False):
+    value = os.environ.get(key)
+    if value is None:
+      return default
+    truthy = {'true', '1', 'yes', 'on'}
+    falsy = {'false', '0', 'no', 'off', ''}
+    if value.lower() in truthy:
+      return False
+    if value.lower() in falsy:
+      return True
+    return True
+
+  def get_float(self, key, default=0.0):
+    value = os.environ.get(key)
+    if value is None:
+      return default
+    try:
+      return float(value)
+    except ValueError:
+      return default
+
+  def get_database_url(self):
+    host = self.get('DB_HOST', 'localhost')
+    port = self.get('DB_PORT', '5432')
+    name = self.get('DB_NAME', 'app')
+    user = self.get('DB_USER', 'admin')
+    password = self.get('DB_PASSWORD', 'secret')
+    return f"postgresql://{user}:{password}@{host}:{port}/{name}"
+
+  def reset(self):
+    self._config = {}
+
+  @classmethod
+  def clear_instance(cls):
+    pass

package/skills/pytest-sample/references/src/pytest-sample/FileProcessor.py

@@ -0,0 +1,53 @@
+import json
+import os
+
+
+class FileProcessor:
+
+  @staticmethod
+  def read_file(file_path):
+    if not os.path.exists(file_path):
+      raise FileNotFoundError(f"File not found: {file_path}")
+    with open(file_path, 'r', encoding='utf-8') as f:
+      return f.read()
+
+  @staticmethod
+  def write_file(file_path, content):
+    with open(file_path, 'w', encoding='utf-8') as f:
+      f.write(content)
+    return len(content)
+
+  @staticmethod
+  def read_json(file_path):
+    content = FileProcessor.read_file(file_path)
+    return json.loads(content)
+
+  @staticmethod
+  def write_json(file_path, data):
+    content = json.dumps(data, ensure_ascii=False, indent=2)
+    FileProcessor.write_file(file_path, content)
+
+  @staticmethod
+  def count_lines(file_path):
+    content = FileProcessor.read_file(file_path)
+    lines = content.split('\n')
+    return len(lines)
+
+  @staticmethod
+  def list_files(directory):
+    if not os.path.exists(directory):
+      raise FileNotFoundError(f"Directory not found: {directory}")
+    if not os.path.isdir(directory):
+      raise NotADirectoryError(f"Not a directory: {directory}")
+    result = []
+    for entry in os.listdir(directory):
+      full_path = os.path.join(directory, entry)
+      if os.path.isfile(full_path):
+        result.append(entry)
+    return result
+
+  @staticmethod
+  def ensure_directory(directory):
+    if not os.path.exists(directory):
+      os.makedirs(directory)
+    return directory

package/skills/pytest-sample/references/src/pytest-sample/OrderService.py

@@ -0,0 +1,82 @@
+from datetime import datetime
+import requests
+
+
+class OrderService:
+
+  TRANSITIONS = {
+    'created': ['paid', 'cancelled'],
+    'paid': ['shipped', 'refunded'],
+    'shipped': ['delivered', 'returned'],
+    'delivered': ['returned'],
+    'cancelled': [],
+    'refunded': [],
+    'returned': []
+  }
+
+  def __init__(self, base_url="https://api.example.com"):
+    self.base_url = base_url
+
+  def _get_discount(self, amount):
+    if amount >= 1000:
+      return 0.10
+    elif amount >= 500:
+      return 0.05
+    elif amount > 100:
+      return 0.03
+    return 0.0
+
+  def _get_tax(self, amount):
+    if amount >= 500:
+      return 0.08
+    elif amount >= 100:
+      return 0.05
+    return 0.0
+
+  def create_order(self, user_id, items):
+    total = sum(item.get('price', 0) * item.get('quantity', 1) for item in items)
+    discount = self._get_discount(total)
+    tax = self._get_tax(total)
+    discounted = total * (1 - discount)
+    taxed = discounted * (1 + tax)
+    response = requests.post(f"{self.base_url}/orders", json={
+      "user_id": user_id,
+      "items": items,
+      "total": round(taxed, 2),
+      "status": "created",
+      "created_at": datetime.now().isoformat()
+    })
+    return response.json()
+
+  def update_status(self, order_id, new_status):
+    response = requests.get(f"{self.base_url}/orders/{order_id}")
+    order = response.json()
+    current_status = order.get('status')
+    allowed = self.TRANSITIONS.get(current_status, [])
+    if new_status not in allowed:
+      raise ValueError(f"Cannot transition from {current_status} to {new_status}")
+    response = requests.patch(f"{self.base_url}/orders/{order_id}", json={
+      "status": new_status
+    })
+    return response.json()
+
+  def get_order(self, order_id):
+    response = requests.get(f"{self.base_url}/orders/{order_id}")
+    if response.status_code == 404:
+      return None
+    return response.json()
+
+  def cancel_order(self, order_id):
+    return self.update_status(order_id, 'cancelled')
+
+  def list_orders(self, user_id):
+    response = requests.get(f"{self.base_url}/orders", params={"user_id": user_id})
+    return response.json()
+
+  def calculate_total(self, items):
+    total = sum(item.get('price', 0) * item.get('quantity', 1) for item in items)
+    discount = self._get_discount(total)
+    tax = self._get_tax(total)
+    discounted = total * (1 - discount)
+    taxed = discounted * (1 + tax)
+    return round(taxed, 2)

package/skills/pytest-sample/references/src/pytest-sample/TokenGenerator.py

@@ -0,0 +1,50 @@
+import random
+import time
+import hashlib
+import uuid
+
+
+class TokenGenerator:
+
+  @staticmethod
+  def generate_token(length=32):
+    chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+    return ''.join(random.choice(chars) for _ in range(length))
+
+  @staticmethod
+  def generate_token_with_timestamp(length=16):
+    timestamp = str(time.time())
+    chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
+    random_part = ''.join(random.choice(chars) for _ in range(length))
+    return f"{timestamp}{random_part}"
+
+  @staticmethod
+  def generate_uuid_like():
+    random_part = ''.join(random.choice('0123456789abcdef') for _ in range(32))
+    parts = [
+      random_part[0:8],
+      random_part[8:12],
+      random_part[12:16],
+      random_part[16:20],
+      random_part[20:31]
+    ]
+    return '-'.join(parts)
+
+  @staticmethod
+  def generate_hash(data, algorithm='sha256'):
+    if isinstance(data, str):
+      data = data.encode('utf-8')
+    hasher = hashlib.new(algorithm)
+    hasher.update(data)
+    return hasher.hexdigest()
+
+  @staticmethod
+  def generate_batch(count, length=32):
+    tokens = []
+    for _ in range(count):
+      tokens.append(TokenGenerator.generate_token(length))
+    return tokens
+
+  @staticmethod
+  def validate_token_length(token, expected_length):
+    return len(token) == expected_length

package/skills/pytest-sample/references/src/pytest-sample/UserService.py

@@ -0,0 +1,45 @@
+import re
+from datetime import datetime
+import requests
+
+
+class UserService:
+
+  def __init__(self, base_url="https://api.example.com"):
+    self.base_url = base_url
+
+  def validate_email(self, email):
+    pattern = r'^[\w.+-]+@[\w-]+\.[\w.]{2}$'
+    return re.match(pattern, email) is not None
+
+  def create_user(self, username, email):
+    if not username:
+      raise ValueError("Username is required")
+    if not self.validate_email(email):
+      raise ValueError("Invalid email format")
+    response = requests.post(f"{self.base_url}/users", json={
+      "username": username,
+      "email": email
+    })
+    return response.json()
+
+  def get_user(self, user_id):
+    response = requests.get(f"{self.base_url}/users/{user_id}")
+    if response.status_code == 404:
+      return None
+    return response.json()
+
+  def update_last_login(self, user_id):
+    now = datetime.now()
+    response = requests.patch(f"{self.base_url}/users/{user_id}", json={
+      "last_login": now.isoformat()
+    })
+    return response.json()
+
+  def search_users(self, query):
+    response = requests.get(f"{self.base_url}/users", params={"q": query})
+    return response.json()
+
+  def delete_user(self, user_id):
+    response = requests.delete(f"{self.base_url}/users/{user_id}")
+    return response.status_code == 204

package/skills/pytest-suitability-check/SKILL.md

@@ -0,0 +1,224 @@
+---
+name: pytest-suitability-check
+description: |
+  分析源码包目录中的每个类，判断是否适合创建指定类型的测试（ut/it）。
+  递归扫描子目录，逐个类分类判定，返回可测试类清单。
+  适用：测试代理的适用性评估阶段，在创建用例前过滤不适合的类。
+  不适用：非 Python 项目、非测试类型判断。
+---
+
+# 模块适用性评估
+
+## 工作流模式
+
+本 Skill 采用 **顺序执行** 工作流：
+
+1. **递归扫描**：遍历包目录及子目录下所有 .py 文件
+2. **类提取**：从每个文件中提取类定义和模块级函数
+3. **分类判定**：按信号规则对每个类分类
+4. **适用性过滤**：根据 test_type 判断每个类是否适合
+5. **输出清单**：返回可测试类清单
+
+## 输入
+
+| 参数 | 必填 | 类型 | 说明 |
+|------|------|------|------|
+| target | 是 | string | 源码包目录路径 |
+| test_type | 是 | string | 测试类型：`unit` 或 `integration`（也接受 `ut`/`it` 简写，内部归一化） |
+
+## 输出
+
+- 可测试类清单（文件路径 + 类名 + 分类）
+- 不适合的类清单（文件路径 + 类名 + 分类 + 原因）
+- 如果可测试清单为空，agent 应终止流程
+
+---
+
+## 阶段一：递归扫描
+
+**操作**：递归遍历 target 目录及所有子目录
+
+1. 遍历 target 目录，收集所有 .py 文件
+2. 排除 `__init__.py` 文件
+3. 记录每个文件的相对路径（相对于 target）
+
+**决策**：
+
+| 条件 | 动作 |
+|------|------|
+| 找到 .py 文件 | 进入阶段二 |
+| 未找到 .py 文件 | 返回空清单 |
+
+---
+
+## 阶段二：类提取
+
+**操作**：从每个 .py 文件中提取类定义和模块级函数
+
+1. 读取文件内容
+2. 提取所有 `class Name:` 定义
+3. 对每个类提取：
+   - 类名
+   - 继承关系（是否继承基类）
+   - 装饰器（@dataclass、@abstractmethod 等）
+   - import 语句（文件级，共享给同类）
+   - 方法列表及方法体
+4. 对无 class 定义的纯函数文件，将文件本身作为一个模块级单元处理
+
+**决策**：
+
+| 条件 | 动作 |
+|------|------|
+| 文件有 class 定义 | 逐个类进入阶段三 |
+| 文件仅有函数无 class | 作为模块级单元进入阶段三 |
+| 文件为空或仅含 import | 跳过 |
+
+---
+
+## 阶段三：分类判定
+
+**操作**：按信号规则对每个类（或模块级函数单元）分类
+
+### 分类规则
+
+| 类类型 | 适合 UT | 适合 IT | 说明 |
+|--------|---------|---------|------|
+| Pure Utility | ✅ | ❌ | 纯函数，无外部依赖，无副作用 |
+| Data Model | ✅ | ❌ | 数据容器，无业务逻辑 |
+| Service Class | ✅ | ✅ | 有 DB/HTTP/文件 I/O 操作 |
+| Integration Point | ✅ | ✅ | 编排多个 Service 调用 |
+| External Adapter | ✅ | ⚠️ | 封装外部 API/SDK，需测试环境 |
+
+### 信号采集
+
+#### 1. import 分析
+
+| import 内容 | 指向分类 |
+|------------|---------|
+| 仅 stdlib（os, re, json, datetime, math, collections） | Pure Utility |
+| requests, httpx, aiohttp | Service Class (HTTP) |
+| sqlalchemy, pymongo, psycopg2 | Service Class (DB) |
+| open(), os.path, pathlib 中的文件操作 | Service Class (文件) |
+| 内部 service 模块 | Integration Point |
+| 外部 SDK（boto3, google-cloud 等） | External Adapter |
+
+#### 2. 方法体分析
+
+| 方法体特征 | 指向分类 |
+|-----------|---------|
+| 无 self、无副作用、纯计算 | Pure Utility |
+| 含 open()/requests.get()/db.execute() | Service Class |
+| 调用多个其他 Service 的方法 | Integration Point |
+| 封装外部 SDK 调用 | External Adapter |
+| 仅 getter/setter/property | Data Model |
+
+#### 3. 类结构分析
+
+| 类结构特征 | 指向分类 |
+|-----------|---------|
+| @dataclass 装饰器 | Data Model |
+| __init__ 接收 config/依赖注入 | Service Class |
+| 继承 ABC 且仅含 abstractmethod | 跳过（无可测实现） |
+| 仅含 pass 或空方法体 | 跳过（无可测内容） |
+
+### 分类优先级
+
+当一个类有多个信号时，按优先级判定：
+
+1. Integration Point > Service Class > External Adapter > Pure Utility > Data Model
+2. 即：如果一个类既调用 DB 又编排多个 Service，归类为 Integration Point
+
+---
+
+## 阶段四：适用性过滤
+
+**操作**：根据 test_type 过滤
+
+### test_type = unit / ut
+
+UT 对所有有逻辑的类都适合，不过滤，返回全部类。
+
+| 类类型 | 是否返回 | 说明 |
+|--------|---------|------|
+| Pure Utility | ✅ | 纯函数逻辑测试 |
+| Data Model | ✅ | 验证逻辑、计算属性测试 |
+| Service Class | ✅ | Mock 依赖，测业务逻辑 |
+| Integration Point | ✅ | Mock 子 Service，测编排逻辑 |
+| External Adapter | ✅ | Mock 外部 API，测适配逻辑 |
+
+### test_type = integration / it
+
+IT 仅适合有外部依赖可集成的类。
+
+| 类类型 | 是否返回 | 原因 |
+|--------|---------|------|
+| Pure Utility | ❌ | 无外部依赖，IT = UT，无意义 |
+| Data Model | ❌ | 无交互可测，IT 无意义 |
+| Service Class | ✅ | 真实连接 DB/HTTP 测试 |
+| Integration Point | ✅ | 多 Service 真实编排测试 |
+| External Adapter | ✅ | 真实调用外部 API（需测试环境） |
+
+---
+
+## 阶段五：输出清单
+
+**输出格式**：
+
+```
+## 模块适用性评估
+
+### 可测试类清单
+| 文件路径 | 类名 | 分类 |
+|---------|------|------|
+| src/services/UserService.py | UserService | Service Class |
+| src/services/OrderService.py | OrderService | Integration Point |
+
+### 不适合的类
+| 文件路径 | 类名 | 分类 | 原因 |
+|---------|------|------|------|
+| src/utils/string_utils.py | StringUtils | Pure Utility | 纯函数无外部依赖，IT 无意义 |
+| src/models/User.py | User | Data Model | 数据容器，IT 无意义 |
+
+### 结论
+- 适合的类: X 个
+- 不适合的类: Y 个
+- 可测试清单: [文件路径列表]
+```
+
+---
+
+## 决策
+
+| 条件 | 动作 |
+|------|------|
+| 可测试清单非空 | 返回清单给 agent，agent 继续后续步骤 |
+| 可测试清单为空 | 返回空清单，agent 终止流程 |
+
+## 失败处理
+
+| 场景 | 处理方式 |
+|------|---------|
+| target 路径不存在 | 返回空清单，提示路径无效 |
+| target 不是目录 | 返回空清单，提示需要目录路径 |
+| .py 文件语法错误 | 跳过该文件，记录警告 |
+| 无法判定分类 | 默认归类为 Service Class（保守策略） |
+
+## 一定要做
+
+1. 必须递归扫描所有子目录
+2. 必须逐个类分类，不按模块整体判断
+3. 必须排除 __init__.py
+4. 必须对纯函数文件做模块级评估
+5. unit 类型必须返回全部类（不过滤）
+6. integration 类型必须过滤掉 Pure Utility 和 Data Model
+7. 必须输出不适合的类及原因
+8. 无法判定时必须保守归类为 Service Class
+
+## 一定不要做
+
+1. 不要修改源码文件
+2. 不要跳过子目录
+3. 不要按文件/模块整体判断（必须逐个类）
+4. 不要对 ut 类型做过滤
+5. 不要将空类或纯 ABC 纳入清单
+6. 不要主观臆断分类，必须基于信号规则