FileSecureTransfer/recv/test_recv.py

from unittest.mock import patch

import recv
from sender import sender


@patch('builtins.input', side_effect=['./public.pem', 'aes', 'ecb'])
def test_handleLetter_aes_ecb(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass


@patch('builtins.input', side_effect=['./public.pem', 'aes', 'cbc'])
def test_handleLetter_aes_cbc(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass


@patch('builtins.input', side_effect=['./public.pem', 'sm4', 'ecb'])
def test_handleLetter_sm4_ecb(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass


@patch('builtins.input', side_effect=['./public.pem', 'sm4', 'cbc'])
def test_handleLetter_sm4_cbc(mock_input):
    letter = sender.main()
    recv.handleLetter(letter)
    pass