tests/lib/test_my_db.py
changeset 65 b975a3d7606a
child 66 a0aa025ee6a2
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/lib/test_my_db.py	Tue Dec 30 22:32:10 2008 -0500
@@ -0,0 +1,40 @@
+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()