from unittest.mock import patch import recv from sender import sender from tool import PriKeyHelper @patch('builtins.input', side_effect=['./public.pem', 'sm4', 'cbc']) def test_handleLetter(mock_input): pki = PriKeyHelper.getUserKey() letter = sender.main() recv.handleLetter(letter) pass