✅ handle CMySQLConnection import for compatibility with different MySQL connector builds

Author: techouse

tests/conftest.py

@@ -20,7 +20,6 @@
 from docker.models.containers import Container
 from faker import Faker
 from mysql.connector import MySQLConnection, errorcode
-from mysql.connector.connection_cext import CMySQLConnection
 from mysql.connector.pooling import PooledMySQLConnection
 from requests import HTTPError
 from sqlalchemy import create_engine
@@ -31,6 +30,12 @@
 
 from . import database, factories
 
+try:
+    from mysql.connector.connection_cext import CMySQLConnection
+except ImportError:
+    # Free-threaded Python builds fall back to the pure-Python connector.
+    CMySQLConnection = MySQLConnection
+
 
 def pytest_addoption(parser: "Parser") -> None:
     parser.addoption(

tests/func/sqlite3_to_mysql_test.py

@@ -14,7 +14,6 @@
 from _pytest.logging import LogCaptureFixture
 from faker import Faker
 from mysql.connector import MySQLConnection, errorcode
-from mysql.connector.connection_cext import CMySQLConnection
 from mysql.connector.pooling import PooledMySQLConnection
 from pytest_mock import MockFixture
 from sqlalchemy import MetaData, Table, create_engine, inspect, select, text
@@ -26,6 +25,12 @@
 from sqlite3_to_mysql import SQLite3toMySQL
 from tests.conftest import Helpers, MySQLCredentials
 
+try:
+    from mysql.connector.connection_cext import CMySQLConnection
+except ImportError:
+    # Free-threaded Python builds fall back to the pure-Python connector.
+    CMySQLConnection = MySQLConnection
+
 
 @pytest.mark.usefixtures("sqlite_database", "mysql_instance")
 class TestSQLite3toMySQL: