Python wait_select Examples

Example #1

File: test_cancel.py Project: youguanxinqing/psycopg2cffi

 def test_async_cancel(self):
     async_conn = psycopg2.connect(dsn, async_=True)
     self.assertRaises(psycopg2.OperationalError, async_conn.cancel)
     extras.wait_select(async_conn)
     cur = async_conn.cursor()
     cur.execute("select pg_sleep(10000)")
     self.assertTrue(async_conn.isexecuting())
     async_conn.cancel()
     self.assertRaises(extensions.QueryCanceledError, extras.wait_select,
                       async_conn)
     cur.execute("select 1")
     extras.wait_select(async_conn)
     self.assertEqual(cur.fetchall(), [(1, )])

Example #2

File: test_cancel.py Project: chtd/psycopg2cffi

 def test_async_cancel(self):
     async_conn = psycopg2.connect(dsn, async_=True)
     self.assertRaises(psycopg2.OperationalError, async_conn.cancel)
     extras.wait_select(async_conn)
     cur = async_conn.cursor()
     cur.execute("select pg_sleep(10000)")
     self.assertTrue(async_conn.isexecuting())
     async_conn.cancel()
     self.assertRaises(extensions.QueryCanceledError,
                       extras.wait_select, async_conn)
     cur.execute("select 1")
     extras.wait_select(async_conn)
     self.assertEqual(cur.fetchall(), [(1, )])

Example #3

File: test_green.py Project: youguanxinqing/psycopg2cffi

 def set_stub_wait_callback(self, conn):
     stub = ConnectionStub(conn)
     extensions.set_wait_callback(lambda conn: extras.wait_select(stub))
     return stub

Example #4

File: test_green.py Project: chtd/psycopg2cffi

 def set_stub_wait_callback(self, conn):
     stub = ConnectionStub(conn)
     extensions.set_wait_callback(lambda conn: extras.wait_select(stub))
     return stub