单元测试失败后,只读Python文件

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了单元测试失败后,只读Python文件，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1452字，纯文字阅读大概需要3分钟。

内容图文

我正在尝试创建一个单元测试来测试以下功能：

    def decrypt_keys(self, enc_password):
        with open ('id_rsa','r') as privkeyfile:
            pem_key = privkeyfile.read()
            print(pem_key)
            priv = rsa.PrivateKey.load_pkcs1(pem_key)
        return rsa.decrypt(base64.b64decode(bytes(enc_password, 'utf-8')),priv).decode('utf-8')

但是,单元测试失败并显示以下错误：

ValueError: No PEM start marker "b'-----BEGIN RSA PRIVATE KEY-----'" found

----------------------------------------------------------------------
Ran 1 test in 0.012s

FAILED (errors=1)
b'-----BEGIN RSA PRIVATE KEY-----
 ... rest of private key here ...
-----END RSA PRIVATE KEY-----'

随即可以看出,文件的内容将按照库的预期进行打印.在函数尝试将其作为rsa密钥加载之前,如何确保完全读取文件？

解决方法:

看来您的文件中包含Python二进制字符串引号,这破坏了RSA读取代码,因为它不知道如何解释私钥周围的b’…’.

我建议在文本编辑器中打开私钥文件并删除这些标记.您可能还想调查文件的写入位置,并在代码的其他部分对其进行修复.当您应该调用解码方法时(例如,binary_key_text.decode()),您可能会在字节对象上调用str.或者,可以在写入文件时以二进制模式(‘wb’)打开文件,然后直接写入bytes对象.

您也许可以在阅读Python代码时解决此问题(使用pem_key = pem_key [2：-1]之类的东西,将前两个字符和最后一个字符切掉).但是,如果将其添加到代码中,则仅适用于写错的文件,不适用于普通文件.这就是将错误包含在软件中的方式,以便将来无法修复！ (我想您可以通过pem_key.startswith(“ b’”)和pem_key.endswith(“’”)来保护切片操作,但是如果有问题的文件编写代码已经广泛发布,并且您真的需要保证向后兼容.)

内容总结

以上是互联网集市为您收集整理的单元测试失败后,只读Python文件全部内容，希望文章能够帮你解决单元测试失败后,只读Python文件所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端