[SQL-CVS] r1641 - in SQLObject/trunk/sqlobject: postgres tests

[<sub...@co...>]

Author: phd
Date: 2006-03-09 08:54:54 -0700 (Thu, 09 Mar 2006)
New Revision: 1641

Modified:
   SQLObject/trunk/sqlobject/postgres/pgconnection.py
   SQLObject/trunk/sqlobject/tests/test_parse.py
Log:
Allow non-default path to unix-socket in postgres DB URIs.


Modified: SQLObject/trunk/sqlobject/postgres/pgconnection.py
===================================================================
--- SQLObject/trunk/sqlobject/postgres/pgconnection.py	2006-03-09 07:43:05 UTC (rev 1640)
+++ SQLObject/trunk/sqlobject/postgres/pgconnection.py	2006-03-09 15:54:54 UTC (rev 1641)
@@ -90,6 +90,10 @@
     def connectionFromURI(cls, uri):
         user, password, host, port, path, args = cls._parseURI(uri)
         path = path.strip('/')
+        if (host is None) and path.count('/'): # Non-default unix socket
+            path_parts = path.split('/')
+            host = '/' + '/'.join(path_parts[:-1])
+            path = path_parts[-1]
         return cls(host=host, port=port, db=path, user=user, password=password, **args)
     connectionFromURI = classmethod(connectionFromURI)
 

Modified: SQLObject/trunk/sqlobject/tests/test_parse.py
===================================================================
--- SQLObject/trunk/sqlobject/tests/test_parse.py	2006-03-09 07:43:05 UTC (rev 1640)
+++ SQLObject/trunk/sqlobject/tests/test_parse.py	2006-03-09 15:54:54 UTC (rev 1641)
@@ -8,21 +8,21 @@
 def test_parse():
     _parseURI = DBConnection._parseURI
 
-    user, password, host, port, path, args = _parseURI("mysql://user:password@host/database")
-    assert user == "user"
-    assert password == "password"
+    user, password, host, port, path, args = _parseURI("mysql://host/database")
+    assert user is None
+    assert password is None
     assert host == "host"
     assert port is None
     assert path == "/database"
     assert args == {}
 
-    user, password, host, port, path, args = _parseURI("mysql://host/database")
-    assert user is None
-    assert password is None
+    user, password, host, port, path, args = _parseURI("mysql://user:pass%20word@host/database?unix_socket=/var/mysql/socket")
+    assert user == "user"
+    assert password == "pass%20word"
     assert host == "host"
     assert port is None
     assert path == "/database"
-    assert args == {}
+    assert args == {"unix_socket": "/var/mysql/socket"}
 
     user, password, host, port, path, args = _parseURI("postgres://user@host/database")
     assert user == "user"
@@ -40,6 +40,14 @@
     assert path == "/database"
     assert args == {}
 
+    user, password, host, port, path, args = _parseURI("postgres:///full/path/to/socket/database")
+    assert user == None
+    assert password is None
+    assert host == None
+    assert port is None
+    assert path == "/full/path/to/socket/database"
+    assert args == {}
+
     user, password, host, port, path, args = _parseURI("sqlite:///full/path/to/database")
     assert user is None
     assert password is None