From 660386929f26a75f47264cad36e81305e4cc2a83 Mon Sep 17 00:00:00 2001
From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Mon, 18 Mar 2013 00:24:46 +0000
Subject: [PATCH] Added test to verify Diagnostics reference disposal

---
 tests/test_module.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/tests/test_module.py b/tests/test_module.py
index e92e3b97..168ddaa7 100755
--- a/tests/test_module.py
+++ b/tests/test_module.py
@@ -185,6 +185,31 @@ class ExceptionsTestCase(unittest.TestCase):
         self.assertEqual(e.diag.severity, 'ERROR')
         self.assertEqual(e.diag.statement_position, '15')
 
+    def test_diagnostics_life(self):
+        import gc
+        from weakref import ref
+
+        def tmp():
+            cur = self.conn.cursor()
+            try:
+                cur.execute("select * from nonexist")
+            except psycopg2.Error, exc:
+                return cur, exc
+
+        cur, e = tmp()
+        diag = e.diag
+        w = ref(cur)
+
+        del e, cur
+        gc.collect()
+        assert(w() is not None)
+
+        self.assertEqual(diag.sqlstate, '42P01')
+
+        del diag
+        gc.collect()
+        assert(w() is None)
+
     @skip_before_postgres(9, 3)
     def test_9_3_diagnostics(self):
         cur = self.conn.cursor()