69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
|
import unittest2
|
||
|
|
|
||
|
|
from pymysql.tests import base
|
||
|
|
from pymysql import util
|
||
|
|
|
||
|
|
|
||
|
|
class TestNextset(base.PyMySQLTestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(TestNextset, self).setUp()
|
||
|
|
self.con = self.connections[0]
|
||
|
|
|
||
|
|
def test_nextset(self):
|
||
|
|
cur = self.con.cursor()
|
||
|
|
cur.execute("SELECT 1; SELECT 2;")
|
||
|
|
self.assertEqual([(1,)], list(cur))
|
||
|
|
|
||
|
|
r = cur.nextset()
|
||
|
|
self.assertTrue(r)
|
||
|
|
|
||
|
|
self.assertEqual([(2,)], list(cur))
|
||
|
|
self.assertIsNone(cur.nextset())
|
||
|
|
|
||
|
|
def test_skip_nextset(self):
|
||
|
|
cur = self.con.cursor()
|
||
|
|
cur.execute("SELECT 1; SELECT 2;")
|
||
|
|
self.assertEqual([(1,)], list(cur))
|
||
|
|
|
||
|
|
cur.execute("SELECT 42")
|
||
|
|
self.assertEqual([(42,)], list(cur))
|
||
|
|
|
||
|
|
def test_ok_and_next(self):
|
||
|
|
cur = self.con.cursor()
|
||
|
|
cur.execute("SELECT 1; commit; SELECT 2;")
|
||
|
|
self.assertEqual([(1,)], list(cur))
|
||
|
|
self.assertTrue(cur.nextset())
|
||
|
|
self.assertTrue(cur.nextset())
|
||
|
|
self.assertEqual([(2,)], list(cur))
|
||
|
|
self.assertFalse(bool(cur.nextset()))
|
||
|
|
|
||
|
|
@unittest2.expectedFailure
|
||
|
|
def test_multi_cursor(self):
|
||
|
|
cur1 = self.con.cursor()
|
||
|
|
cur2 = self.con.cursor()
|
||
|
|
|
||
|
|
cur1.execute("SELECT 1; SELECT 2;")
|
||
|
|
cur2.execute("SELECT 42")
|
||
|
|
|
||
|
|
self.assertEqual([(1,)], list(cur1))
|
||
|
|
self.assertEqual([(42,)], list(cur2))
|
||
|
|
|
||
|
|
r = cur1.nextset()
|
||
|
|
self.assertTrue(r)
|
||
|
|
|
||
|
|
self.assertEqual([(2,)], list(cur1))
|
||
|
|
self.assertIsNone(cur1.nextset())
|
||
|
|
|
||
|
|
def test_multi_statement_warnings(self):
|
||
|
|
cursor = self.con.cursor()
|
||
|
|
|
||
|
|
try:
|
||
|
|
cursor.execute('DROP TABLE IF EXISTS a; '
|
||
|
|
'DROP TABLE IF EXISTS b;')
|
||
|
|
except TypeError:
|
||
|
|
self.fail()
|
||
|
|
|
||
|
|
#TODO: How about SSCursor and nextset?
|
||
|
|
# It's very hard to implement correctly...
|