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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
import unittest
import doctest
import os
import tempfile
import thandy.keys
import thandy.formats
import thandy.repository
import thandy.checkJson
import thandy.tests
class CanonicalEncodingTest(unittest.TestCase):
def test_encode(self):
enc = thandy.formats.encodeCanonical
self.assertEquals(enc(''), '""')
self.assertEquals(enc('"'), '"\\""')
self.assertEquals(enc('\t\\\n"\r'),
'"\t\\\\\n\\"\r"')
class CryptoTests(unittest.TestCase):
def test_encrypt(self):
s = "The Secret words are marzipan habidashery zeugma."
password = "the password is swordfish."
encrypted = thandy.keys.encryptSecret(s, password)
self.assertNotEquals(encrypted, s)
self.assert_(encrypted.startswith("GKEY1"))
self.assertEquals(s, thandy.keys.decryptSecret(encrypted, password))
self.assertRaises(thandy.BadPassword, thandy.keys.decryptSecret,
encrypted, "password")
self.assertRaises(thandy.UnknownFormat, thandy.keys.decryptSecret,
"foobar", password)
def test_keystore(self):
passwd = "umfitty noonah"
fname = tempfile.mktemp()
ks = thandy.keys.KeyStore(fname)
key1 = thandy.keys.RSAKey.generate(512)
key2 = thandy.keys.RSAKey.generate(512)
ks.addKey(key1)
ks.addKey(key2)
ks.save(passwd)
ks2 = thandy.keys.KeyStore(fname)
ks2.load(passwd)
self.assertEquals(key1.key.n, ks2.getKey(key1.getKeyID()).key.n)
def suite():
suite = unittest.TestSuite()
suite.addTest(doctest.DocTestSuite(thandy.formats))
suite.addTest(doctest.DocTestSuite(thandy.keys))
suite.addTest(doctest.DocTestSuite(thandy.checkJson))
loader = unittest.TestLoader()
suite.addTest(loader.loadTestsFromModule(thandy.tests))
return suite
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=1).run(suite())
|
