add test for defer_connect

grooverdan · grooverdan · commit 09a416a9e0a4 · 2015-08-30T09:50:02.000+10:00
diff --git a/pymysql/connections.py b/pymysql/connections.py
@@ -651,7 +651,9 @@ def _config(key, arg):
         self.sql_mode = sql_mode
         self.init_command = init_command
         self.max_allowed_packet = max_allowed_packet
-        if not defer_connect:
+        if defer_connect:
+            self.socket = None
+        else:
             self.connect()
 
     def close(self):
diff --git a/pymysql/tests/test_connection.py b/pymysql/tests/test_connection.py
@@ -107,6 +107,26 @@ def test_set_charset(self):
         c.set_charset('utf8')
         # TODO validate setting here
 
+    def test_defer_connect(self):
+        import socket
+        for db in self.databases:
+            d = db.copy()
+            try:
+                sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+                sock.connect(d['unix_socket'])
+            except KeyError:
+                sock = socket.create_connection(
+                                (d.get('host', 'localhost'), d.get('port', 3306)))
+            for k in ['unix_socket', 'host', 'port']:
+                try:
+                    del d[k]
+                except KeyError:
+                    pass
+
+            c = pymysql.connect(defer_connect=True, **d)
+            self.assertFalse(c.open)
+            c.connect(sock)
+
 
 # A custom type and function to escape it
 class Foo(object):
