当前位置：首页 > news >正文

pyuadk测试驱动开发：编写可靠密码学组件的完整流程

news 2026/7/11 19:45:00

pyuadk测试驱动开发：编写可靠密码学组件的完整流程

【免费下载链接】pyuadkpython wrapper of uadk项目地址: https://gitcode.com/openeuler/pyuadk

前往项目官网免费下载：https://ar.openeuler.org/ar/

pyuadk是openEuler社区推出的密码学组件Python封装库，通过测试驱动开发（TDD）方法确保密码学功能的可靠性与安全性。本文将详细介绍如何为pyuadk构建完整的测试体系，从单元测试到集成验证，确保AES、SM4、RSA等核心密码学算法的实现正确性。

为什么密码学组件需要测试驱动开发？

密码学组件的正确性直接关系到数据安全，任何微小的错误都可能导致加密失效或密钥泄露。测试驱动开发（TDD）通过"先测试后编码"的方式，为每个密码学功能建立可验证的安全基线。pyuadk项目中，测试文件test.py和test_async.py分别实现了同步和异步密码学操作的测试用例，覆盖了从算法正确性到性能稳定性的全维度验证。

测试环境搭建：快速上手pyuadk测试框架

准备工作

克隆代码仓库

git clone https://gitcode.com/openeuler/pyuadk cd pyuadk

安装依赖
项目依赖已在requirements.txt中声明，通过以下命令安装：
```
pip install -r requirements.txt
```
测试文件结构
pyuadk的测试体系主要包含：
- test.py：同步密码学操作测试（AES/SM4加密、SM3哈希、RSA密钥生成与签名）
- test_async.py：异步密码学操作测试

核心测试用例设计：以密码学功能为中心

1. 对称加密算法测试（AES/SM4）

对称加密是数据传输的基础保障，pyuadk的测试用例通过双重验证法确保实现正确性：

使用pyuadk加密后，通过标准密码库（如PyCryptodome、gmssl）解密验证结果
验证加密和解密的一致性（加密后解密应还原原始数据）

示例测试逻辑（来自test.py）：

# AES-CBC模式测试 def test_cipher_aes_cbc_sync(self): # 1. 初始化pyuadk加密上下文 self.cipher.ctx_setup(pywd_cipher.CIPHER_ALG.AES, pywd_cipher.CIPHER_MODE.CBC) self.cipher.create_cipher_ctx() self.cipher.wcrypto_set_cipher_key(self.key, len(self.key)) # 2. 执行加密操作 self.cipher.set_opdata(self.iv, pywd_cipher.CIPHER_OP_TYPE.ENCRYPT, self.pt) ret = self.cipher.wcrypto_do_cipher() cth = self.cipher.get_ct() # 获取pyuadk加密结果 # 3. 标准库验证 aes_ecb = AES.new(self.key, AES.MODE_CBC, self.iv) cts = aes_ecb.encrypt(self.pt) # 标准库加密结果 self.assertEqual(cth[0], cts[0]) # 对比结果一致性

2. 哈希算法测试（SM3）

哈希算法的核心需求是输出一致性和抗碰撞性。pyuadk测试用例通过对比标准库哈希结果，确保SM3算法实现符合国家标准：

关键测试点（来自test.py）：

普通哈希模式（NORMAL）：验证SM3哈希值与gmssl库结果一致
HMAC模式：验证带密钥的哈希计算正确性

def test_digest_sm3_normal(self): self.digest.ctx_setup(pywd_digest.DIGEST_ALG.SM3, pywd_digest.DIGEST_MODE.NORMAL) self.digest.create_digest_ctx() self.digest.set_opdata(self.pt, 16) ret = self.digest.wcrypto_do_digest() digest_h = self.digest.get_digest() # pyuadk计算结果 digest_s = sm3_hash(list(self.pt)) # gmssl标准结果 # 验证哈希值一致性（示例中对比首字节） self.assertEqual(digest_h[0], int(digest_s[:2], base=16))

3. RSA非对称加密测试

RSA算法涉及密钥生成、签名、验证等复杂流程，测试用例需覆盖：

密钥生成（普通模式与CRT优化模式）
签名与验证的一致性
公私钥参数的正确性

CRT模式密钥生成测试（来自test.py）：

def test_rsa_genkey_crt(self): self.rsa.ctx_setup(1) # 1表示CRT模式 self.rsa.create_rsa_ctx() self.rsa.wcrypto_new_kg_in(self.e, self.p, self.q) # 输入素数p/q self.rsa.set_opdata(pywd_rsa.RSA_OP_TYPE.GENKEY) ret = self.rsa.wcrypto_do_rsa() dq, dp, qinv = self.rsa.get_kg_out_crt() # 获取CRT参数 # 验证生成的dp/dq/qinv与标准库一致 self.assertEqual(int.from_bytes(dp, byteorder="big"), self.private_key.private_numbers().dmp1)

测试执行与结果验证

运行测试用例

通过Python unittest框架执行测试：

python test.py

预期输出

测试通过后，控制台将显示各算法测试结果：

aes cbc test pass aes ecb test pass sm4 ecb test pass sm4 cbc test pass sm3 normal test pass sm3 hmac test pass rsa common genkey test pass rsa ctr genkey test pass rsa sign common test pass rsa sign crt test pass rsa verify common and crt test pass

测试驱动开发的最佳实践

覆盖所有密码学操作
确保每个API（如加密/解密、哈希、签名）都有对应的测试用例，参考test.py中的TestPyWdCipherSync、TestPyWdDigest、TestPyWdRSA类结构。
使用权威标准库作为参照
测试中通过PyCryptodome（AES/RSA）、gmssl（SM4/SM3）等权威库验证结果，避免"自己测试自己"的逻辑漏洞。
异常场景测试
补充边界测试用例，如：
- 密钥长度不合法（如AES-128使用15字节密钥）
- 数据长度非块大小整数倍（如AES-CBC模式下明文长度为17字节）