You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
791 B
33 lines
791 B
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
|