From e9fc17cf57ec14e652f0effe4e85279ea5f2aba2 Mon Sep 17 00:00:00 2001 From: "Sybren A. St?vel" Date: Tue, 19 Jul 2011 23:53:59 +0200 Subject: Added loading of DER and PEM encoded private keys --- tests/test_load_save_keys.py | 47 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 tests/test_load_save_keys.py (limited to 'tests') diff --git a/tests/test_load_save_keys.py b/tests/test_load_save_keys.py new file mode 100644 index 0000000..695de72 --- /dev/null +++ b/tests/test_load_save_keys.py @@ -0,0 +1,47 @@ +'''Unittest for saving and loading keys.''' + +import base64 +import unittest + +import rsa.key + +B64PRIV_DER = 'MC4CAQACBQDeKYlRAgMBAAECBQDHn4npAgMA/icCAwDfxwIDANcXAgInbwIDAMZt' +PRIVATE_DER = base64.decodestring(B64PRIV_DER) + +PRIVATE_PEM = ''' +-----BEGIN CONFUSING STUFF----- +Cruft before the key + +-----BEGIN RSA PRIVATE KEY----- +%s +-----END RSA PRIVATE KEY----- + +Stuff after the key +-----END CONFUSING STUFF----- +''' % B64PRIV_DER + + +class DerTest(unittest.TestCase): + '''Test saving and loading DER keys.''' + + def test_load_private_key(self): + '''Test loading private DER keys.''' + + key = rsa.key.load_private_key_der(PRIVATE_DER) + expected = rsa.key.PrivateKey(3727264081, 65537, 3349121513, 65063, 57287) + + self.assertEqual(expected, key) + + +class PemTest(unittest.TestCase): + '''Test saving and loading PEM keys.''' + + + def test_load_private_key(self): + '''Test loading private PEM files.''' + + key = rsa.key.load_private_key_pem(PRIVATE_PEM) + expected = rsa.key.PrivateKey(3727264081, 65537, 3349121513, 65063, 57287) + + self.assertEqual(expected, key) + -- cgit v1.2.1