fix test

nakagami · Aug 22, 2024 · bcf5930 · bcf5930
1 parent f4c403a
commit bcf5930
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/cymysql/tests/test_issues.py b/cymysql/tests/test_issues.py
@@ -232,13 +232,12 @@ def test_issue_36(self):
             conn.kill(kill_id)
         except cymysql.InternalError:
             exc, value, tb = sys.exc_info()
-            # MySQL 8.0
+            if value.errno == 1047:
+                # MySQL 9.0
+                self.assertEqual(value.errmsg, 'Unknown command')
+                return
             if value.errno == 1317:
                 self.assertEqual(value.errmsg, 'Query execution was interrupted')
-            # MySQL 9.0
-            elif value.errno == 1047:
-                self.assertEqual(value.errmsg, 'Unknown command')
-            return
         # make sure this connection has broken
         try:
             c.execute("show tables")