2 import unittest |
2 import unittest |
3 import my_db |
3 import my_db |
4 import pgdb |
4 import pgdb |
5 |
5 |
6 class TestMyDbConnexion(unittest.TestCase): |
6 class TestMyDbConnexion(unittest.TestCase): |
7 def setUp(self): |
7 def setUp(self): |
8 " Setup a test-db environment." |
8 " Setup a test-db environment." |
9 self.db = my_db.connect(testconfig.CONFIG, "test") |
9 self.db = my_db.connect(testconfig.CONFIG, "test") |
10 self.prefs = { 'keywords_box' : 10, |
10 self.prefs = { 'keywords_box' : 10, |
11 'keywords_sort' : 'count', |
11 'keywords_sort' : 'count', |
12 'keywords_reverse': True, |
12 'keywords_reverse': True, |
13 'fullname': 'Test User', |
13 'fullname': 'Test User', |
14 'default_view': 1 } |
14 'default_view': 1 } |
15 |
15 |
16 def tearDown(self): |
16 def tearDown(self): |
17 " Tear down the test-db environment" |
17 " Tear down the test-db environment" |
18 self.db.delete_user(self.db.userid) |
18 self.db.delete_user(self.db.userid) |
19 pass |
|
20 |
19 |
21 def testConnect(self): |
20 def testConnect(self): |
22 self.assertNotEqual(self.db, None) |
21 self.assertNotEqual(self.db, None) |
23 self.assertNotEqual(self.db.userid, -1) |
22 self.assertNotEqual(self.db.userid, -1) |
24 |
23 |
25 def testGetPreferences(self): |
24 def testGetPreferences(self): |
26 prefs = self.db.get_preferences() |
25 prefs = self.db.get_preferences() |
27 keys = prefs.keys() |
26 keys = prefs.keys() |
28 keys.sort() |
27 keys.sort() |
29 expKeys = self.prefs.keys() |
28 expKeys = self.prefs.keys() |
30 expKeys.sort() |
29 expKeys.sort() |
31 self.assertEqual(expKeys, keys) |
30 self.assertEqual(expKeys, keys) |
32 |
31 |
33 def testSetPreferences(self): |
32 def testSetPreferences(self): |
34 self.db.set_preferences(self.prefs) |
33 self.db.set_preferences(self.prefs) |
35 prefs = self.db.get_preferences() |
34 prefs = self.db.get_preferences() |
36 self.assertEqual(prefs, self.prefs) |
35 self.assertEqual(prefs, self.prefs) |
|
36 |
|
37 def testAddKeyword(self): |
|
38 kw = self.db.add_keyword('keyword') |
|
39 self.assertEqual(self.db.get_keyword(kw), 'keyword') |
|
40 |
|
41 def testRemoveKeyword(self): |
|
42 kw = self.db.add_keyword('keyword') |
|
43 self.db.remove_keyword(kw) |
|
44 kws = self.db.get_all_keywords() |
|
45 # There is always '--' by default |
|
46 self.assertEqual(len(kws), 1) |
|
47 |
|
48 def testMapKeywords(self): |
|
49 kws = ('kw1', 'kw2', 'kw3', 'kw4') |
|
50 kwids = map(self.db.add_keyword, kws[:-1]) |
|
51 expected_map = { kws[-1]: -1 } |
|
52 for kw, id in zip(kws[:-1], kwids): |
|
53 expected_map[kw] = id |
|
54 kw_map = self.db.map_keywords(kws) |
|
55 self.assertEqual(expected_map, kw_map) |
37 |
56 |
38 if __name__ == '__main__': |
57 if __name__ == '__main__': |
39 unittest.main() |
58 unittest.main() |