New version with tester.
authorfabien@tzone.org
Tue, 30 Dec 2008 23:16:05 -0500
changeset 66 a0aa025ee6a2
parent 65 b975a3d7606a
child 69 d79722064d8d
New version with tester.
lib/my_db.py
tests/lib/test_my_db.py
--- a/lib/my_db.py	Tue Dec 30 22:32:10 2008 -0500
+++ b/lib/my_db.py	Tue Dec 30 23:16:05 2008 -0500
@@ -9,15 +9,16 @@
 
 class MyDbConnexion:
 	def __init__(self, config):
+		self.userid = None 
 		self.cnx = pgdb.connect(database = config.db_name,
 			host = config.db_host+':%d'%config.db_port,
 			user = config.db_user, password = config.db_passwd)
 		self.crs = self.cnx.cursor();
 	def connect_user(self, name):
 		if (self.userid == None):
-			self.userid = self._get_user_id(username)
+			self.userid = self._get_user_id(name)
 			if self.userid == None:
-				self.userid = self._create_user(username)
+				self.userid = self._create_user(name)
 			else:
 				self.userid = self.userid[0]
 		return self.userid
@@ -67,7 +68,7 @@
 			DELETE FROM preferences WHERE userid = %d;
 			DELETE FROM keywords WHERE userid = %d;
 			DELETE FROM db_sequence WHERE userid = %d;
-			""" % ((userid,)*5))
+			""" % ((userid,)*6))
 		self.cnx.commit()
 	def get_preferences(self):
 		self.crs.execute("""
@@ -398,4 +399,7 @@
 
 def connect(config, username):
 	db = MyDbConnexion(config)
-	db.connect_user(username)
+	userid = db.connect_user(username)
+	if userid == None:
+		raise "Can't connect user!"
+	return db
--- a/tests/lib/test_my_db.py	Tue Dec 30 22:32:10 2008 -0500
+++ b/tests/lib/test_my_db.py	Tue Dec 30 23:16:05 2008 -0500
@@ -1,10 +1,6 @@
-if __name__ == '__main__':
-    import sys
-    sys.path.insert(0, "../../lib")
-
+import testconfig
 import unittest
 import my_db
-import testconfig
 import pgdb
 
 class TestMyDbConnexion(unittest.TestCase):
@@ -19,22 +15,25 @@
     
     def tearDown(self):
         " Tear down the test-db environment"
-        self.db.delete_user('test')
+        self.db.delete_user(self.db.userid)
         pass
 
     def testConnect(self):
-        self.assertNotEqual(db, None)
-        self.assertNotEqual(db.userid, -1)
+        self.assertNotEqual(self.db, None)
+        self.assertNotEqual(self.db.userid, -1)
 
     def testGetPreferences(self):
         prefs = self.db.get_preferences()
-        self.assertEqual(prefs, self.prefs)
+	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()