import testconfig
import unittest
import my_db
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 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 testSetPreferences(self):
        self.db.set_preferences(self.prefs)
        prefs = self.db.get_preferences()
        self.assertEqual(prefs, self.prefs)

if __name__ == '__main__':
    unittest.main()