import repository from arizona
[raven.git] / 2.0 / python / test / storkvalidatepubkeyTest.py
1 """
2 <Module>
3    storkvalidatepubkeyTest
4
5 <Author>
6    Collin Reynolds
7
8 <Started>
9    October 23, 2007
10
11 <Purpose>
12    Test module for storkvalidatepubkey.py
13 """
14
15 import arizonaunittest
16 import arizonaconfig
17 import arizonareport
18 import os
19 import storkvalidatepubkey
20
21 class test(arizonaunittest.TestCase):
22     
23    #Test the only method in the test, which checks via arizonacrypt whether the key is valid
24    def test_main(self):
25       #Test with blank filename
26       self.set_cmdline([""])
27       self.assertException(SystemExit,storkvalidatepubkey.main)
28       self.assertEquals( "error:  is not a file.", arizonareport.message)
29         
30       #Test a valid key generated with storkutil.py
31       self.set_cmdline(["test/testkeys.publickey"])
32       self.assertException(SystemExit,storkvalidatepubkey.main)
33       self.assertEquals("test/testkeys.publickey is a valid public key file",arizonareport.message)
34         
35       #Test invalid key, some bytes deleted in a valid key
36       self.set_cmdline(["test/invalid.publickey"])        
37       self.assertException(SystemExit,storkvalidatepubkey.main)
38       self.assertEquals("test/invalid.publickey is NOT a valid public key file",arizonareport.message)
39         
40       #Test invalid key, some bytes changed in valid key
41       self.set_cmdline(["test/invalid2.publickey"])        
42       self.assertException(SystemExit,storkvalidatepubkey.main)
43       self.assertEquals("test/invalid2.publickey is NOT a valid public key file",arizonareport.message)
44
45 if __name__=='__main__':
46    arizonaconfig.init_options()
47    arizonaunittest.main()