Partial submission, to include tests.
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()