1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#!/usr/bin/env python
"""PGP test program.
Copyright (c) 1999 Ng Pheng Siong. All rights reserved."""
import unittest
from M2Crypto import EVP, PGP
from cStringIO import StringIO
class PGPTestCase(unittest.TestCase):
def test_simple(self):
pkr = PGP.load_pubring('tests/pubring.pgp')
daft = pkr['daft']
daft_pkt = daft._pubkey_pkt.pack()
s1 = EVP.MessageDigest('sha1')
s1.update(daft_pkt)
s1f = `s1.final()`
buf = StringIO(daft_pkt)
ps = PGP.packet_stream(buf)
dift_pkt = ps.read()
s2 = EVP.MessageDigest('sha1')
s2.update(dift_pkt.pack())
s2f = `s2.final()`
self.assertEqual(s1f, s2f)
def suite():
suite = unittest.TestSuite()
suite.addTest(unittest.makeSuite(PGPTestCase))
return suite
if __name__ == '__main__':
Rand.load_file('randpool.dat', -1)
unittest.TextTestRunner().run(suite())
Rand.save_file('randpool.dat')
|
