1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
import unittest
import pymysql
_mysql = pymysql
from pymysql.constants import FIELD_TYPE
from pymysql.tests import base
class TestDBAPISet(unittest.TestCase):
def test_set_equality(self):
self.assertTrue(pymysql.STRING == pymysql.STRING)
def test_set_inequality(self):
self.assertTrue(pymysql.STRING != pymysql.NUMBER)
def test_set_equality_membership(self):
self.assertTrue(FIELD_TYPE.VAR_STRING == pymysql.STRING)
def test_set_inequality_membership(self):
self.assertTrue(FIELD_TYPE.DATE != pymysql.STRING)
class CoreModule(unittest.TestCase):
"""Core _mysql module features."""
def test_NULL(self):
"""Should have a NULL constant."""
self.assertEqual(_mysql.NULL, "NULL")
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_thread_safe(self):
self.assertTrue(isinstance(_mysql.thread_safe(), int))
class CoreAPI(unittest.TestCase):
"""Test _mysql interaction internals."""
def setUp(self):
kwargs = base.PyMySQLTestCase.databases[0].copy()
kwargs["read_default_file"] = "~/.my.cnf"
self.conn = _mysql.connect(**kwargs)
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 integral value."
)
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(pymysql.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):
assert 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 an str."
)
if __name__ == "__main__":
unittest.main()
|
