from pathlib import Path
import shutil

base = Path(__file__).parent
path = base / 'test_visual_handoff_visibility.py'
source = path.read_text(encoding='utf-8')
assert 'test_plain_cancel_cannot_bypass' not in source
source = source.replace("\n\nif __name__ == '__main__':", '\n' + (base / 'visual_handoff_extra_tests.txt').read_text(encoding='utf-8') + "\n\nif __name__ == '__main__':")
path.write_text(source, encoding='utf-8')
for root in (Path('C:/wechat_rpa'), Path('C:/kefu/wechat_rpa')):
    shutil.copy2(path, root / path.name)