Python 文件在单元测试失败后是只读的

Question

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

    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 密钥之前完整读取文件？

Answer 1

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

我建议在文本编辑器中打开私钥文件并删除这些标记。您可能还想调查文件的写入位置，如果它是您代码的其他部分，则修复它。您可能在 bytes 对象上调用 str，而您应该调用 decode 方法（例如 binary_key_text.decode()）。或者，您可以在编写文件时以二进制模式 ('wb') 打开文件，然后直接写入 bytes 对象。

您或许可以在阅读 Python 代码时解决这个问题（使用类似 pem_key = pem_key[2:-1] 的代码，删除前两个字符和最后一个字符）。但是，如果您将其添加到您的代码中，那么它将仅适用于错误写入的文件，而不适用于正常文件。这就是 bug 是如何被铭记在软件中的，以至于它们在未来无法修复！（我想你可以使用 if pem_key.startswith("b'") and pem_key.endswith("'") 来保护切片操作，但我只会在错误的文件编写代码被广泛发布并且你真的需要向后保证时才会打扰兼容性。)

Python 文件在单元测试失败后是只读的

Python file is read only after unit test failure

python

python-unittest