fix test

nakagami · Aug 22, 2024 · 9892f60 · 9892f60
1 parent 10f116f
commit 9892f60
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/cymysql/tests/test_issues.py b/cymysql/tests/test_issues.py
@@ -230,10 +230,14 @@ def test_issue_36(self):
         # now nuke the connection
         try:
             conn.kill(kill_id)
-        except cymysql.InternalError:   # MySQL 9.0
+        except cymysql.InternalError:
             exc, value, tb = sys.exc_info()
-            self.assertEqual(value.errno, 1047)
-            self.assertEqual(value.errmsg, 'Unknown command')
+            # MySQL 8.0
+            if value.error == 1317:
+                self.assertEqual(value.errmsg, 'Query execution was interrupted'')
+            # MySQL 9.0
+            elif value.error == 1047:
+                self.assertEqual(value.errmsg, 'Unknown command')
             return
         # make sure this connection has broken
         try: