Add test to reproduce PyMySQL#399

methane · methane · commit abcb5f9b99a4 · 2016-01-04T21:27:29.000+09:00
diff --git a/pymysql/tests/test_connection.py b/pymysql/tests/test_connection.py
@@ -418,7 +418,7 @@ def test_connection_gone_away(self):
         with self.assertRaises(pymysql.OperationalError) as cm:
             cur.execute("SELECT 1+1")
         # error occures while reading, not writing because of socket buffer.
-        #self.assertEquals(cm.exception.args[0], 2006)
+        #self.assertEqual(cm.exception.args[0], 2006)
         self.assertIn(cm.exception.args[0], (2006, 2013))
 
     def test_init_command(self):
@@ -566,3 +566,11 @@ def test_previous_cursor_not_closed(self):
         cur2 = con.cursor()
         with self.assertRaises(pymysql.ProgrammingError):
             cur2.execute("SELECT 3")
+
+    def test_commit_during_multi_result(self):
+        con = self.connections[0]
+        cur = con.cursor()
+        cur.execute("SELECT 1; SELECT 2")
+        con.commit()
+        cur.execute("SELECT 3")
+        self.assertEqual(cur.fetchone()[0], 3)
