Fix: Fix import issues in test/test_connection.py

isrgomez · isrgomez · commit 7cd313cfdb2c · 2018-05-18T17:48:17.000-05:00
An import error trying to import connection_cext on Python 2.7
was the cause to not able to run unittests on Python 2.7.
diff --git a/tests/test_connection.py b/tests/test_connection.py
@@ -40,7 +40,13 @@
 import tests
 from . import PY2
 
-from mysql.connector.connection_cext import HAVE_CMYSQL, CMySQLConnection
+try:
+    from mysql.connector.connection_cext import HAVE_CMYSQL, CMySQLConnection
+except ImportError:
+    # Test without C Extension
+    CMySQLConnection = None
+    HAVE_CMYSQL = False
+
 from mysql.connector.conversion import (MySQLConverterBase, MySQLConverter)
 from mysql.connector import (connect, connection, network, errors,
                              constants, cursor, abstracts, catch23)
