ykxmz/_tests/test_selfcheck.py

# -*- coding: utf-8 -*-
"""验证「手表点连不上 -> 电脑打开网络自检」这条链路。

真起一个被控端，广播 LANREMOTE_SELFCHECK，确认：
  * 电脑回 ACK（带自己的名字）
  * 电脑日志里记下了这次请求
"""
from __future__ import annotations

import json
import shutil
import socket
import subprocess
import sys
import time
from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(ROOT / "windows"))

PASSED = 0
FAILED = 0
PORT = 8841
DISCOVERY_PORT = 8766


def check(name: str, ok: bool, detail: str = "") -> None:
    global PASSED, FAILED
    if ok:
        PASSED += 1
        print(f"  [PASS] {name}")
    else:
        FAILED += 1
        print(f"  [FAIL] {name}  {detail}")


def main() -> int:
    data_dir = ROOT / "windows" / "data"
    if data_dir.exists():
        shutil.rmtree(data_dir, ignore_errors=True)

    env = dict(**__import__("os").environ)
    env["PYTHONIOENCODING"] = "utf-8"

    proc = subprocess.Popen(
        # --add-device：静默运行 + 临时开放配对。
            # 不能用默认启动方式 —— 没设置过配对码会弹设置窗口，卡住。
            [sys.executable, "run_agent.py", "--port", str(PORT),
             "--new-code", "--add-device", "10"],
        cwd=str(ROOT / "windows"), stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL)

    try:
        # 等就绪
        deadline = time.time() + 25
        ready = False
        while time.time() < deadline:
            try:
                sock = socket.create_connection(("127.0.0.1", PORT), timeout=1)
                sock.close()
                ready = True
                break
            except OSError:
                time.sleep(0.4)
        check("被控端已就绪", ready)
        if not ready:
            return 1

        print("\n  —— 广播自检请求 ——")
        request = json.dumps({"lanremote": "LANREMOTE_SELFCHECK",
                              "proto": 1}).encode("utf-8")
        sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
        sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
        sock.settimeout(4)
        answers: list[str] = []
        try:
            for _ in range(3):
                sock.sendto(request, ("127.0.0.1", DISCOVERY_PORT))
                time.sleep(0.15)
            deadline = time.time() + 4
            while time.time() < deadline:
                try:
                    data, addr = sock.recvfrom(2048)
                except socket.timeout:
                    break
                text = data.decode("utf-8", errors="replace")
                if "LANREMOTE_ACK" in text:
                    answers.append(text)
        finally:
            sock.close()

        check("收到了 ACK 应答", bool(answers), "一个都没收到")
        if answers:
            print(f"      应答内容: {answers[0]}")
            try:
                obj = json.loads(answers[0])
                check("ACK 里带了电脑名",
                      bool(obj.get("name")), str(obj))
                check("ACK 的 magic 正确",
                      obj.get("lanremote") == "LANREMOTE_ACK", str(obj.get("lanremote")))
            except ValueError as exc:
                check("ACK 是合法 JSON", False, str(exc))

        # 日志里应该有记录
        time.sleep(1.0)
        # 被控端现在跑静默模式，日志写在 silent.log；旧版是 运行日志.txt
        log_text = ""
        for name in ("silent.log", "运行日志.txt"):
            log_path = data_dir / name
            if log_path.is_file():
                try:
                    log_text += log_path.read_text(encoding="utf-8",
                                                   errors="replace")
                except OSError:
                    pass
        check("日志里记下了这次自检请求",
              "自检" in log_text and "请求" in log_text,
              log_text[-200:] if log_text else "(没有日志)")
    finally:
        proc.terminate()
        try:
            proc.wait(timeout=6)
        except subprocess.TimeoutExpired:
            proc.kill()
        shutil.rmtree(data_dir, ignore_errors=True)

    print()
    print("=" * 60)
    print(f"通过 {PASSED} 项，失败 {FAILED} 项")
    return 1 if FAILED else 0


if __name__ == "__main__":
    sys.exit(main())