from pathlib import Path
root=Path('C:/kefu/wechat_rpa')
p=root/'test_visual_review_assistant.py'
text=p.read_text(encoding='utf-8')
text=text.replace('        self.assistant.resumed.return_value = True', '        self.assistant.resumed.return_value = True\n        self.assistant.has_pending.return_value = False')
text=text.replace('        self.bot._reply_needs_review = mock.Mock(return_value=(True, \'new high risk\'))', '        self.bot._reply_needs_review = mock.Mock(return_value=(True, \'new high risk\'))')
text=text.replace("self.assertEqual(self.bot._reply_review_decision(self.state, 'q', 'a'), (True, False, ''))", "self.assertEqual(self.bot._reply_review_decision(self.state, 'q', 'a'), (True, True, 'new high risk'))")
extra=(Path(__file__).parent/'extra_visual_tests.txt').read_text(encoding='utf-8')
text=text.replace("\nif __name__ == '__main__':", '\n'+extra+"\nif __name__ == '__main__':")
for target in (root, Path('C:/wechat_rpa')):
    (target/'test_visual_review_assistant.py').write_text(text,encoding='utf-8',newline='\n')