Added test to verify executemany doesn't propagate exceptions.

This commit is contained in:
Daniele Varrazzo 2010-07-09 15:23:02 +01:00
parent 4a78e4068b
commit bd6a4139da
2 changed files with 31 additions and 0 deletions

View File

@ -37,6 +37,7 @@ import test_dates
import test_psycopg2_dbapi20 import test_psycopg2_dbapi20
import test_quote import test_quote
import test_connection import test_connection
import test_cursor
import test_transaction import test_transaction
import types_basic import types_basic
import types_extras import types_extras
@ -54,6 +55,7 @@ def test_suite():
suite.addTest(test_psycopg2_dbapi20.test_suite()) suite.addTest(test_psycopg2_dbapi20.test_suite())
suite.addTest(test_quote.test_suite()) suite.addTest(test_quote.test_suite())
suite.addTest(test_connection.test_suite()) suite.addTest(test_connection.test_suite())
suite.addTest(test_cursor.test_suite())
suite.addTest(test_transaction.test_suite()) suite.addTest(test_transaction.test_suite())
suite.addTest(types_basic.test_suite()) suite.addTest(types_basic.test_suite())
suite.addTest(types_extras.test_suite()) suite.addTest(types_extras.test_suite())

29
tests/test_cursor.py Normal file
View File

@ -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()