diff options
Diffstat (limited to 'tests/test_pem.py')
-rw-r--r-- | tests/test_pem.py | 39 |
1 files changed, 20 insertions, 19 deletions
diff --git a/tests/test_pem.py b/tests/test_pem.py index dd03cca..7440431 100644 --- a/tests/test_pem.py +++ b/tests/test_pem.py @@ -19,14 +19,14 @@ from rsa.pem import _markers import rsa.key # 512-bit key. Too small for practical purposes, but good enough for testing with. -public_key_pem = ''' +public_key_pem = """ -----BEGIN PUBLIC KEY----- MFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBAKH0aYP9ZFuctlPnXhEyHjgc8ltKKx9M 0c+h4sKMXwjhjbQAZdtWIw8RRghpUJnKj+6bN2XzZDazyULxgPhtax0CAwEAAQ== -----END PUBLIC KEY----- -''' +""" -private_key_pem = ''' +private_key_pem = """ -----BEGIN RSA PRIVATE KEY----- MIIBOwIBAAJBAKH0aYP9ZFuctlPnXhEyHjgc8ltKKx9M0c+h4sKMXwjhjbQAZdtW Iw8RRghpUJnKj+6bN2XzZDazyULxgPhtax0CAwEAAQJADwR36EpNzQTqDzusCFIq @@ -36,7 +36,7 @@ F8p1J98BEbtjU2mEZIVCMn6vQuhWdl8CIDRL4IJl4eGKlB0QP0JJF1wpeGO/R76l DaPF5cMM7k3NAiEAss28m/ck9BWBfFVdNjx/vsdFZkx2O9AX9EJWoBSnSgECIQCa +sVQMUVJFGsdE/31C7wCIbE3IpB7ziABZ7mN+V3Dhg== -----END RSA PRIVATE KEY----- -''' +""" # Private key components prime1 = 96275860229939261876671084930484419185939191875438854026071315955024109172739 @@ -45,9 +45,10 @@ prime2 = 88103681619592083641803383393198542599284510949756076218404908654323473 class TestMarkers(unittest.TestCase): def test_values(self): - self.assertEqual(_markers('RSA PRIVATE KEY'), - (b'-----BEGIN RSA PRIVATE KEY-----', - b'-----END RSA PRIVATE KEY-----')) + self.assertEqual( + _markers("RSA PRIVATE KEY"), + (b"-----BEGIN RSA PRIVATE KEY-----", b"-----END RSA PRIVATE KEY-----"), + ) class TestBytesAndStrings(unittest.TestCase): @@ -58,7 +59,7 @@ class TestBytesAndStrings(unittest.TestCase): self.assertEqual(prime1 * prime2, key.n) def test_bytes_public(self): - key = rsa.key.PublicKey.load_pkcs1_openssl_pem(public_key_pem.encode('ascii')) + key = rsa.key.PublicKey.load_pkcs1_openssl_pem(public_key_pem.encode("ascii")) self.assertEqual(prime1 * prime2, key.n) def test_unicode_private(self): @@ -66,7 +67,7 @@ class TestBytesAndStrings(unittest.TestCase): self.assertEqual(prime1 * prime2, key.n) def test_bytes_private(self): - key = rsa.key.PrivateKey.load_pkcs1(private_key_pem.encode('ascii')) + key = rsa.key.PrivateKey.load_pkcs1(private_key_pem.encode("ascii")) self.assertEqual(prime1, key.p) self.assertEqual(prime2, key.q) @@ -76,24 +77,24 @@ class TestByteOutput(unittest.TestCase): def test_bytes_public(self): key = rsa.key.PublicKey.load_pkcs1_openssl_pem(public_key_pem) - self.assertIsInstance(key.save_pkcs1(format='DER'), bytes) - self.assertIsInstance(key.save_pkcs1(format='PEM'), bytes) + self.assertIsInstance(key.save_pkcs1(format="DER"), bytes) + self.assertIsInstance(key.save_pkcs1(format="PEM"), bytes) def test_bytes_private(self): key = rsa.key.PrivateKey.load_pkcs1(private_key_pem) - self.assertIsInstance(key.save_pkcs1(format='DER'), bytes) - self.assertIsInstance(key.save_pkcs1(format='PEM'), bytes) + self.assertIsInstance(key.save_pkcs1(format="DER"), bytes) + self.assertIsInstance(key.save_pkcs1(format="PEM"), bytes) class TestByteInput(unittest.TestCase): """Tests that PEM and DER can be loaded from bytes.""" def test_bytes_public(self): - key = rsa.key.PublicKey.load_pkcs1_openssl_pem(public_key_pem.encode('ascii')) - self.assertIsInstance(key.save_pkcs1(format='DER'), bytes) - self.assertIsInstance(key.save_pkcs1(format='PEM'), bytes) + key = rsa.key.PublicKey.load_pkcs1_openssl_pem(public_key_pem.encode("ascii")) + self.assertIsInstance(key.save_pkcs1(format="DER"), bytes) + self.assertIsInstance(key.save_pkcs1(format="PEM"), bytes) def test_bytes_private(self): - key = rsa.key.PrivateKey.load_pkcs1(private_key_pem.encode('ascii')) - self.assertIsInstance(key.save_pkcs1(format='DER'), bytes) - self.assertIsInstance(key.save_pkcs1(format='PEM'), bytes) + key = rsa.key.PrivateKey.load_pkcs1(private_key_pem.encode("ascii")) + self.assertIsInstance(key.save_pkcs1(format="DER"), bytes) + self.assertIsInstance(key.save_pkcs1(format="PEM"), bytes) |