1 if __name__ == '__main__':

     2     import sys

     3     sys.path.insert(0, "../../lib")

     4 

     5 import unittest

     6 import my_db

     7 import testconfig

     8 import pgdb

     9 

    10 class TestMyDbConnexion(unittest.TestCase):

    11     def setUp(self):

    12         " Setup a test-db environment."

    13         self.db = my_db.connect(testconfig.CONFIG, "test")

    14         self.prefs = { 'keywords_box' : 10,

    15                        'keywords_sort' : 'count',

    16                        'keywords_reverse': True,

    17                        'fullname': 'Test User',

    18                        'default_view': 1 }

    19     

    20     def tearDown(self):

    21         " Tear down the test-db environment"

    22         self.db.delete_user('test')

    23         pass

    24 

    25     def testConnect(self):

    26         self.assertNotEqual(db, None)

    27         self.assertNotEqual(db.userid, -1)

    28 

    29     def testGetPreferences(self):

    30         prefs = self.db.get_preferences()

    31         self.assertEqual(prefs, self.prefs)

    32 

    33     def testSetPreferences(self):

    34         self.db.set_preferences(self.prefs)

    35         prefs = self.db.get_preferences()

    36         self.assertEqual(prefs, self.prefs)

    37 

    38 

    39 if __name__ == '__main__':

    40     unittest.main()