From 8363ac8607eca7398e568e1336154e1262a995a0 Mon Sep 17 00:00:00 2001 From: Erlend Egeberg Aasland Date: Fri, 4 Jun 2021 19:36:08 +0200 Subject: bpo-44041: Add test for sqlite3 column count (GH-25907) --- Lib/sqlite3/test/dbapi.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/Lib/sqlite3/test/dbapi.py b/Lib/sqlite3/test/dbapi.py index 77fafe0..4bda6aa 100644 --- a/Lib/sqlite3/test/dbapi.py +++ b/Lib/sqlite3/test/dbapi.py @@ -555,6 +555,17 @@ class CursorTests(unittest.TestCase): ] self.assertEqual(results, expected) + def test_column_count(self): + # Check that column count is updated correctly for cached statements + select = "select * from test" + res = self.cu.execute(select) + old_count = len(res.description) + # Add a new column and execute the cached select query again + self.cu.execute("alter table test add newcol") + res = self.cu.execute(select) + new_count = len(res.description) + self.assertEqual(new_count - old_count, 1) + class ThreadTests(unittest.TestCase): def setUp(self): -- cgit v0.12