JezK
Edit File: test_load_local.py
from pymysql import cursors, OperationalError, Warning from pymysql.tests import base import os import warnings import unittest2 __all__ = ["TestLoadLocal"] @unittest2.skip("Not working on Debian") class TestLoadLocal(base.PyMySQLTestCase): def test_no_file(self): """Test load local infile when the file does not exist""" conn = self.connections[0] c = conn.cursor() c.execute("CREATE TABLE test_load_local (a INTEGER, b INTEGER)") try: self.assertRaises( OperationalError, c.execute, ("LOAD DATA LOCAL INFILE 'no_data.txt' INTO TABLE " "test_load_local fields terminated by ','") ) finally: c.execute("DROP TABLE test_load_local") c.close() def test_load_file(self): """Test load local infile with a valid file""" conn = self.connections[0] c = conn.cursor() c.execute("CREATE TABLE test_load_local (a INTEGER, b INTEGER)") filename = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'load_local_data.txt') try: c.execute( ("LOAD DATA LOCAL INFILE '{0}' INTO TABLE " + "test_load_local FIELDS TERMINATED BY ','").format(filename) ) c.execute("SELECT COUNT(*) FROM test_load_local") self.assertEqual(22749, c.fetchone()[0]) finally: c.execute("DROP TABLE test_load_local") def test_unbuffered_load_file(self): """Test unbuffered load local infile with a valid file""" conn = self.connections[0] c = conn.cursor(cursors.SSCursor) c.execute("CREATE TABLE test_load_local (a INTEGER, b INTEGER)") filename = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'load_local_data.txt') try: c.execute( ("LOAD DATA LOCAL INFILE '{0}' INTO TABLE " + "test_load_local FIELDS TERMINATED BY ','").format(filename) ) c.execute("SELECT COUNT(*) FROM test_load_local") self.assertEqual(22749, c.fetchone()[0]) finally: c.close() conn.close() conn.connect() c = conn.cursor() c.execute("DROP TABLE test_load_local") def test_load_warnings(self): """Test load local infile produces the appropriate warnings""" conn = self.connections[0] c = conn.cursor() c.execute("CREATE TABLE test_load_local (a INTEGER, b INTEGER)") filename = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data', 'load_local_warn_data.txt') try: with warnings.catch_warnings(record=True) as w: warnings.simplefilter('always') c.execute( ("LOAD DATA LOCAL INFILE '{0}' INTO TABLE " + "test_load_local FIELDS TERMINATED BY ','").format(filename) ) self.assertEqual(w[0].category, Warning) expected_message = "Incorrect integer value" if expected_message not in str(w[-1].message): self.fail("%r not in %r" % (expected_message, w[-1].message)) finally: c.execute("DROP TABLE test_load_local") c.close() if __name__ == "__main__": import unittest unittest.main()