from unittest.mock import patch

import recv
from senders import sender


@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem','aes', 'ecb'])
def test_handleLetter_aes_ecb(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass


@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem', 'aes', 'cbc'])
def test_handleLetter_aes_cbc(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass


@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem', 'sm4', 'ecb'])
def test_handleLetter_sm4_ecb(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass


@patch('builtins.input', side_effect=['../README.md',r'D:\Users\Timmoc\AppData\Roaming\fst\public.pem', 'sm4', 'cbc'])
def test_handleLetter_sm4_cbc(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass