1 files changed, 35 insertions, 0 deletions
diff --git a/src/mailman_pgp/pgp/tests/test_inline.py b/src/mailman_pgp/pgp/tests/test_inline.py
index f8831c9..d144991 100644
--- a/src/mailman_pgp/pgp/tests/test_inline.py
+++ b/src/mailman_pgp/pgp/tests/test_inline.py
@@ -88,3 +88,38 @@ class TestEncryption(PGPTestCase, unittest.TestCase):
         wrapped = self.wrapper(message)
         self.assertEqual(wrapped.decrypt(key).message,
                          bytearray(clear, 'latin-1'))
+
+
+class TestKeys(PGPTestCase, unittest.TestCase):
+    def setUp(self):
+        self.wrapper = InlineWrapper
+
+    @parameterized.expand([
+        (load_message('data/inline_privkey.eml'),
+         True),
+        (load_message('data/inline_pubkey.eml'),
+         True),
+        (load_message('data/clear.eml'),
+         False),
+        (load_message('data/inline_signed.eml'),
+         False)
+    ])
+    def test_has_keys(self, message, has_keys):
+        wrapped = self.wrapper(message)
+        self.assertEqual(wrapped.has_keys(), has_keys)
+
+    @parameterized.expand([
+        (load_message('data/inline_privkey.eml'),
+         [load_key('data/rsa_1024.priv.asc')]),
+        (load_message('data/inline_pubkey.eml'),
+         [load_key('data/rsa_1024.pub.asc')])
+    ])
+    def test_keys(self, message, keys):
+        wrapped = self.wrapper(message)
+        loaded = list(wrapped.keys())
+        loaded_fingerprints = list(map(lambda key: key.fingerprint, loaded))
+
+        self.assertEqual(len(loaded), len(keys))
+
+        fingerprints = list(map(lambda key: key.fingerprint, keys))
+        self.assertListEqual(loaded_fingerprints, fingerprints)