修正base64问题

recv/recv.py

@@ -40,9 +40,9 @@ def handleLetter(letter: Letter):
     elif type == config.EncryptType.SM4_CBC:
         data = base64.b64decode(SM4.decrypt_cbc_with_iv(letter.fileBase64, key))
     elif type == config.EncryptType.AES_ECB:
-        data = AES.AESUtils(key).decrypt(key, letter.fileBase64, "ecb")
+        data = base64.b64decode(AES.AESUtils(key).decrypt(key, letter.fileBase64, "ecb"))
     elif type == config.EncryptType.AES_CBC:
-        data = AES.AESUtils(key).decrypt(key, letter.fileBase64,"cbc")
+        data = base64.b64decode(AES.AESUtils(key).decrypt(key, letter.fileBase64,"cbc"))
     else:
         raise KeyError("不支持的对称加密算法")
 

recv/test_recv.py

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