psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-02-07 12:50:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

write test for fetch_result flag

Browse Source
This commit is contained in:
Ivan Kotelnikov 2018-11-22 12:30:37 +02:00
parent 9a8d7d5a6a
commit e2b99d3a8e
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_fast_executemany.py
View File

@ -229,6 +229,15 @@ class TestExecuteValues(FastExecuteTestMixin, testutils.ConnectingTestCase):
cur.execute("select id, data from testfast where id = 3") cur.execute("select id, data from testfast where id = 3")
self.assertEqual(cur.fetchone(), (3, snowman)) self.assertEqual(cur.fetchone(), (3, snowman))
def test_returning(self):
cur = self.conn.cursor()
result = psycopg2.extras.execute_values(cur,
"insert into testfast (id, val) values %s returning id",
((i, i * 10) for i in range(25)),
page_size=10, fetch_many=True)
# result contains all returned pages
self.assertEqual([r[0] for r in result], list(range(25)))
def test_invalid_sql(self): def test_invalid_sql(self):
cur = self.conn.cursor() cur = self.conn.cursor()
self.assertRaises(ValueError, psycopg2.extras.execute_values, cur, self.assertRaises(ValueError, psycopg2.extras.execute_values, cur,