diff options
author | Petri Lehtinen <petri@digip.org> | 2011-05-09 12:24:09 +0200 |
---|---|---|
committer | Petri Lehtinen <petri@digip.org> | 2011-05-09 12:24:09 +0200 |
commit | 0ae448f24e2986e07a70a7763c1fe87a89181bbe (patch) | |
tree | 4cab9ef6d15627a925dd41bcabe8561fb47604ff /Lib/sqlite3 | |
parent | 2693f42ab8b17773b98a7ccc90e912f5d45fdd26 (diff) | |
download | cpython-0ae448f24e2986e07a70a7763c1fe87a89181bbe.tar.gz |
Issue #10811: Fix recursive usage of cursors. Instead of crashing, raise a ProgrammingError now.
Diffstat (limited to 'Lib/sqlite3')
-rw-r--r-- | Lib/sqlite3/test/regression.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/Lib/sqlite3/test/regression.py b/Lib/sqlite3/test/regression.py index 7d0553d8f1..c7551e35a1 100644 --- a/Lib/sqlite3/test/regression.py +++ b/Lib/sqlite3/test/regression.py @@ -281,6 +281,28 @@ class RegressionTests(unittest.TestCase): # Lone surrogate cannot be encoded to the default encoding (utf8) "\uDC80", collation_cb) + def CheckRecursiveCursorUse(self): + """ + http://bugs.python.org/issue10811 + + Recursively using a cursor, such as when reusing it from a generator led to segfaults. + Now we catch recursive cursor usage and raise a ProgrammingError. + """ + con = sqlite.connect(":memory:") + + cur = con.cursor() + cur.execute("create table a (bar)") + cur.execute("create table b (baz)") + + def foo(): + cur.execute("insert into a (bar) values (?)", (1,)) + yield 1 + + with self.assertRaises(sqlite.ProgrammingError): + cur.executemany("insert into b (baz) values (?)", + ((i,) for i in foo())) + + def suite(): regression_suite = unittest.makeSuite(RegressionTests, "Check") return unittest.TestSuite((regression_suite,)) |