Fixed segfault in large object close.

Check that the connection is not closed/faulty before attempting lo_close.

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/test_lobject.py b/tests/test_lobject.py
index 056496e..b3e3880 100755
--- a/tests/test_lobject.py
+++ b/tests/test_lobject.py
@@ -60,6 +60,10 @@ class LargeObjectMixin(object):
     def tearDown(self):
         if self.tmpdir:
             shutil.rmtree(self.tmpdir, ignore_errors=True)
+
+        if self.conn.closed:
+            return
+
         if self.lo_oid is not None:
             self.conn.rollback()
             try:
@@ -106,6 +110,11 @@ class LargeObjectTests(LargeObjectMixin, unittest.TestCase):
         self.assertEqual(lo2.oid, lo.oid)
         self.assertEqual(lo2.closed, True)
 
+    def test_close_connection_gone(self):
+        lo = self.conn.lobject()
+        self.conn.close()
+        lo.close()
+
     def test_create_with_oid(self):
         # Create and delete a large object to get an unused Oid.
         lo = self.conn.lobject()