py-wlcli 0.8.0__py3-none-any.whl

wl_cli/commands/testcommands/test_reporter.py

@@ -0,0 +1,181 @@
+from __future__ import annotations
+
+from ...utils.logging import log_info_pair
+from .utils import (
+    coverage_totals,
+    file_entries,
+    fmt_pct,
+    get_env_info,
+    parse_junit,
+    read_json,
+    read_xml,
+    write_md,
+)
+
+# Re-export private functions for backward compatibility (used in tests)
+_coverage_totals = coverage_totals
+_file_entries = file_entries
+_fmt_pct = fmt_pct
+_get_env_info = get_env_info
+_parse_junit = parse_junit
+_read_json = read_json
+_read_xml = read_xml
+_write_md = write_md
+
+
+def generate_reports(
+    coverage_json_path: str,
+    junit_xml_path: str,
+    out_cov: str,
+    out_err: str,
+    project_root: str,
+) -> str:
+    _log_inputs(coverage_json_path, junit_xml_path, out_cov, out_err)
+    cov = _read_json(coverage_json_path)
+    junit = _read_xml(junit_xml_path)
+    env_info = _get_env_info()
+
+    line_pct, branch_pct, num_lines, num_branches, num_files = _coverage_totals(cov)
+    files = _file_entries(cov, project_root)
+    top_files = sorted(files, key=lambda x: x[1] or 0.0, reverse=True)[:10]
+    low_files = sorted(files, key=lambda x: x[1] or 0.0)[:10]
+
+    cov_lines = _build_coverage_lines(
+        env_info,
+        line_pct,
+        branch_pct,
+        num_lines,
+        num_branches,
+        num_files,
+        top_files,
+        low_files,
+    )
+    _write_md(out_cov, cov_lines)
+    _log_written(out_cov)
+
+    tests, failures, errors, skipped, warnings, time_spent, cases = _parse_junit(junit)
+    err_lines = _build_error_lines(
+        env_info, tests, failures, errors, skipped, warnings, time_spent, cases
+    )
+    _write_md(out_err, err_lines)
+    _log_written(out_err)
+
+    summary = f"总行覆盖率 {fmt_pct(line_pct)} | 分支覆盖率 {fmt_pct(branch_pct)} | 文件数 {num_files} | 失败用例 {failures + errors}"
+    return summary
+
+
+def _log_inputs(
+    coverage_json_path: str, junit_xml_path: str, out_cov: str, out_err: str
+) -> None:
+    try:
+        log_info_pair(
+            f"Generating coverage report to: {out_cov}",
+            f"生成覆盖率报告到: {out_cov}",
+        )
+        log_info_pair(
+            f"Generating error report to: {out_err}",
+            f"生成错误报告到: {out_err}",
+        )
+        log_info_pair(
+            f"Coverage JSON source: {coverage_json_path}",
+            f"覆盖率JSON来源: {coverage_json_path}",
+        )
+        log_info_pair(
+            f"JUnit XML source: {junit_xml_path}",
+            f"JUnit XML来源: {junit_xml_path}",
+        )
+    except Exception:
+        pass
+
+
+def _log_written(path: str) -> None:
+    try:
+        log_info_pair(
+            f"Coverage Markdown written: {path}",
+            f"覆盖率Markdown已写入: {path}",
+        )
+    except Exception:
+        pass
+
+
+def _build_coverage_lines(
+    env_info,
+    line_pct,
+    branch_pct,
+    num_lines,
+    num_branches,
+    num_files,
+    top_files,
+    low_files,
+) -> list[str]:
+    cov_lines: list[str] = []
+    cov_lines.append("# WL 测试覆盖率报告")
+    cov_lines.append("")
+    cov_lines.append("## 测试环境")
+    cov_lines.extend([f"- {x}" for x in env_info])
+    cov_lines.append("")
+    cov_lines.append("## 总览")
+    cov_lines.append(f"- 行覆盖率: {fmt_pct(line_pct)}")
+    cov_lines.append(f"- 分支覆盖率: {fmt_pct(branch_pct)}")
+    cov_lines.append(f"- 语句数: {num_lines if num_lines else 'N/A'}")
+    cov_lines.append(f"- 分支数: {num_branches if num_branches else 'N/A'}")
+    cov_lines.append(f"- 文件数: {num_files}")
+    cov_lines.append("")
+    cov_lines.append("## Top 文件")
+    for rel, lp, bp in top_files:
+        cov_lines.append(f"- `{rel}` | 行 {fmt_pct(lp)} | 分支 {fmt_pct(bp)}")
+    cov_lines.append("")
+    cov_lines.append("## 低覆盖率文件")
+    for rel, lp, bp in low_files:
+        cov_lines.append(f"- `{rel}` | 行 {fmt_pct(lp)} | 分支 {fmt_pct(bp)}")
+    return cov_lines
+
+
+def _build_error_lines(
+    env_info, tests, failures, errors, skipped, warnings, time_spent, cases
+) -> list[str]:
+    err_lines: list[str] = []
+    err_lines.append("# WL 测试错误报告")
+    err_lines.append("")
+    err_lines.append("## 测试环境")
+    err_lines.extend([f"- {x}" for x in env_info])
+    err_lines.append("")
+    err_lines.append("## 测试统计")
+    err_lines.append(f"- 总用例: {tests}")
+    err_lines.append(f"- 失败: {failures}")
+    err_lines.append(f"- 错误: {errors}")
+    err_lines.append(f"- 跳过: {skipped}")
+    err_lines.append(f"- 警告: {warnings}")
+    err_lines.append(f"- 总时长: {time_spent:.2f}s")
+    err_lines.append("")
+
+    failure_cases = [c for c in cases if c.get("kind") == "failure"]
+    error_cases = [c for c in cases if c.get("kind") == "error"]
+    warning_cases = [c for c in cases if c.get("kind") == "warning"]
+
+    if failure_cases:
+        err_lines.append("## 失败明细")
+        _append_case_lines(err_lines, failure_cases)
+
+    if error_cases:
+        err_lines.append("## 错误明细")
+        _append_case_lines(err_lines, error_cases)
+
+    if warning_cases:
+        err_lines.append("## 警告明细")
+        _append_case_lines(err_lines, warning_cases)
+
+    return err_lines
+
+
+def _append_case_lines(err_lines: list[str], cases: list) -> None:
+    for c in cases:
+        loc = c.get("file") or c.get("classname") or ""
+        err_lines.append(
+            f"- 用例: `{c.get('name', '')}` @ `{loc}` [{c.get('kind', '')}]"
+        )
+        err_lines.append("")
+        err_lines.append("```")
+        err_lines.append(c.get("text", ""))
+        err_lines.append("```")
+        err_lines.append("")

