diff options
| -rw-r--r-- | tests/test_openpgp.py | 60 |
1 files changed, 52 insertions, 8 deletions
diff --git a/tests/test_openpgp.py b/tests/test_openpgp.py index 11a6f6e..3771bed 100644 --- a/tests/test_openpgp.py +++ b/tests/test_openpgp.py @@ -30,6 +30,26 @@ jCvJNJ7pU8YnJSRTQDH0PZEupAdzDU/AhGSrBz5+Jr7N0pQIxq4duE/Q -----END PGP PUBLIC KEY BLOCK----- ''' +MALFORMED_PUBLIC_KEY = u''' +-----BEGIN PGP PUBLIC KEY BLOCK----- + +mQENBFnwXJMBCACgaTVz+d10TGL9zR920sb0GBFsitAJ5ZFzO4E0cg3SHhwI+reM +JQ6LLKmHowY/E1dl5FBbnJoRMxXP7/eScQ7HlhYj1gMPN5XiS2pkPwVkmJKBDV42 +DLwoytC+ot0frRTJvSdEPCX81BNMgFiBSpkeZfXqb9XmU03bh6mFnrdd4CsHpTQG +csVXHK8QKhaxuqmHTALdpSzKCb/r0N/Z3sQExZhfLcBf/9UUVXj44Nwc6ooqZLRi +zHydxwQdxNu0aOFGEBn9WTi8Slf7MfR/pF0dI8rs9w6zMzVEq0lhDPpKFGDveoGf +g/+TpvBNXZ7DWH23GM4kID3pk4LLMc24U1PhABEBAAG0D2dlbWF0byB0ZXN0IGtl +eYkBRgQTAQoAMBYhBIHhLBa9jc1gvhgIRRNogOcqexOEBQJZ8FyTAhsDBQsJCg0E +AxUKCAIeAQIXgAAKCRATaIDnKnsThCnkB/0fhTH230idhlfZhFbVgTLxrj4rpsGg +20K8HkMaWzshsONdKkqYaYuRcm2UQZ0Kg5rm9jQsGYuAnzH/7XwmOleY95ycVfBk +je9aXF6BEoGick6C/AK5w77vd1kcBtJDrT4I7vwD4wRkyUdCkpVMVT4z4aZ7lHJ4 +ECrrrI/mg0b+sGRyHfXPvIPp7F2959L/dpbhBZDfMOFC0A9LBQBJldKFbQLg3xzX +4tniz/BBrp7KjTOMKU0sufsedI50xc6cvCYCwJElqo86vv69klZHahE/k9nJaUAM +jCvJNJ7pU8YnJSRTQDH0PZEupAdzDU/AhGSrBz5+Jr7N0pQIxq4duE/Q +=r7JK +-----END PGP PUBLIC KEY BLOCK----- +''' + SIGNED_MANIFEST = u''' -----BEGIN PGP SIGNED MESSAGE----- Hash: SHA512 @@ -209,6 +229,32 @@ class OpenPGPContextManagerTest(unittest.TestCase): Test the context manager API for OpenPGPEnvironment. """ + def test_import_key(self): + with gemato.openpgp.OpenPGPEnvironment() as env: + try: + env.import_key( + io.BytesIO(PUBLIC_KEY.encode('utf8'))) + except gemato.exceptions.OpenPGPNoImplementation as e: + raise unittest.SkipTest(str(e)) + + def test_import_malformed_key(self): + with gemato.openpgp.OpenPGPEnvironment() as env: + try: + self.assertRaises(RuntimeError, + env.import_key, + io.BytesIO(MALFORMED_PUBLIC_KEY.encode('utf8'))) + except gemato.exceptions.OpenPGPNoImplementation as e: + raise unittest.SkipTest(str(e)) + + def test_import_no_keys(self): + with gemato.openpgp.OpenPGPEnvironment() as env: + try: + self.assertRaises(RuntimeError, + env.import_key, + io.BytesIO(b'')) + except gemato.exceptions.OpenPGPNoImplementation as e: + raise unittest.SkipTest(str(e)) + def test_verify_manifest(self): with io.BytesIO(SIGNED_MANIFEST.encode('utf8')) as f: with gemato.openpgp.OpenPGPEnvironment() as env: @@ -224,13 +270,11 @@ class OpenPGPContextManagerTest(unittest.TestCase): raise unittest.SkipTest(str(e)) def test_double_close(self): - with io.BytesIO(SIGNED_MANIFEST.encode('utf8')) as f: - with gemato.openpgp.OpenPGPEnvironment() as env: - env.close() + with gemato.openpgp.OpenPGPEnvironment() as env: + env.close() def test_home_after_close(self): - with io.BytesIO(SIGNED_MANIFEST.encode('utf8')) as f: - with gemato.openpgp.OpenPGPEnvironment() as env: - env.close() - with self.assertRaises(RuntimeError): - env.home + with gemato.openpgp.OpenPGPEnvironment() as env: + env.close() + with self.assertRaises(RuntimeError): + env.home |