Added support for saving private keys in DER and PEM format
diff --git a/tests/test_load_save_keys.py b/tests/test_load_save_keys.py
index 695de72..c13d059 100644
--- a/tests/test_load_save_keys.py
+++ b/tests/test_load_save_keys.py
@@ -20,6 +20,12 @@
-----END CONFUSING STUFF-----
''' % B64PRIV_DER
+CLEAN_PRIVATE_PEM = '''\
+-----BEGIN RSA PRIVATE KEY-----
+%s
+-----END RSA PRIVATE KEY-----
+''' % B64PRIV_DER
+
class DerTest(unittest.TestCase):
'''Test saving and loading DER keys.'''
@@ -32,6 +38,13 @@
self.assertEqual(expected, key)
+ def test_save_private_key(self):
+ '''Test saving private DER keys.'''
+
+ key = rsa.key.PrivateKey(3727264081, 65537, 3349121513, 65063, 57287)
+ der = rsa.key.save_private_key_der(key)
+
+ self.assertEqual(PRIVATE_DER, der)
class PemTest(unittest.TestCase):
'''Test saving and loading PEM keys.'''
@@ -45,3 +58,11 @@
self.assertEqual(expected, key)
+ def test_save_private_key(self):
+ '''Test saving private PEM files.'''
+
+ key = rsa.key.PrivateKey(3727264081, 65537, 3349121513, 65063, 57287)
+ pem = rsa.key.save_private_key_pem(key)
+
+ self.assertEqual(CLEAN_PRIVATE_PEM, pem)
+
