JezK
Edit File: test_MySQLdb_nonstandard.py
import sys try: import unittest2 as unittest except ImportError: import unittest import pymysql _mysql = pymysql from pymysql.constants import FIELD_TYPE from pymysql.tests import base from pymysql._compat import PY2, long_type if not PY2: basestring = str 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__, basestring)) 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(), basestring)) 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, long_type)), "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(), basestring), "Should return a string.") def test_host_info(self): assert isinstance(self.conn.get_host_info(), basestring), "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): if sys.version_info[0] == 2: self.assertTrue(isinstance(self.conn.get_server_info(), basestring), "Should return an str.") else: self.assertTrue(isinstance(self.conn.get_server_info(), basestring), "Should return an str.") if __name__ == "__main__": unittest.main()