From: Erlend Egeberg Aasland <erlend.aasland@innova.no>
Date: Fri, 4 Jun 2021 17:36:08 +0000 (+0200)
Subject: bpo-44041: Add test for sqlite3 column count (GH-25907)
X-Git-Tag: v3.11.0a1~942
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=8363ac8607eca7398e568e1336154e1262a995a0;p=thirdparty%2FPython%2Fcpython.git

bpo-44041: Add test for sqlite3 column count (GH-25907)
---

diff --git a/Lib/sqlite3/test/dbapi.py b/Lib/sqlite3/test/dbapi.py
index 77fafe093002..4bda6aa393e3 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):