tests/lib/test_my_db.py
author fabien@tzone.org
Wed, 31 Dec 2008 00:00:09 -0500
changeset 68 c1f1491f098c
parent 65 b975a3d7606a
child 66 a0aa025ee6a2
permissions -rw-r--r--
Some small correction and update.

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