diff -r f1b93ddec9e0 -r b975a3d7606a tests/lib/test_my_db.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/lib/test_my_db.py Tue Dec 30 22:32:10 2008 -0500 @@ -0,0 +1,40 @@ +if __name__ == '__main__': + import sys + sys.path.insert(0, "../../lib") + +import unittest +import my_db +import testconfig +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('test') + pass + + def testConnect(self): + self.assertNotEqual(db, None) + self.assertNotEqual(db.userid, -1) + + def testGetPreferences(self): + prefs = self.db.get_preferences() + self.assertEqual(prefs, self.prefs) + + def testSetPreferences(self): + self.db.set_preferences(self.prefs) + prefs = self.db.get_preferences() + self.assertEqual(prefs, self.prefs) + + +if __name__ == '__main__': + unittest.main()