Fixed ping to handle failures properly.

Fixes #115 in the right way.
FSX · Aug 26, 2015 · 5e83907 · 5e83907
1 parent 7c3aefa
commit 5e83907
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/momoko/connection.py b/momoko/connection.py
@@ -556,7 +556,10 @@ def on_ping_done(ping_fut):
                 try:
                     ping_fut.result()
                 except psycopg2.Error as error:
-                    ping_future.set_exc_info(error)
+                    if conn.closed:
+                        ping_future.set_exception(self._no_conn_availble_error)
+                    else:
+                        ping_future.set_exc_info(sys.exc_info())
                     self.putconn(conn)
                 else:
                     ping_future.set_result(conn)