Add test cases for PublicKeyEncryptedSessionKeyPacket

Signed-off-by: Dan McGee <dpmcgee@gmail.com>
toofishes · May 12, 2012 · daba6ac · daba6ac
1 parent b3a91fc
commit daba6ac
Show file tree

Hide file tree

Showing 3 changed files with 59 additions and 0 deletions.
diff --git a/pgpdump/test.py b/pgpdump/test.py
@@ -390,6 +390,34 @@ def test_parse_v3_elgamal_pk(self):
         self.assertEqual(888716292, packet.raw_creation_time)
 
 
+class EncryptedPacketsTestCase(TestCase, Helper):
+    def test_parse_sessionkey_elg(self):
+        '''This file contains a public key and message encrypted with an
+        ElGamal Encrypt-Only key.'''
+        asc_data = self.load_data('sessionkey_elg.asc')
+        data = AsciiData(asc_data)
+        packets = list(data.packets())
+        self.assertEqual(2, len(packets))
+        session_key = packets[0]
+        self.assertEqual(3, session_key.session_key_version)
+        self.assertEqual(b"B705D3A4C3751D38", session_key.key_id)
+        self.assertEqual(16, session_key.raw_pub_algorithm)
+        self.assertEqual("ElGamal Encrypt-Only", session_key.pub_algorithm)
+
+    def test_parse_sessionkey_rsa(self):
+        '''This file contains a public key and message encrypted with a RSA
+        Encrypt or Sign key.'''
+        asc_data = self.load_data('sessionkey_rsa.asc')
+        data = AsciiData(asc_data)
+        packets = list(data.packets())
+        self.assertEqual(2, len(packets))
+        session_key = packets[0]
+        self.assertEqual(3, session_key.session_key_version)
+        self.assertEqual(b"1C39A7BD114BFFA5", session_key.key_id)
+        self.assertEqual(1, session_key.raw_pub_algorithm)
+        self.assertEqual("RSA Encrypt or Sign", session_key.pub_algorithm)
+
+
 class PacketTestCase(TestCase):
     def test_lookup_type(self):
         self.assertEqual("Signature Packet", TAG_TYPES[2][0])

diff --git a/testdata/sessionkey_elg.asc b/testdata/sessionkey_elg.asc
@@ -0,0 +1,18 @@
+-----BEGIN PGP MESSAGE-----
+Version: GnuPG v2.0.19 (GNU/Linux)
+
+hQIOA7cF06TDdR04EAf/S4Y0EIc+gZpGMMIeRbWKBHgKDlTYSbcZCcRRpWQuhNp3
++3aAvtufR18rlmMPHKIZXP9TMqA84GIqexHPNufTzoDbPkcNcPk/fiO09gK4GewD
+44Gp2iDV5JBAeNMF84KzYSpCTshUNdkfWdLE+XwwxxEFkHvpRGOAz3IUDqARRCXy
+wJZ2xEyvu/04VSe8l0vv0HFCsr8an+KxLDpfU9kp4p1Qi6eyEzIAM5HBPRCZzHMg
+mOjuRD+Qyp8DoRX62e2kUczH2c1//yBXBqA6DCz1DXG/4XObmzKwD75en40CvmOZ
+wSMF8guIV12tC1ZVJP0iK+0rchaKP8AX2MC9YiaQGgf+KUImjjW9Ik6EtBI9A/H5
+1i/EbcflZq8JscH17jkhHB1f7WUaBMQZnLUtPoj/28U3gQb0O9Fc8PD0EBhsenhz
+jkIrJXmnMLW9qhdD/GeymIK1GQqVSMBqXXCS0XJINNCW/ZWIgPL9fKtYhAEhA3uj
+EILo1vxncOQBIhIKmPctVffB54/hoNakr3hsqZBoGYe3iLipF+W0Tg+/xqjdxVal
+szeFjmkbNClX+ou8Opl560ZJen9gujmljLQUbfWfqAfAzpHg5tkbaKY5I+U03GE2
+crcwA5p6D8HObZvOS6GPJWKgXbpAOWX4Q9gFUsQx8K5K1sDk9lhXjtBlvsg+UZgS
+m9JKARkukSZKKf1FBHailjJlbpgBt3Uq6QTLJVCqnvwoHZ62+El2Qo4AytcGmKC3
+JqKOwpEwu7k7BD9azZJldRzO94lPxEyuOW20zxQ=
+=6Tor
+-----END PGP MESSAGE-----
diff --git a/testdata/sessionkey_rsa.asc b/testdata/sessionkey_rsa.asc
@@ -0,0 +1,13 @@
+-----BEGIN PGP MESSAGE-----
+Version: GnuPG v2.0.19 (GNU/Linux)
+
+hQEMAxw5p70RS/+lAQf+PKkPn604EcH56EEWdP64SebRVBkIED+Fj1NMLBOu9KCG
+keUWYe2LBfxTHK5Con1oR0rbWk3V3TXZMUs8S6l7VJ2mF9dVXftOhhqxU3s2Idjj
+QHifQreYLk73FHvONdrIAYIEKcWAyK3s1wZ1WqobRMSXTk/cA7b9YUAQkoctc0Ix
+EtgLF/azlR1pc6w97wZI5MKrUnAnCqfbDd9TCHvPE3OhQzceTicHs+q5MLrLd1M3
+zBL2AD+mO6kVs3HvEy/jPd16wAE2SNakeDtJzypcU4PoDy0vpbNOTKnQievRDRCA
+58qxnf0r0Yf/R5e6Jgfaf/k64YxZDstnTAigN8I/09JJAcl0dyZmrXLpC4DPTKrN
+6UgOjeE2+o6Z6OHvHNMaZSibmZ3vbTQG/C7xl+8mNNZEywcjLSx43zgmWNY8ajH0
+COwqvQVAe7A9kA==
+=DQda
+-----END PGP MESSAGE-----