diff -r d79722064d8d -r 1798859f7f6c tests/lib/test_my_db.py --- a/tests/lib/test_my_db.py Wed Dec 31 00:07:25 2008 -0500 +++ b/tests/lib/test_my_db.py Wed Dec 31 01:00:21 2008 -0500 @@ -4,36 +4,55 @@ import pgdb class TestMyDbConnexion(unittest.TestCase): - def setUp(self): - " Setup a test-db environment." - self.db = my_db.connect(testconfig.CONFIG, "test") - self.prefs = { 'keywords_box' : 10, - 'keywords_sort' : 'count', - 'keywords_reverse': True, - 'fullname': 'Test User', - 'default_view': 1 } - - def tearDown(self): - " Tear down the test-db environment" - self.db.delete_user(self.db.userid) - pass + def setUp(self): + " Setup a test-db environment." + self.db = my_db.connect(testconfig.CONFIG, "test") + self.prefs = { 'keywords_box' : 10, + 'keywords_sort' : 'count', + 'keywords_reverse': True, + 'fullname': 'Test User', + 'default_view': 1 } + + def tearDown(self): + " Tear down the test-db environment" + self.db.delete_user(self.db.userid) + + def testConnect(self): + self.assertNotEqual(self.db, None) + self.assertNotEqual(self.db.userid, -1) + + def testGetPreferences(self): + prefs = self.db.get_preferences() + keys = prefs.keys() + keys.sort() + expKeys = self.prefs.keys() + expKeys.sort() + self.assertEqual(expKeys, keys) - def testConnect(self): - self.assertNotEqual(self.db, None) - self.assertNotEqual(self.db.userid, -1) + def testSetPreferences(self): + self.db.set_preferences(self.prefs) + prefs = self.db.get_preferences() + self.assertEqual(prefs, self.prefs) + + def testAddKeyword(self): + kw = self.db.add_keyword('keyword') + self.assertEqual(self.db.get_keyword(kw), 'keyword') - def testGetPreferences(self): - prefs = self.db.get_preferences() - keys = prefs.keys() - keys.sort() - expKeys = self.prefs.keys() - expKeys.sort() - self.assertEqual(expKeys, keys) + def testRemoveKeyword(self): + kw = self.db.add_keyword('keyword') + self.db.remove_keyword(kw) + kws = self.db.get_all_keywords() + # There is always '--' by default + self.assertEqual(len(kws), 1) - def testSetPreferences(self): - self.db.set_preferences(self.prefs) - prefs = self.db.get_preferences() - self.assertEqual(prefs, self.prefs) + def testMapKeywords(self): + kws = ('kw1', 'kw2', 'kw3', 'kw4') + kwids = map(self.db.add_keyword, kws[:-1]) + expected_map = { kws[-1]: -1 } + for kw, id in zip(kws[:-1], kwids): + expected_map[kw] = id + kw_map = self.db.map_keywords(kws) + self.assertEqual(expected_map, kw_map) if __name__ == '__main__': unittest.main()