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()