[project @ robey@lag.net-20060627064106-71a2bfb5a39a2913]

add PKey.from_private_key to read from a file object

1 files changed, 6 insertions, 0 deletions

diff --git a/tests/test_pkey.py b/tests/test_pkey.py
index d4b826ad..d5419cde 100644
--- a/tests/test_pkey.py
+++ b/tests/test_pkey.py
@@ -91,6 +91,9 @@ class KeyTest (unittest.TestCase):
         s = StringIO.StringIO()
         key.write_private_key(s)
         self.assertEquals(RSA_PRIVATE_OUT, s.getvalue()) 
+        s.seek(0)
+        key2 = RSAKey.from_private_key(s)
+        self.assertEquals(key, key2)
 
     def test_3_load_rsa_password(self):
         key = RSAKey.from_private_key_file('tests/test_rsa_password.key', 'television')
@@ -113,6 +116,9 @@ class KeyTest (unittest.TestCase):
         s = StringIO.StringIO()
         key.write_private_key(s)
         self.assertEquals(DSS_PRIVATE_OUT, s.getvalue())
+        s.seek(0)
+        key2 = DSSKey.from_private_key(s)
+        self.assertEquals(key, key2)
 
     def test_5_load_dss_password(self):
         key = DSSKey.from_private_key_file('tests/test_dss_password.key', 'television')