Add basic test for GC.

James William Pye · James William Pye · commit 91ecdf8a2253 · 2009-04-30T03:13:28.000-07:00
Make sure that the weakref callback is getting called.
diff --git a/postgresql/test/test_driver.py b/postgresql/test/test_driver.py
@@ -9,6 +9,7 @@
 import time
 import datetime
 import decimal
+import gc
 from itertools import chain, islice
 from operator import itemgetter
 
@@ -317,6 +318,27 @@ def testItsClosed(self):
 		c.close()
 		self.db.close()
 
+	def testGarbage(self):
+		ps = self.db.prepare('select 1')
+		sid = ps.statement_id
+		ci = ps.chunks()
+		ci_id = ci.cursor.cursor_id
+		c = ps.declare()
+		cid = c.cursor_id
+		# make sure there are no remaining xact references..
+		self.db._pq_complete()
+		# ci and c both hold references to ps, so they must
+		# be removed before we can observe the effects __del__
+		del c
+		gc.collect()
+		self.failUnless(self.db.typio.encode(cid) in self.db._closeportals)
+		del ci
+		gc.collect()
+		self.failUnless(self.db.typio.encode(ci_id) in self.db._closeportals)
+		del ps
+		gc.collect()
+		self.failUnless(self.db.typio.encode(sid) in self.db._closestatements)
+
 	def testStatementCall(self):
 		ps = self.db.prepare("SELECT 1")
 		r = ps()
