def test_conpy175(self):
default_conf= conf()
c1 = mariadb.connect(**default_conf)
str= '"' * 4194304
newstr= c1.escape_string(str);
self.assertEqual(newstr, '\\"' * 4194304)
c1.close()
def test_change_user(self):
if is_skysql():
self.skipTest("SkySQL failure")
if self.connection.server_version < 100103:
self.skipTest("CREATE OR REPLACE USER not supported")
if is_maxscale():
self.skipTest("MAXSCALE doesn't get new user immediately")
if self.connection.server_name == "localhost":
curs = self.connection.cursor(buffered=True)
curs.execute(
"select * from information_schema.plugins where plugin_name='unix_socket' and plugin_status='ACTIVE'"
)
if curs.rowcount > 0:
del curs
self.skipTest("unix_socket is active")
del curs
default_conf = conf()
cursor = self.connection.cursor()
cursor.execute("drop user if exists foo")
if is_mysql() and self.connection.server_version < 80000:
cursor.execute("create user foo@'%'")
cursor.execute("GRANT ALL on `" + default_conf["database"] +
"`.* TO foo@'%' IDENTIFIED BY 'heyPassw-!µ20§rd'")
else:
cursor.execute(
"create user foo@'%' IDENTIFIED BY 'heyPassw-!µ20§rd'")
cursor.execute("GRANT ALL on `" + default_conf["database"] +
"`.* TO foo@'%'")
new_conn = create_connection()
new_conn.change_user("foo", "heyPassw-!µ20§rd", "")
self.assertEqual("foo", new_conn.user)
cursor.execute("drop user foo")
del new_conn
del cursor
def test_change_user(self):
if self.connection.server_version < 100103:
self.skipTest("CREATE OR REPLACE USER not supported")
if os.environ.get("MAXSCALE_VERSION"):
self.skipTest("MAXSCALE doesn't get new user immediately")
if self.connection.server_name == "localhost":
curs = self.connection.cursor(buffered=True)
curs.execute(
"select * from information_schema.plugins where plugin_name='unix_socket' and plugin_status='ACTIVE'"
)
if curs.rowcount > 0:
del curs
self.skipTest("unix_socket is active")
del curs
default_conf = conf()
cursor = self.connection.cursor()
cursor.execute("drop user if exists foo")
cursor.execute("create user foo@'%'")
cursor.execute("GRANT ALL on `" + default_conf["database"] +
"`.* TO foo@'%'")
new_conn = create_connection()
new_conn.change_user("foo", "", "")
self.assertEqual("foo", new_conn.user)
cursor.execute("drop user foo")
del new_conn
del cursor
def test_conpy36(self):
if platform.system() == "Windows":
self.skipTest("unix_socket not supported on Windows")
default_conf = conf()
try:
conn= mariadb.connect(user=default_conf["user"], unix_socket="/does_not_exist/x.sock", port=default_conf["port"], host=default_conf["host"])
except (mariadb.OperationalError,):
pass
def test_conpy36(self):
if platform.system() == "Windows":
self.skipTest("unix_socket not supported on Windows")
default_conf = conf()
try:
conn= mariadb.connect(user=default_conf["user"], unix_socket="/does_not_exist/x.sock")
except mariadb.DatabaseError:
pass
def test_compress(self):
default_conf= conf()
new_conn = mariadb.connect(user=default_conf["user"], database=default_conf["database"], compress=True)
cursor=new_conn.cursor()
cursor.execute("SHOW SESSION STATUS LIKE 'compression'")
row=cursor.fetchone()
self.assertEqual(row[1], "ON")
del cursor
del new_conn
def test_init_command(self):
default_conf= conf()
new_conn = mariadb.connect(user=default_conf["user"], database=default_conf["database"], init_command="SET @a:=1")
cursor=new_conn.cursor()
cursor.execute("SELECT @a")
row=cursor.fetchone()
self.assertEqual(row[0], 1)
del cursor
del new_conn
def test_init_command(self):
default_conf = conf()
new_conn = mariadb.connect(**default_conf, init_command="SET @a:=1")
#, port=default_conf["port"], host=default_conf["host"], password=default_conf["password"])
cursor = new_conn.cursor()
cursor.execute("SELECT @a")
row = cursor.fetchone()
self.assertEqual(row[0], 1)
del cursor
del new_conn
def test_local_infile(self):
default_conf= conf()
new_conn = mariadb.connect(user=default_conf["user"], database=default_conf["database"], local_infile=False)
cursor=new_conn.cursor()
cursor.execute("CREATE TEMPORARY TABLE t1 (a int)")
try:
cursor.execute("LOAD DATA LOCAL INFILE 'x.x' INTO TABLE t1")
except mariadb.ProgrammingError:
pass
del cursor
del new_conn
def test_local_infile(self):
default_conf = conf()
new_conn = mariadb.connect(**default_conf, local_infile=False)
cursor = new_conn.cursor()
cursor.execute("CREATE TEMPORARY TABLE t1 (a int)")
try:
cursor.execute("LOAD DATA LOCAL INFILE 'x.x' INTO TABLE t1")
except (mariadb.OperationalError, ):
pass
del cursor
del new_conn
def test_ed25519(self):
if is_skysql():
self.skipTest("Test fail on SkySQL")
default_conf = conf()
if os.environ.get("MAXSCALE_VERSION"):
self.skipTest("MAXSCALE doesn't support ed25519 for now")
if self.connection.server_version < 100122:
self.skipTest("ed25519 not supported")
conn = self.connection
curs = conn.cursor(buffered=True)
if self.connection.server_name == "localhost":
curs.execute(
"select * from information_schema.plugins where plugin_name='unix_socket' and plugin_status='ACTIVE'"
)
if curs.rowcount > 0:
del curs
self.skipTest("unix_socket is active")
cursor = conn.cursor()
try:
cursor.execute("INSTALL SONAME 'auth_ed25519'")
except (mariadb.DatabaseError, mariadb.OperationalError):
self.skipTest("Server couldn't load auth_ed25519")
cursor.execute("DROP USER IF EXISTS eduser")
if self.connection.server_version < 100400:
cursor.execute(
"CREATE USER eduser@'%' IDENTIFIED VIA ed25519 "
"USING '6aW9C7ENlasUfymtfMvMZZtnkCVlcb1ssxOLJ0kj/AA'")
else:
cursor.execute("CREATE USER eduser@'%' IDENTIFIED VIA ed25519 "
"USING PASSWORD('MySup8%rPassw@ord')")
cursor.execute("GRANT ALL on " + default_conf["database"] +
".* to eduser@'%'")
conn2 = create_connection({
"user": "******",
"password": "******"
})
cursor.execute("DROP USER IF EXISTS eduser")
try:
create_connection({
"user": "******",
"password": "******",
"plugin_dir": "wrong_plugin_dir"
})
self.fail(
"wrong plugin directory, must not have found authentication plugin"
)
except (mariadb.OperationalError):
pass
cursor.execute("DROP USER IF EXISTS eduser")
del cursor, conn2
def test_compress(self):
default_conf = conf()
new_conn = mariadb.connect(**default_conf, compress=True)
cursor = new_conn.cursor()
cursor.execute("SHOW SESSION STATUS LIKE 'compression'")
row = cursor.fetchone()
if os.environ.get("MAXSCALE_VERSION"):
self.assertEqual(row[1], "OFF")
else:
self.assertEqual(row[1], "ON")
del cursor
del new_conn
def test_conpy155(self):
default_conf= conf()
c1 = mariadb.connect(**default_conf)
version= c1.get_server_version()
self.assertEqual(c1.get_server_version(), version)
self.assertEqual(c1.get_server_version(), version)
self.assertEqual(c1.get_server_version(), version)
self.assertEqual(c1.get_server_version(), version)
c1.close()
try:
version= c1.get_server_version()
except mariadb.ProgrammingError:
pass
def test_local_infile(self):
default_conf = conf()
new_conn = mariadb.connect(user=default_conf["user"],
database=default_conf["database"],
local_infile=False,
port=default_conf["port"],
host=default_conf["host"])
cursor = new_conn.cursor()
cursor.execute("CREATE TEMPORARY TABLE t1 (a int)")
try:
cursor.execute("LOAD DATA LOCAL INFILE 'x.x' INTO TABLE t1")
except (mariadb.OperationalError, ):
pass
del cursor
del new_conn
def test_connection_default_file(self):
if os.path.exists("client.cnf"):
os.remove("client.cnf")
default_conf = conf()
f = open("client.cnf", "w+")
f.write("[client]\n")
f.write("host=%s\n" % default_conf["host"])
f.write("port=%i\n" % default_conf["port"])
f.write("database=%s\n" % default_conf["database"])
f.close()
new_conn = mariadb.connect(user=default_conf["user"], default_file="./client.cnf")
self.assertEqual(new_conn.database, default_conf["database"])
del new_conn
os.remove("client.cnf")
def test_compress(self):
default_conf = conf()
new_conn = mariadb.connect(user=default_conf["user"],
database=default_conf["database"],
compress=True,
port=default_conf["port"],
host=default_conf["host"])
cursor = new_conn.cursor()
cursor.execute("SHOW SESSION STATUS LIKE 'compression'")
row = cursor.fetchone()
if os.environ.get("MAXSCALE_VERSION"):
self.assertEqual(row[1], "OFF")
else:
self.assertEqual(row[1], "ON")
del cursor
del new_conn
def test_schema(self):
if self.connection.server_version < 100103:
self.skipTest("CREATE OR REPLACE SCHEMA not supported")
if self.connection.server_version < 100202:
self.skipTest("session tracking not supported")
if os.environ.get("MAXSCALE_VERSION"):
self.skipTest("MAXSCALE doesn't tell schema change for now")
default_conf = conf()
conn = self.connection
self.assertEqual(conn.database, default_conf["database"])
cursor = conn.cursor()
cursor.execute("CREATE OR REPLACE SCHEMA test1")
cursor.execute("USE test1")
self.assertEqual(conn.database, "test1")
conn.database = default_conf["database"]
self.assertEqual(conn.database, default_conf["database"])
def test_conpy101(self):
default_conf = conf()
c1 = mariadb.connect(user=default_conf["user"],
database=default_conf["database"],
port=default_conf["port"],
host=default_conf["host"])
self.assertEqual(c1.autocommit, False)
c1 = mariadb.connect(user=default_conf["user"],
database=default_conf["database"],
port=default_conf["port"],
host=default_conf["host"],
autocommit=False)
self.assertEqual(c1.autocommit, False)
c1 = mariadb.connect(user=default_conf["user"],
database=default_conf["database"],
port=default_conf["port"],
host=default_conf["host"],
autocommit=True)
self.assertEqual(c1.autocommit, True)
def main():
default_conf = conf()
conn = dbdrv.connect(**default_conf)
init_db(conn, dbdrv.paramstyle)
conn.close()
def main():
default_conf = conf()
conn = dbdrv.connect(**default_conf)
run_test(test_suite(), conn)
conn.close()
class DatabaseAPI20Test(unittest.TestCase):
''' Test a database self.driver for DB API 2.0 compatibility.
This implementation tests Gadfly, but the TestCase
is structured so that other self.drivers can subclass this
test case to ensure compiliance with the DB-API. It is
expected that this TestCase may be expanded in the future
if ambiguities or edge conditions are discovered.
The 'Optional Extensions' are not yet being tested.
self.drivers should subclass this test, overriding setUp, tearDown,
self.driver, connect_args and connect_kw_args. Class specification
should be as follows:
import dbapi20
class mytest(dbapi20.DatabaseAPI20Test):
[...]
Don't 'import DatabaseAPI20Test from dbapi20', or you will
confuse the unit tester - just 'import dbapi20'.
'''
# The self.driver module. This should be the module where the 'connect'
# method is to be found
driver = mariadb
connect_args = () # List of arguments to pass to connect
connect_kw_args = conf() # Keyword arguments for connect
table_prefix = 'dbapi20test_' # If you need to specify a prefix for tables
ddl1 = 'create table %sbooze (name varchar(20))' % table_prefix
ddl2 = 'create table %sbarflys (name varchar(20))' % table_prefix
xddl1 = 'drop table %sbooze' % table_prefix
xddl2 = 'drop table %sbarflys' % table_prefix
lowerfunc = 'lower' # Name of stored procedure to convert string->lowercase
# Some drivers may need to override these helpers, for example adding
# a 'commit' after the execute.
def executeDDL1(self, cursor):
cursor.execute(self.ddl1)
def executeDDL2(self, cursor):
cursor.execute(self.ddl2)
def setUp(self):
''' self.drivers should override this method to perform required setup
if any is necessary, such as creating the database.
'''
pass
def tearDown(self):
''' self.drivers should override this method to perform required cleanup
if any is necessary, such as deleting the test database.
The default drops the tables that may be created.
'''
con = self._connect()
try:
cur = con.cursor()
for ddl in (self.xddl1, self.xddl2):
try:
cur.execute(ddl)
con.commit()
except self.driver.Error:
# Assume table didn't exist. Other tests will check if
# execute is busted.
pass
finally:
con.close()
def _connect(self):
try:
return self.driver.connect(**self.connect_kw_args)
except AttributeError:
self.fail("No connect method found in self.driver module")
def test_connect(self):
con = self._connect()
con.close()
def test_apilevel(self):
try:
# Must exist
apilevel = self.driver.apilevel
# Must equal 2.0
self.assertEqual(apilevel, '2.0')
except AttributeError:
self.fail("Driver doesn't define apilevel")
def test_threadsafety(self):
try:
# Must exist
threadsafety = self.driver.threadsafety
# Must be a valid value
self.assertTrue(threadsafety in (0, 1, 2, 3))
except AttributeError:
self.fail("Driver doesn't define threadsafety")
def test_paramstyle(self):
try:
# Must exist
paramstyle = self.driver.paramstyle
# Must be a valid value
self.assertTrue(paramstyle in (
'qmark', 'numeric', 'named', 'format', 'pyformat'
))
except AttributeError:
self.fail("Driver doesn't define paramstyle")
def test_Exceptions(self):
# Make sure required exceptions exist, and are in the
# defined hierarchy.
self.assertTrue(issubclass(self.driver.Warning, Exception))
self.assertTrue(issubclass(self.driver.Error, Exception))
self.assertTrue(
issubclass(self.driver.InterfaceError, self.driver.Error)
)
self.assertTrue(
issubclass(self.driver.DatabaseError, self.driver.Error)
)
self.assertTrue(
issubclass(self.driver.OperationalError, self.driver.Error)
)
self.assertTrue(
issubclass(self.driver.IntegrityError, self.driver.Error)
)
self.assertTrue(
issubclass(self.driver.InternalError, self.driver.Error)
)
self.assertTrue(
issubclass(self.driver.ProgrammingError, self.driver.Error)
)
self.assertTrue(
issubclass(self.driver.NotSupportedError, self.driver.Error)
)
def test_ExceptionsAsConnectionAttributes(self):
# OPTIONAL EXTENSION
# Test for the optional DB API 2.0 extension, where the exceptions
# are exposed as attributes on the Connection object
# I figure this optional extension will be implemented by any
# driver author who is using this test suite, so it is enabled
# by default.
con = self._connect()
drv = self.driver
self.assertTrue(con.Warning is drv.Warning)
self.assertTrue(con.Error is drv.Error)
self.assertTrue(con.InterfaceError is drv.InterfaceError)
self.assertTrue(con.DatabaseError is drv.DatabaseError)
self.assertTrue(con.OperationalError is drv.OperationalError)
self.assertTrue(con.IntegrityError is drv.IntegrityError)
self.assertTrue(con.InternalError is drv.InternalError)
self.assertTrue(con.ProgrammingError is drv.ProgrammingError)
self.assertTrue(con.NotSupportedError is drv.NotSupportedError)
def test_commit(self):
con = self._connect()
try:
# Commit must work, even if it doesn't do anything
con.commit()
finally:
con.close()
def test_rollback(self):
con = self._connect()
# If rollback is defined, it should either work or throw
# the documented exception
if hasattr(con, 'rollback'):
try:
con.rollback()
except self.driver.NotSupportedError:
pass
def test_cursor(self):
con = self._connect()
try:
cur = con.cursor()
finally:
con.close()
def test_cursor_isolation(self):
con = self._connect()
try:
# Make sure cursors created from the same connection have
# the documented transaction isolation level
cur1 = con.cursor()
cur2 = con.cursor()
self.executeDDL1(cur1)
cur1.execute("insert into %sbooze values ('Victoria Bitter')" % (
self.table_prefix
))
cur2.execute("select name from %sbooze" % self.table_prefix)
booze = cur2.fetchall()
self.assertEqual(len(booze), 1)
self.assertEqual(len(booze[0]), 1)
self.assertEqual(booze[0][0], 'Victoria Bitter')
finally:
con.close()
def test_description(self):
con = self._connect()
try:
cur = con.cursor()
self.executeDDL1(cur)
self.assertEqual(cur.description, None,
'cursor.description should be none after executing a '
'statement that can return no rows (such as DDL)'
)
cur.execute('select name from %sbooze' % self.table_prefix)
self.assertEqual(len(cur.description), 1,
'cursor.description describes too many columns'
)
self.assertEqual(len(cur.description[0]), 8,
'cursor.description[x] tuples must have 8 elements'
)
self.assertEqual(cur.description[0][0].lower(), 'name',
'cursor.description[x][0] must return column name'
)
self.assertEqual(cur.description[0][1], self.driver.STRING,
'cursor.description[x][1] must return column type. Got %r'
% cur.description[0][1]
)
# Make sure self.description gets reset
self.executeDDL2(cur)
self.assertEqual(cur.description, None,
'cursor.description not being set to None when executing '
'no-result statements (eg. DDL)'
)
finally:
con.close()
def test_rowcount(self):
con = self._connect()
try:
cur = con.cursor(buffered=True)
self.executeDDL1(cur)
self.assertEqual(cur.rowcount, -1,
'cursor.rowcount should be -1 after executing no-result '
'statements'
)
cur.execute("insert into %sbooze values ('Victoria Bitter')" % (
self.table_prefix
))
self.assertTrue(cur.rowcount in (-1, 1),
'cursor.rowcount should == number or rows inserted, or '
'set to -1 after executing an insert statement'
)
cur.execute("select name from %sbooze" % self.table_prefix)
self.assertTrue(cur.rowcount in (-1, 1),
'cursor.rowcount should == number of rows returned, or '
'set to -1 after executing a select statement'
)
self.executeDDL2(cur)
self.assertEqual(cur.rowcount, -1,
'cursor.rowcount not being reset to -1 after executing '
'no-result statements'
)
finally:
con.close()
lower_func = 'lower'
def test_close(self):
con = self._connect()
try:
cur = con.cursor()
finally:
con.close()
# cursor.execute should raise an Error if called after connection
# closed
self.assertRaises(self.driver.Error, self.executeDDL1, cur)
# connection.commit should raise an Error if called after connection'
# closed.'
self.assertRaises(self.driver.Error, con.commit)
# connection.close should raise an Error if called more than once
self.assertRaises(self.driver.Error, con.close)
def test_execute(self):
con = self._connect()
try:
cur = con.cursor()
self._paraminsert(cur)
finally:
con.close()
def _paraminsert(self, cur):
self.executeDDL1(cur)
cur.execute("insert into %sbooze values ('Victoria Bitter')" % (
self.table_prefix
))
self.assertTrue(cur.rowcount in (-1, 1))
if self.driver.paramstyle == 'qmark':
cur.execute(
'insert into %sbooze values (?)' % self.table_prefix,
("Cooper's",)
)
elif self.driver.paramstyle == 'numeric':
cur.execute(
'insert into %sbooze values (:1)' % self.table_prefix,
("Cooper's",)
)
elif self.driver.paramstyle == 'named':
cur.execute(
'insert into %sbooze values (:beer)' % self.table_prefix,
{'beer': "Cooper's"}
)
elif self.driver.paramstyle == 'format':
cur.execute(
'insert into %sbooze values (%%s)' % self.table_prefix,
("Cooper's",)
)
elif self.driver.paramstyle == 'pyformat':
cur.execute(
'insert into %sbooze values (%%(beer)s)' % self.table_prefix,
{'beer': "Cooper's"}
)
else:
self.fail('Invalid paramstyle')
self.assertTrue(cur.rowcount in (-1, 1))
cur.execute('select name from %sbooze' % self.table_prefix)
res = cur.fetchall()
self.assertEqual(len(res), 2, 'cursor.fetchall returned too few rows')
beers = [res[0][0], res[1][0]]
beers.sort()
self.assertEqual(beers[0], "Cooper's",
'cursor.fetchall retrieved incorrect data, or data inserted '
'incorrectly'
)
self.assertEqual(beers[1], "Victoria Bitter",
'cursor.fetchall retrieved incorrect data, or data inserted '
'incorrectly'
)
def test_executemany(self):
if os.environ.get("MAXSCALE_VERSION"):
self.skipTest("MAXSCALE doesn't support BULK yet")
con = self._connect()
try:
cur = con.cursor()
self.executeDDL1(cur)
largs = [("Cooper's",), ("Boag's",)]
margs = [{'beer': "Cooper's"}, {'beer': "Boag's"}]
if self.driver.paramstyle == 'qmark':
cur.executemany(
'insert into %sbooze values (?)' % self.table_prefix,
largs
)
elif self.driver.paramstyle == 'numeric':
cur.executemany(
'insert into %sbooze values (:1)' % self.table_prefix,
largs
)
elif self.driver.paramstyle == 'named':
cur.executemany(
'insert into %sbooze values (:beer)' % self.table_prefix,
margs
)
elif self.driver.paramstyle == 'format':
cur.executemany(
'insert into %sbooze values (%%s)' % self.table_prefix,
largs
)
elif self.driver.paramstyle == 'pyformat':
cur.executemany(
'insert into %sbooze values (%%(beer)s)' % (
self.table_prefix
),
margs
)
else:
self.fail('Unknown paramstyle')
self.assertTrue(cur.rowcount in (-1, 2),
'insert using cursor.executemany set cursor.rowcount to '
'incorrect value %r' % cur.rowcount
)
cur.execute('select name from %sbooze' % self.table_prefix)
res = cur.fetchall()
self.assertEqual(len(res), 2,
'cursor.fetchall retrieved incorrect number of rows'
)
beers = [res[0][0], res[1][0]]
beers.sort()
self.assertEqual(beers[0], "Boag's", 'incorrect data retrieved')
self.assertEqual(beers[1], "Cooper's", 'incorrect data retrieved')
finally:
con.close()
def test_fetchone(self):
con = self._connect()
try:
cur = con.cursor(buffered=True)
# cursor.fetchone should raise an Error if called before
# executing a select-type query
self.assertRaises(self.driver.Error, cur.fetchone)
# cursor.fetchone should raise an Error if called after
# executing a query that cannot return rows
self.executeDDL1(cur)
self.assertRaises(self.driver.Error, cur.fetchone)
cur.execute('select name from %sbooze' % self.table_prefix)
self.assertEqual(cur.fetchone(), None,
'cursor.fetchone should return None if a query retrieves '
'no rows'
)
self.assertTrue(cur.rowcount in (-1, 0))
# cursor.fetchone should raise an Error if called after
# executing a query that cannot return rows
cur.execute("insert into %sbooze values ('Victoria Bitter')" % (
self.table_prefix
))
self.assertRaises(self.driver.Error, cur.fetchone)
cur.execute('select name from %sbooze' % self.table_prefix)
r = cur.fetchone()
self.assertEqual(len(r), 1,
'cursor.fetchone should have retrieved a single row'
)
self.assertEqual(r[0], 'Victoria Bitter',
'cursor.fetchone retrieved incorrect data'
)
self.assertEqual(cur.fetchone(), None,
'cursor.fetchone should return None if no more rows available'
)
self.assertTrue(cur.rowcount in (-1, 1))
finally:
con.close()
samples = [
'Carlton Cold',
'Carlton Draft',
'Mountain Goat',
'Redback',
'Victoria Bitter',
'XXXX'
]
def _populate(self):
''' Return a list of sql commands to setup the DB for the fetch
tests.
'''
populate = [
"insert into %sbooze values ('%s')" % (self.table_prefix, s)
for s in self.samples
]
return populate
def test_fetchmany(self):
con = self._connect()
try:
cur = con.cursor()
# cursor.fetchmany should raise an Error if called without
# issuing a query
self.assertRaises(self.driver.Error, cur.fetchmany, 4)
self.executeDDL1(cur)
for sql in self._populate():
cur.execute(sql)
cur.execute('select name from %sbooze' % self.table_prefix)
r = cur.fetchmany()
self.assertEqual(len(r), 1,
'cursor.fetchmany retrieved incorrect number of rows, '
'default of arraysize is one.'
)
cur.arraysize = 10
r = cur.fetchmany(3) # Should get 3 rows
self.assertEqual(len(r), 3,
'cursor.fetchmany retrieved incorrect number of rows'
)
r = cur.fetchmany(4) # Should get 2 more
self.assertEqual(len(r), 2,
'cursor.fetchmany retrieved incorrect number of rows'
)
r = cur.fetchmany(4) # Should be an empty sequence
self.assertEqual(len(r), 0,
'cursor.fetchmany should return an empty sequence after '
'results are exhausted'
)
self.assertTrue(cur.rowcount in (-1, 6))
# Same as above, using cursor.arraysize
cur.arraysize = 4
cur.execute('select name from %sbooze' % self.table_prefix)
r = cur.fetchmany() # Should get 4 rows
self.assertEqual(len(r), 4,
'cursor.arraysize not being honoured by fetchmany'
)
r = cur.fetchmany() # Should get 2 more
self.assertEqual(len(r), 2)
r = cur.fetchmany() # Should be an empty sequence
self.assertEqual(len(r), 0)
self.assertTrue(cur.rowcount in (-1, 6))
cur.arraysize = 6
cur.execute('select name from %sbooze' % self.table_prefix)
rows = cur.fetchmany() # Should get all rows
self.assertTrue(cur.rowcount in (-1, 6))
self.assertEqual(len(rows), 6)
self.assertEqual(len(rows), 6)
rows = [r[0] for r in rows]
rows.sort()
# Make sure we get the right data back out
for i in range(0, 6):
self.assertEqual(rows[i], self.samples[i],
'incorrect data retrieved by cursor.fetchmany'
)
rows = cur.fetchmany() # Should return an empty list
self.assertEqual(len(rows), 0,
'cursor.fetchmany should return an empty sequence if '
'called after the whole result set has been fetched'
)
self.assertTrue(cur.rowcount in (-1, 6))
self.executeDDL2(cur)
cur.execute('select name from %sbarflys' % self.table_prefix)
r = cur.fetchmany() # Should get empty sequence
self.assertEqual(len(r), 0,
'cursor.fetchmany should return an empty sequence if '
'query retrieved no rows'
)
self.assertTrue(cur.rowcount in (-1, 0))
finally:
con.close()
def test_fetchall(self):
con = self._connect()
try:
cur = con.cursor()
# cursor.fetchall should raise an Error if called
# without executing a query that may return rows (such
# as a select)
self.assertRaises(self.driver.Error, cur.fetchall)
self.executeDDL1(cur)
for sql in self._populate():
cur.execute(sql)
# cursor.fetchall should raise an Error if called
# after executing a a statement that cannot return rows
self.assertRaises(self.driver.Error, cur.fetchall)
cur.execute('select name from %sbooze' % self.table_prefix)
rows = cur.fetchall()
self.assertTrue(cur.rowcount in (-1, len(self.samples)))
self.assertEqual(len(rows), len(self.samples),
'cursor.fetchall did not retrieve all rows'
)
rows = [r[0] for r in rows]
rows.sort()
for i in range(0, len(self.samples)):
self.assertEqual(rows[i], self.samples[i],
'cursor.fetchall retrieved incorrect rows'
)
rows = cur.fetchall()
self.assertEqual(
len(rows), 0,
'cursor.fetchall should return an empty list if called '
'after the whole result set has been fetched'
)
self.assertTrue(cur.rowcount in (-1, len(self.samples)))
self.executeDDL2(cur)
cur.execute('select name from %sbarflys' % self.table_prefix)
rows = cur.fetchall()
self.assertTrue(cur.rowcount in (-1, 0))
self.assertEqual(len(rows), 0,
'cursor.fetchall should return an empty list if '
'a select query returns no rows'
)
finally:
con.close()
def test_mixedfetch(self):
con = self._connect()
try:
cur = con.cursor()
self.executeDDL1(cur)
for sql in self._populate():
cur.execute(sql)
cur.execute('select name from %sbooze' % self.table_prefix)
rows1 = cur.fetchone()
rows23 = cur.fetchmany(2)
rows4 = cur.fetchone()
rows56 = cur.fetchall()
self.assertTrue(cur.rowcount in (-1, 6))
self.assertEqual(len(rows23), 2,
'fetchmany returned incorrect number of rows'
)
self.assertEqual(len(rows56), 2,
'fetchall returned incorrect number of rows'
)
rows = [rows1[0]]
rows.extend([rows23[0][0], rows23[1][0]])
rows.append(rows4[0])
rows.extend([rows56[0][0], rows56[1][0]])
rows.sort()
for i in range(0, len(self.samples)):
self.assertEqual(rows[i], self.samples[i],
'incorrect data retrieved or inserted'
)
finally:
con.close()
def help_nextset_setUp(self, cur):
''' Should create a procedure called deleteme
that returns two result sets, first the
number of rows in booze then "name from booze"
'''
raise NotImplementedError('Helper not implemented')
# sql="""
# create procedure deleteme as
# begin
# select count(*) from booze
# select name from booze
# end
# """
# cur.execute(sql)
def help_nextset_tearDown(self, cur):
'If cleaning up is needed after nextSetTest'
raise NotImplementedError('Helper not implemented')
# cur.execute("drop procedure deleteme")
def test_arraysize(self):
# Not much here - rest of the tests for this are in test_fetchmany
con = self._connect()
try:
cur = con.cursor()
self.assertTrue(hasattr(cur, 'arraysize'),
'cursor.arraysize must be defined'
)
finally:
con.close()
def test_setinputsizes(self):
con = self._connect()
try:
cur = con.cursor()
cur.setinputsizes((25,))
self._paraminsert(cur) # Make sure cursor still works
finally:
con.close()
def test_None(self):
con = self._connect()
try:
cur = con.cursor()
self.executeDDL1(cur)
cur.execute('insert into %sbooze values (NULL)' % self.table_prefix)
cur.execute('select name from %sbooze' % self.table_prefix)
r = cur.fetchall()
self.assertEqual(len(r), 1)
self.assertEqual(len(r[0]), 1)
self.assertEqual(r[0][0], None, 'NULL value not returned as None')
finally:
con.close()
def main():
default_conf = conf()
conn = mysql.connector.connect(**default_conf)
run_test(test_suite(), conn)
conn.close()
def test_conpy101(self):
default_conf = conf()
c1 = mariadb.connect(**default_conf)
self.assertEqual(c1.autocommit, False)
c1 = mariadb.connect(**default_conf, autocommit=True)
self.assertEqual(c1.autocommit, True)
def main():
default_conf = conf()
conn = dbdrv.connect(**default_conf)
run_test(test_suite(dbdrv.paramstyle), conn, dbdrv.paramstyle)
conn.close()
def main():
default_conf = conf()
conn = dbdrv.connect(**default_conf)
init_db(conn)
conn.close()
