psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-06-13 09:33:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merged tests from piro and jan async fixes branches

Browse Source
This commit is contained in:
Federico Di Gregorio 2010-04-14 10:01:37 +02:00
parent 5f4ef5da13
commit 249b3ef88f
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_async.py
View File

@ -301,7 +301,6 @@ class AsyncTests(unittest.TestCase):
curs = self.conn.cursor() curs = self.conn.cursor()
for mb in 1, 5, 10, 20, 50: for mb in 1, 5, 10, 20, 50:
size = mb * 1024 * 1024 size = mb * 1024 * 1024
print "\nplease wait: sending", mb, "MB query to the server",
stub = PollableStub(curs) stub = PollableStub(curs)
curs.execute("select %s;", ('x' * size,)) curs.execute("select %s;", ('x' * size,))
self.wait(stub) self.wait(stub)
@ -324,7 +323,7 @@ class AsyncTests(unittest.TestCase):
# polling a cursor that's not currently executing is an error # polling a cursor that's not currently executing is an error
self.assertRaises(psycopg2.ProgrammingError, cur2.poll) self.assertRaises(psycopg2.ProgrammingError, cur2.poll)
self.wait_for_query(cur1) self.wait(cur1)
self.assertEquals(cur1.fetchone()[0], 1) self.assertEquals(cur1.fetchone()[0], 1)
def test_async_fetch_wrong_cursor(self): def test_async_fetch_wrong_cursor(self):
@ -332,7 +331,7 @@ class AsyncTests(unittest.TestCase):
cur2 = self.conn.cursor() cur2 = self.conn.cursor()
cur1.execute("select 1") cur1.execute("select 1")
self.wait_for_query(cur1) self.wait(cur1)
self.assertFalse(self.conn.executing()) self.assertFalse(self.conn.executing())
# fetching from a cursor with no results is an error # fetching from a cursor with no results is an error
self.assertRaises(psycopg2.ProgrammingError, cur2.fetchone) self.assertRaises(psycopg2.ProgrammingError, cur2.fetchone)