mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
194 lines
8.2 KiB
Python
194 lines
8.2 KiB
Python
from pymysql.tests import base
|
|
from pymysql import util
|
|
|
|
import time
|
|
import datetime
|
|
|
|
class TestConversion(base.PyMySQLTestCase):
|
|
def test_datatypes(self):
|
|
""" test every data type """
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
c.execute("create table test_datatypes (b bit, i int, l bigint, f real, s varchar(32), u varchar(32), bb blob, d date, dt datetime, ts timestamp, td time, t time, st datetime)")
|
|
try:
|
|
# insert values
|
|
v = (True, -3, 123456789012, 5.7, "hello'\" world", u"Espa\xc3\xb1ol", "binary\x00data".encode(conn.charset), datetime.date(1988,2,2), datetime.datetime.now(), datetime.timedelta(5,6), datetime.time(16,32), time.localtime())
|
|
c.execute("insert into test_datatypes (b,i,l,f,s,u,bb,d,dt,td,t,st) values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)", v)
|
|
c.execute("select b,i,l,f,s,u,bb,d,dt,td,t,st from test_datatypes")
|
|
r = c.fetchone()
|
|
self.assertEqual(util.int2byte(1), r[0])
|
|
self.assertEqual(v[1:8], r[1:8])
|
|
# mysql throws away microseconds so we need to check datetimes
|
|
# specially. additionally times are turned into timedeltas.
|
|
self.assertEqual(datetime.datetime(*v[8].timetuple()[:6]), r[8])
|
|
self.assertEqual(v[9], r[9]) # just timedeltas
|
|
self.assertEqual(datetime.timedelta(0, 60 * (v[10].hour * 60 + v[10].minute)), r[10])
|
|
self.assertEqual(datetime.datetime(*v[-1][:6]), r[-1])
|
|
|
|
c.execute("delete from test_datatypes")
|
|
|
|
# check nulls
|
|
c.execute("insert into test_datatypes (b,i,l,f,s,u,bb,d,dt,td,t,st) values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)", [None] * 12)
|
|
c.execute("select b,i,l,f,s,u,bb,d,dt,td,t,st from test_datatypes")
|
|
r = c.fetchone()
|
|
self.assertEqual(tuple([None] * 12), r)
|
|
|
|
c.execute("delete from test_datatypes")
|
|
|
|
# check sequence type
|
|
c.execute("insert into test_datatypes (i, l) values (2,4), (6,8), (10,12)")
|
|
c.execute("select l from test_datatypes where i in %s order by i", ((2,6),))
|
|
r = c.fetchall()
|
|
self.assertEqual(((4,),(8,)), r)
|
|
finally:
|
|
c.execute("drop table test_datatypes")
|
|
|
|
def test_dict(self):
|
|
""" test dict escaping """
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
c.execute("create table test_dict (a integer, b integer, c integer)")
|
|
try:
|
|
c.execute("insert into test_dict (a,b,c) values (%(a)s, %(b)s, %(c)s)", {"a":1,"b":2,"c":3})
|
|
c.execute("select a,b,c from test_dict")
|
|
self.assertEqual((1,2,3), c.fetchone())
|
|
finally:
|
|
c.execute("drop table test_dict")
|
|
|
|
def test_string(self):
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
c.execute("create table test_dict (a text)")
|
|
test_value = "I am a test string"
|
|
try:
|
|
c.execute("insert into test_dict (a) values (%s)", test_value)
|
|
c.execute("select a from test_dict")
|
|
self.assertEqual((test_value,), c.fetchone())
|
|
finally:
|
|
c.execute("drop table test_dict")
|
|
|
|
def test_integer(self):
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
c.execute("create table test_dict (a integer)")
|
|
test_value = 12345
|
|
try:
|
|
c.execute("insert into test_dict (a) values (%s)", test_value)
|
|
c.execute("select a from test_dict")
|
|
self.assertEqual((test_value,), c.fetchone())
|
|
finally:
|
|
c.execute("drop table test_dict")
|
|
|
|
|
|
def test_big_blob(self):
|
|
""" test tons of data """
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
c.execute("create table test_big_blob (b blob)")
|
|
try:
|
|
data = "pymysql" * 1024
|
|
c.execute("insert into test_big_blob (b) values (%s)", (data,))
|
|
c.execute("select b from test_big_blob")
|
|
self.assertEqual(data.encode(conn.charset), c.fetchone()[0])
|
|
finally:
|
|
c.execute("drop table test_big_blob")
|
|
|
|
class TestCursor(base.PyMySQLTestCase):
|
|
# this test case does not work quite right yet, however,
|
|
# we substitute in None for the erroneous field which is
|
|
# compatible with the DB-API 2.0 spec and has not broken
|
|
# any unit tests for anything we've tried.
|
|
|
|
#def test_description(self):
|
|
# """ test description attribute """
|
|
# # result is from MySQLdb module
|
|
# r = (('Host', 254, 11, 60, 60, 0, 0),
|
|
# ('User', 254, 16, 16, 16, 0, 0),
|
|
# ('Password', 254, 41, 41, 41, 0, 0),
|
|
# ('Select_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Insert_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Update_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Delete_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Create_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Drop_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Reload_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Shutdown_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Process_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('File_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Grant_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('References_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Index_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Alter_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Show_db_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Super_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Create_tmp_table_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Lock_tables_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Execute_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Repl_slave_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Repl_client_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Create_view_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Show_view_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Create_routine_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Alter_routine_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Create_user_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Event_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('Trigger_priv', 254, 1, 1, 1, 0, 0),
|
|
# ('ssl_type', 254, 0, 9, 9, 0, 0),
|
|
# ('ssl_cipher', 252, 0, 65535, 65535, 0, 0),
|
|
# ('x509_issuer', 252, 0, 65535, 65535, 0, 0),
|
|
# ('x509_subject', 252, 0, 65535, 65535, 0, 0),
|
|
# ('max_questions', 3, 1, 11, 11, 0, 0),
|
|
# ('max_updates', 3, 1, 11, 11, 0, 0),
|
|
# ('max_connections', 3, 1, 11, 11, 0, 0),
|
|
# ('max_user_connections', 3, 1, 11, 11, 0, 0))
|
|
# conn = self.connections[0]
|
|
# c = conn.cursor()
|
|
# c.execute("select * from mysql.user")
|
|
#
|
|
# self.assertEqual(r, c.description)
|
|
|
|
def test_fetch_no_result(self):
|
|
""" test a fetchone() with no rows """
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
c.execute("create table test_nr (b varchar(32))")
|
|
try:
|
|
data = "pymysql"
|
|
c.execute("insert into test_nr (b) values (%s)", (data,))
|
|
self.assertEqual(None, c.fetchone())
|
|
finally:
|
|
c.execute("drop table test_nr")
|
|
|
|
def test_aggregates(self):
|
|
""" test aggregate functions """
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
try:
|
|
c.execute('create table test_aggregates (i integer)')
|
|
for i in xrange(0, 10):
|
|
c.execute('insert into test_aggregates (i) values (%s)', (i,))
|
|
c.execute('select sum(i) from test_aggregates')
|
|
r, = c.fetchone()
|
|
self.assertEqual(sum(range(0,10)), r)
|
|
finally:
|
|
c.execute('drop table test_aggregates')
|
|
|
|
def test_single_tuple(self):
|
|
""" test a single tuple """
|
|
conn = self.connections[0]
|
|
c = conn.cursor()
|
|
try:
|
|
c.execute("create table mystuff (id integer primary key)")
|
|
c.execute("insert into mystuff (id) values (1)")
|
|
c.execute("insert into mystuff (id) values (2)")
|
|
c.execute("select id from mystuff where id in %s", ((1,),))
|
|
self.assertEqual([(1,)], list(c.fetchall()))
|
|
finally:
|
|
c.execute("drop table mystuff")
|
|
|
|
__all__ = ["TestConversion","TestCursor"]
|
|
|
|
if __name__ == "__main__":
|
|
import unittest
|
|
unittest.main()
|