# -*- coding: utf-8 -*-
"""把 Java 测试的第五段改成顺序场景：先断开第一个，再用令牌重连。

原来的写法是两个连接同时活着 —— 新的把旧的顶掉。那是并发边界情况。
实际的用法是"手表关了再打开"，也就是顺序的。先把这个主流程测通。
"""
from __future__ import annotations

import sys
from pathlib import Path

TEST = Path(r"C:\Users\shu\Desktop\deepseekagent\LanRemote\_tests\java"
            r"\JavaRelayTest.java")

OLD = '''        // ---------------- 5. 令牌免配对重连 ----------------
        System.out.println();
        System.out.println("五、用已保存的令牌重连（下次打开不用再输码）");
        RemoteClient again = new RemoteClient();'''

NEW = '''        // ---------------- 5. 令牌免配对重连 ----------------
        System.out.println();
        System.out.println("五、用已保存的令牌重连（下次打开不用再输码）");
        // 先断开第一条连接 —— 这才是实际用法（手表关了再打开）
        client.close();
        check("断开后 isConnected 为 false", !client.isConnected());
        Thread.sleep(1500);
        RemoteClient again = new RemoteClient();'''


def main() -> int:
    text = TEST.read_text(encoding="utf-8")
    if OLD not in text:
        print("  [失败] 锚点没找到")
        return 1
    TEST.write_text(text.replace(OLD, NEW, 1), encoding="utf-8")
    print("  第五段改成顺序场景")
    return 0


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