Merge previous changeset and ensure everything seems to work.
authorfabien@tzone.org
Wed, 31 Dec 2008 00:07:25 -0500
changeset 69 d79722064d8d
parent 68 c1f1491f098c (current diff)
parent 66 a0aa025ee6a2 (diff)
child 70 1798859f7f6c
Merge previous changeset and ensure everything seems to work.
lib/templates.py
--- a/lib/my_db.py	Wed Dec 31 00:00:09 2008 -0500
+++ b/lib/my_db.py	Wed Dec 31 00:07:25 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/lib/templates.py	Wed Dec 31 00:00:09 2008 -0500
+++ b/lib/templates.py	Wed Dec 31 00:07:25 2008 -0500
@@ -1,7 +1,7 @@
 from htmltmpl import TemplateManager, TemplateProcessor
 from urlparse import urljoin
 from os import environ, path
-from config import template_dir
+from config import CONFIG
 
 def get_curl():
 	url = "http"
@@ -15,7 +15,7 @@
 class Template:
 	def __init__(self, template, prefs, status = "msg", status_msg = ""):
 		self.tproc = TemplateProcessor()
-		self.tmpl = TemplateManager().prepare(path.join(template_dir, template))
+		self.tmpl = TemplateManager().prepare(path.join(CONFIG.template_dir, template))
     		self.set("remote_user", prefs["fullname"])
     		self.set("kw_size", prefs['keywords_box'])
     		self.set("kw_sort", prefs['keywords_sort'])
--- a/tests/lib/test_my_db.py	Wed Dec 31 00:00:09 2008 -0500
+++ b/tests/lib/test_my_db.py	Wed Dec 31 00:07:25 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()