Add test for _mysql.escape_string()

methane · methane · commit 0dfce0651464 · 2015-02-24T14:51:34.000+09:00
diff --git a/tests/test_MySQLdb_nonstandard.py b/tests/test_MySQLdb_nonstandard.py
@@ -41,6 +41,12 @@ def test_client_info(self):
     def test_thread_safe(self):
         self.assertTrue(isinstance(_mysql.thread_safe(), int))
 
+    def test_escape_string(self):
+        self.assertEqual(_mysql.escape_string(b'foo"bar'),
+                         b'foo\\"bar', "escape byte string")
+        self.assertEqual(_mysql.escape_string(u'foo"bar'),
+                         b'foo\\"bar', "escape unicode string")
+
 
 class CoreAPI(unittest.TestCase):
     """Test _mysql interaction internals."""