wl_cli/commands/testcommands/test_result_handler.py

@@ -0,0 +1,96 @@
+"""
+Test result handling utilities for test command.
+"""
+
+import sys
+
+
+class TestResultHandler:
+    """Handle test results and exit codes."""
+
+    def handle_test_result(self, result, report_mode: bool) -> int:
+        """
+        Handle the test result based on return code.
+
+        Reports the outcome to the user and returns the exit code to hand back
+        to the caller. Exiting is deliberately left to the caller so that
+        post-run work (report generation, per-file cache attribution) still
+        happens on a failing suite.
+
+        Args:
+            result: The subprocess result object
+            report_mode: Whether we're in report mode
+
+        Returns:
+            The exit code to exit with (pytest's return code, unchanged)
+        """
+        if result.returncode == 0:
+            # Tests passed
+            if not report_mode:
+                print("All tests passed successfully!")
+        else:
+            # Tests failed
+            if not report_mode:
+                self._print_test_failure_info(result)
+        return int(result.returncode)
+
+    def _print_test_failure_info(self, result) -> None:
+        """
+        Print test failure information, filtering out progress bars.
+
+        Args:
+            result: The subprocess result object
+        """
+        # In concise mode, only show the short test summary info
+        # Skip the long progress bar to reduce token consumption
+        if hasattr(result, "stdout") and result.stdout:
+            output = self._filter_test_output(result.stdout, result)
+            # Only print if there's something left to print
+            if output.strip():
+                print(output)
+        if hasattr(result, "stderr") and result.stderr:
+            # stderr usually doesn't contain the summary
+            # Only output if it's not empty and not already in stdout
+            stderr = result.stderr.strip()
+            if stderr:
+                print(stderr, file=sys.stderr)
+        print(f"Tests failed with return code {result.returncode}")
+
+    def _filter_test_output(self, output: str, result) -> str:
+        """
+        Filter test output to remove progress bars and keep only useful information.
+
+        Args:
+            output: The test output to filter
+            result: The subprocess result object
+
+        Returns:
+            Filtered test output
+        """
+        # Find the short test summary info section
+        marker = "short test summary info"
+        if marker in output:
+            # Extract everything from the marker onwards
+            return output[output.index(marker) :]
+        else:
+            # If no marker found, filter out progress bar lines manually
+            # Filter lines that look like:
+            # - progress percentage: "... [XX%]"
+            # - progress dots: "filename.py ........." with many dots
+            lines = output.splitlines()
+            filtered_lines = []
+            for line in lines:
+                # Skip collection progress lines with percentage
+                # Matches both "[ 100%]" and "[100%]"
+                if "[" in line and "%]" in line:
+                    continue
+                # Skip lines like "filename.py ........." that contain many dots
+                # These are the per-file progress dots from pytest
+                if "." in line and line.count(".") > 7:
+                    continue
+                filtered_lines.append(line)
+            filtered_output = "\n".join(filtered_lines)
+            # If after filtering output is empty and there's stderr, don't print anything here
+            if not filtered_output.strip() and hasattr(result, "stderr"):
+                return ""
+            return filtered_output