from pathlib import Path
import os
import shutil

for root in (Path(r'C:\kefu\wechat_rpa'), Path(r'C:\wechat_rpa')):
    backup = root / 'backups/review-visibility-20260916/qt'
    for name in ('test_queue_log.py', 'test_qt_console_bridge.py'):
        path = root / name
        if not (backup / name).exists():
            shutil.copy2(path, backup / name)
        text = path.read_text(encoding='utf-8')
        if name == 'test_queue_log.py':
            old = 'self.page._state_text({"send_state": "uncertain"}), "发送待核对"'
            assert old in text
            text = text.replace(old, 'self.page._state_text({"send_state": "uncertain"}), "发送待核对（需人工）"')
        else:
            start = text.index('    def test_handoff_only_reports_tasks_the_backend_accepted(self):')
            end = text.index('    def test_queue_actions_select_exact_key_before_forwarding', start)
            text = text[:start] + '''    def test_handoff_only_reports_tasks_the_backend_accepted(self):
        thread = SimpleNamespace(
            is_alive=lambda: True,
            handoff_pending_tasks=mock.Mock(return_value={
                "handed_off": ["done"], "scheduled": [],
                "protected": ["protected"], "error": "",
            }),
        )
        host = SimpleNamespace(
            _thread=thread,
            queue_page=SimpleNamespace(pending_states={}, refresh_data=mock.Mock()),
            append_log=mock.Mock(), _schedule_console_push=mock.Mock(),
        )
        with mock.patch("wechat_gui_qt.QMessageBox.information") as notice:
            MainWindow._handoff_queue_tasks(host, ["done", "protected"])
        thread.handoff_pending_tasks.assert_called_once_with(["done", "protected"])
        host.append_log.assert_called_once_with(
            "已将 1 条任务转交人工，保留发送记录，不再自动重发。", "notify",
        )
        notice.assert_called_once()
        host.queue_page.refresh_data.assert_called_once_with()

''' + text[end:]
        temp = path.with_suffix('.audit-tmp')
        temp.write_text(text, encoding='utf-8')
        os.replace(temp, path)
    print(str(root) + ': updated regression contracts')