Ultra File Manager

Edit: /lib/python2.7/site-packages/pymysql/tests/test_nextset.py

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...

Ukuran: 1.8 KB