Added test to verify executemany doesn't propagate exceptions.

2 files changed, 31 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
index 2a79948..1c8f6f2 100755
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -37,6 +37,7 @@ import test_dates
 import test_psycopg2_dbapi20
 import test_quote
 import test_connection
+import test_cursor
 import test_transaction
 import types_basic
 import types_extras
@@ -54,6 +55,7 @@ def test_suite():
     suite.addTest(test_psycopg2_dbapi20.test_suite())
     suite.addTest(test_quote.test_suite())
     suite.addTest(test_connection.test_suite())
+    suite.addTest(test_cursor.test_suite())
     suite.addTest(test_transaction.test_suite())
     suite.addTest(types_basic.test_suite())
     suite.addTest(types_extras.test_suite())
diff --git a/tests/test_cursor.py b/tests/test_cursor.py
new file mode 100644
index 0000000..38c368a
--- /dev/null
+++ b/tests/test_cursor.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+
+import unittest
+import psycopg2
+import psycopg2.extensions
+import tests
+
+class CursorTests(unittest.TestCase):
+
+    def connect(self):
+        return psycopg2.connect(tests.dsn)
+
+    def test_executemany_propagate_exceptions(self):
+        conn = self.connect()
+        cur = conn.cursor()
+        cur.execute("create temp table test_exc (data int);")
+        def buggygen():
+            yield 1/0
+        self.assertRaises(ZeroDivisionError,
+            cur.executemany, "insert into test_exc values (%s)", buggygen())
+        cur.close()
+        conn.close()
+
+
+def test_suite():
+    return unittest.TestLoader().loadTestsFromName(__name__)
+
+if __name__ == "__main__":
+    unittest.main()