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