From 90085efa4c5acb3ebdb862157c9ecf10aafa6530 Mon Sep 17 00:00:00 2001 From: "Sybren A. St?vel" Date: Wed, 20 Jul 2011 01:11:34 +0200 Subject: Added support for saving private keys in DER and PEM format --- tests/test_load_save_keys.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) (limited to 'tests') 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 @@ Stuff after the key -----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 @@ class DerTest(unittest.TestCase): 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 @@ class PemTest(unittest.TestCase): 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) + -- cgit v1.2.1