在Python中如何利用pytest测试加密与解密算法-Python教程-PHP中文网

pytest写加解密单元测试需固定随机源、显式控制填充与编码、用bytes或hex比对，encrypt返回(ciphertext, iv)供decrypt复用，参数化覆盖模式与密钥长度，并mock kms时注意base64编码和异常分支。

在python中如何利用pytest测试加密与解密算法

pytest怎么写加密/解密的单元测试

直接用 pytest 测试加解密逻辑完全可行，关键不是“能不能”，而是“怎么避免测得假阳性”——比如密文看起来不同但实际等价、IV没重用、填充被忽略、编码不一致导致断言失败。

核心做法：固定随机源、显式控制填充与编码、把加解密当成纯函数来测（输入→输出确定）。

所有涉及随机的步骤（如生成 IV、salt、密钥）必须用 secrets.token_bytes() 或 os.urandom()，但测试中要 mock 掉，或用固定字节替代（例如 b"iv-16-bytes-now"）
不要依赖 encrypt(...).decode('utf-8') 后再断言字符串——加密输出是二进制，应统一用 bytes 比较或 hex 编码后比对
解密失败时，别只检查是否抛异常；要确认抛的是你预期的异常类型（如 CryptographyError 而非 ValueError）

为什么test_decrypt(test_encrypt())会失败

常见现象：用 encrypt(data) 得到密文，再传给 decrypt()，结果报错或返回乱码。问题往往不在算法本身，而在状态残留或上下文错配。

AES-CBC 等模式要求 IV 与加密时完全一致，但若 encrypt() 内部每次生成新 IV 且不返回，decrypt() 就无从得知——必须让 encrypt() 返回 (ciphertext, iv) 元组
PKCS#7 填充在加密时自动添加，解密后需手动去除；若底层库（如 cryptography.hazmat.primitives.padding）没调用 unpadder.update() + unpadder.finalize()，就会多出填充字节
密钥若用字符串硬编码（如 "mykey123"），而实际算法要求 32 字节 AES-256 密钥，hashlib.sha256(key.encode()).digest() 是常见补救，但测试中必须复现同一哈希过程，不能一边用 SHA256、另一边直接截取前32位

如何用pytest参数化覆盖不同密钥长度和模式

不同密钥长度（128/192/256）、不同模式（CBC/CTR/GCM）行为差异大，硬写多个 test 函数易重复。用 @pytest.mark.parametrize 最省力，但要注意参数组合的真实有效性。

python全能编程助手

SkillSub Pro - Python 题解与代码注释双功能技能功能概述SkillSub Pro - Python 题解与代码注释双功能技能是一项面向实际任务的技能，主要用于SkillSub Pro 是一个 Python 题解生成与代码注释的双功能合体技能，专为学生、算法学习者和开发者设计；✅ 一个技能，两种用途：；核心要点📝 题解模式：输入题目/题号，自动生成完整 Python 题解（含详细注释、解题思路、复杂度分析）；💬 注释模式：输入 Python 代码，自动添加详细中。它将相关步骤、

下载

@pytest.mark.parametrize("mode,key_len,expected_error", [
    ("CBC", 16, None),
    ("GCM", 32, None),
    ("CBC", 24, ValueError),  # AES 不支持 192-bit key 的 CBC？错，它支持；但你的实现可能没适配
])
def test_aes_with_mode_and_key(mode, key_len, expected_error):
    key = b"x" * key_len
    iv_or_nonce = b"0123456789abcdef"[: (12 if mode == "GCM" else 16)]
    if expected_error:
        with pytest.raises(expected_error):
            encrypt(b"data", key, iv_or_nonce, mode=mode)
    else:
        ciphertext = encrypt(b"data", key, iv_or_nonce, mode=mode)
        assert decrypt(ciphertext, key, iv_or_nonce, mode=mode) == b"data"

注意：mode 和 key_len 必须符合底层库约束（例如 cryptography 中 GCM 要求 nonce 长度通常为 12 字节，而非 16）；参数列表里混入非法组合反而会让测试失焦。

mock外部密钥管理服务时容易漏掉什么

如果加解密依赖 KMS（如 AWS KMS、HashiCorp Vault），本地测试不能真调 API。用 pytest-mock 或 unittest.mock.patch 模拟响应没问题，但常漏三点：

模拟返回的密文是 base64 编码字符串，而你的解密函数可能默认接收 bytes —— 必须在 mock 里做 base64.b64encode(...).decode() 或保持类型一致
KMS 的 decrypt() 响应里密文是 Plaintext 字段，但字段值仍是 base64，不是明文；mock 返回 {"Plaintext": b"raw"} 是错的，应返回 {"Plaintext": base64.b64encode(b"raw")}
没覆盖 KeyUnavailableError 这类网络异常的 fallback 行为——比如降级用本地密钥加解密，这种分支必须显式触发并断言

加解密测试最麻烦的从来不是“跑通”，而是“跑通了却掩盖了 IV 重用、填充错误或编码隐式转换”。每条测试用例背后，都该清楚知道它在保卫哪一行真实部署中会出事的代码。