Premiere version : mise en route du suivi.
[auf_roundup.git] / test / test_userauditor.py
1 # $Id: test_userauditor.py,v 1.4 2007-09-12 21:11:14 jpend Exp $
2
3 import os, unittest, shutil
4 from db_test_base import setupTracker
5
6 class UserAuditorTest(unittest.TestCase):
7 def setUp(self):
8 self.dirname = '_test_user_auditor'
9 self.instance = setupTracker(self.dirname)
10 self.db = self.instance.open('admin')
11
12 try:
13 import pytz
14 self.pytz = True
15 except ImportError:
16 self.pytz = False
17
18 self.db.user.create(username='kyle', address='kyle@example.com',
19 realname='Kyle Broflovski', roles='User')
20
21 def tearDown(self):
22 self.db.close()
23 try:
24 shutil.rmtree(self.dirname)
25 except OSError, error:
26 if error.errno not in (errno.ENOENT, errno.ESRCH): raise
27
28 def testBadTimezones(self):
29 self.assertRaises(ValueError, self.db.user.create, username='eric', timezone='24')
30
31 userid = self.db.user.lookup('kyle')
32
33 self.assertRaises(ValueError, self.db.user.set, userid, timezone='3000')
34 self.assertRaises(ValueError, self.db.user.set, userid, timezone='24')
35 self.assertRaises(ValueError, self.db.user.set, userid, timezone='-24')
36 self.assertRaises(ValueError, self.db.user.set, userid, timezone='-3000')
37
38 if self.pytz:
39 try:
40 from pytz import UnknownTimeZoneError
41 except:
42 UnknownTimeZoneError = ValueError
43 self.assertRaises(UnknownTimeZoneError, self.db.user.set, userid, timezone='MiddleOf/Nowhere')
44
45 def testGoodTimezones(self):
46 self.db.user.create(username='test_user01', timezone='12')
47
48 if self.pytz:
49 self.db.user.create(username='test_user02', timezone='MST')
50
51 userid = self.db.user.lookup('kyle')
52
53 # TODO: roundup should accept non-integer offsets since those are valid
54 # this is the offset for Tehran, Iran
55 #self.db.user.set(userid, timezone='3.5')
56
57 self.db.user.set(userid, timezone='-23')
58 self.db.user.set(userid, timezone='23')
59 self.db.user.set(userid, timezone='0')
60
61 if self.pytz:
62 self.db.user.set(userid, timezone='US/Eastern')
63
64 def testBadEmailAddresses(self):
65 userid = self.db.user.lookup('kyle')
66 self.assertRaises(ValueError, self.db.user.set, userid, address='kyle @ example.com')
67 self.assertRaises(ValueError, self.db.user.set, userid, address='one@example.com,two@example.com')
68 self.assertRaises(ValueError, self.db.user.set, userid, address='weird@@example.com')
69 self.assertRaises(ValueError, self.db.user.set, userid, address='embedded\nnewline@example.com')
70 # verify that we check alternates as well
71 self.assertRaises(ValueError, self.db.user.set, userid, alternate_addresses='kyle @ example.com')
72 # make sure we accept local style addresses
73 self.db.user.set(userid, address='kyle')
74 # verify we are case insensitive
75 self.db.user.set(userid, address='kyle@EXAMPLE.COM')
76
77 def testUniqueEmailAddresses(self):
78 self.db.user.create(username='kenny', address='kenny@example.com', alternate_addresses='sp_ken@example.com')
79 self.assertRaises(ValueError, self.db.user.create, username='test_user01', address='kenny@example.com')
80 uid = self.db.user.create(username='eric', address='eric@example.com')
81 self.assertRaises(ValueError, self.db.user.set, uid, address='kenny@example.com')
82
83 # make sure we check alternates
84 self.assertRaises(ValueError, self.db.user.set, uid, address='kenny@example.com')
85 self.assertRaises(ValueError, self.db.user.set, uid, address='sp_ken@example.com')
86 self.assertRaises(ValueError, self.db.user.set, uid, alternate_addresses='kenny@example.com')
87
88 def testBadRoles(self):
89 userid = self.db.user.lookup('kyle')
90 self.assertRaises(ValueError, self.db.user.set, userid, roles='BadRole')
91 self.assertRaises(ValueError, self.db.user.set, userid, roles='User,BadRole')
92
93 def testGoodRoles(self):
94 userid = self.db.user.lookup('kyle')
95 # make sure we handle commas in weird places
96 self.db.user.set(userid, roles='User,')
97 self.db.user.set(userid, roles=',User')
98 # make sure we strip whitespace
99 self.db.user.set(userid, roles=' User ')
100 # check for all-whitespace (treat as no role)
101 self.db.user.set(userid, roles=' ')
102
103 def test_suite():
104 suite = unittest.TestSuite()
105 suite.addTest(unittest.makeSuite(UserAuditorTest))
106 return suite
107
108 if __name__ == '__main__':
109 runner = unittest.TextTestRunner()
110 unittest.main(testRunner=runner)
111
112 # vim: filetype=python sts=4 sw=4 et si