mirror of
https://github.com/PyMySQL/mysqlclient.git
synced 2025-08-14 18:12:35 +08:00
147 lines
4.2 KiB
Python
147 lines
4.2 KiB
Python
import unittest
|
|
|
|
from MySQLdb import _mysql
|
|
import MySQLdb
|
|
from MySQLdb.constants import FIELD_TYPE
|
|
from configdb import connection_factory
|
|
import warnings
|
|
|
|
warnings.simplefilter("ignore")
|
|
|
|
|
|
class TestDBAPISet(unittest.TestCase):
|
|
def test_set_equality(self):
|
|
self.assertTrue(MySQLdb.STRING == MySQLdb.STRING)
|
|
|
|
def test_set_inequality(self):
|
|
self.assertTrue(MySQLdb.STRING != MySQLdb.NUMBER)
|
|
|
|
def test_set_equality_membership(self):
|
|
self.assertTrue(FIELD_TYPE.VAR_STRING == MySQLdb.STRING)
|
|
|
|
def test_set_inequality_membership(self):
|
|
self.assertTrue(FIELD_TYPE.DATE != MySQLdb.STRING)
|
|
|
|
|
|
class TestCoreModule(unittest.TestCase):
|
|
"""Core _mysql module features."""
|
|
|
|
def test_version(self):
|
|
"""Version information sanity."""
|
|
self.assertTrue(isinstance(_mysql.__version__, str))
|
|
|
|
self.assertTrue(isinstance(_mysql.version_info, tuple))
|
|
self.assertEqual(len(_mysql.version_info), 5)
|
|
|
|
def test_client_info(self):
|
|
self.assertTrue(isinstance(_mysql.get_client_info(), str))
|
|
|
|
def test_escape_string(self):
|
|
self.assertEqual(
|
|
_mysql.escape_string(b'foo"bar'), b'foo\\"bar', "escape byte string"
|
|
)
|
|
self.assertEqual(
|
|
_mysql.escape_string('foo"bar'), b'foo\\"bar', "escape unicode string"
|
|
)
|
|
|
|
|
|
class CoreAPI(unittest.TestCase):
|
|
"""Test _mysql interaction internals."""
|
|
|
|
def setUp(self):
|
|
self.conn = connection_factory(use_unicode=True)
|
|
|
|
def tearDown(self):
|
|
self.conn.close()
|
|
|
|
def test_thread_id(self):
|
|
tid = self.conn.thread_id()
|
|
self.assertTrue(isinstance(tid, int), "thread_id didn't return an int.")
|
|
|
|
self.assertRaises(
|
|
TypeError,
|
|
self.conn.thread_id,
|
|
("evil",),
|
|
"thread_id shouldn't accept arguments.",
|
|
)
|
|
|
|
def test_affected_rows(self):
|
|
self.assertEqual(
|
|
self.conn.affected_rows(), 0, "Should return 0 before we do anything."
|
|
)
|
|
|
|
# def test_debug(self):
|
|
# (FIXME) Only actually tests if you lack SUPER
|
|
# self.assertRaises(MySQLdb.OperationalError,
|
|
# self.conn.dump_debug_info)
|
|
|
|
def test_charset_name(self):
|
|
self.assertTrue(
|
|
isinstance(self.conn.character_set_name(), str), "Should return a string."
|
|
)
|
|
|
|
def test_host_info(self):
|
|
self.assertTrue(
|
|
isinstance(self.conn.get_host_info(), str), "Should return a string."
|
|
)
|
|
|
|
def test_proto_info(self):
|
|
self.assertTrue(
|
|
isinstance(self.conn.get_proto_info(), int), "Should return an int."
|
|
)
|
|
|
|
def test_server_info(self):
|
|
self.assertTrue(
|
|
isinstance(self.conn.get_server_info(), str), "Should return a string."
|
|
)
|
|
|
|
def test_client_flag(self):
|
|
conn = connection_factory(
|
|
use_unicode=True, client_flag=MySQLdb.constants.CLIENT.FOUND_ROWS
|
|
)
|
|
|
|
self.assertIsInstance(conn.client_flag, int)
|
|
self.assertTrue(conn.client_flag & MySQLdb.constants.CLIENT.FOUND_ROWS)
|
|
with self.assertRaises(AttributeError):
|
|
conn.client_flag = 0
|
|
|
|
conn.close()
|
|
|
|
def test_fileno(self):
|
|
self.assertGreaterEqual(self.conn.fileno(), 0)
|
|
|
|
def test_context_manager(self):
|
|
with connection_factory() as conn:
|
|
self.assertFalse(conn.closed)
|
|
self.assertTrue(conn.closed)
|
|
|
|
|
|
class TestCollation(unittest.TestCase):
|
|
"""Test charset and collation connection options."""
|
|
|
|
def setUp(self):
|
|
# Initialize a connection with a non-default character set and
|
|
# collation.
|
|
self.conn = connection_factory(
|
|
charset="utf8mb4",
|
|
collation="utf8mb4_esperanto_ci",
|
|
)
|
|
|
|
def tearDown(self):
|
|
self.conn.close()
|
|
|
|
def test_charset_collation(self):
|
|
c = self.conn.cursor()
|
|
c.execute(
|
|
"""
|
|
SHOW VARIABLES WHERE
|
|
Variable_Name="character_set_connection" OR
|
|
Variable_Name="collation_connection";
|
|
"""
|
|
)
|
|
row = c.fetchall()
|
|
charset = row[0][1]
|
|
collation = row[1][1]
|
|
self.assertEqual(charset, "utf8mb4")
|
|
self.assertEqual(collation, "utf8mb4_esperanto_ci")
|