placeholder mysql

Author: sfc-gh-yuwang

tests/integ/datasource/test_jdbc.py

@@ -26,6 +26,11 @@
     POSTGRES_SECRET,
     POSTGRES_URL,
     postgres_expected_data,
+    MYSQL_SECRET,
+    MYSQL_URL,
+)
+from tests.resources.test_data_source_dir.test_mysql_data import (
+    MYSQL_TEST_EXTERNAL_ACCESS_INTEGRATION,
 )
 from tests.resources.test_data_source_dir.test_postgres_data import (
     POSTGRES_TEST_EXTERNAL_ACCESS_INTEGRATION,
@@ -35,6 +40,7 @@
 SELECT_QUERY = "SELECT ID, NUMBER_COL, BINARY_FLOAT_COL, BINARY_DOUBLE_COL, VARCHAR2_COL, CHAR_COL, CLOB_COL, NCHAR_COL, NVARCHAR2_COL, NCLOB_COL, DATE_COL, TIMESTAMP_COL, TIMESTAMP_TZ_COL, TIMESTAMP_LTZ_COL, RAW_COL, GUID_COL FROM ALL_TYPE_TABLE_JDBC"
 EMPTY_QUERY = "SELECT * FROM ALL_TYPE_TABLE_JDBC WHERE 1=0"
 POSTGRES_SELECT_QUERY = "select BIGINT_COL, BIGSERIAL_COL, BIT_COL, BIT_VARYING_COL, BOOLEAN_COL, BOX_COL, BYTEA_COL, CHAR_COL, VARCHAR_COL, CIDR_COL, CIRCLE_COL, DATE_COL, DOUBLE_PRECISION_COL, INET_COL, INTEGER_COL, INTERVAL_COL, JSON_COL, JSONB_COL, LINE_COL, LSEG_COL, MACADDR_COL, MACADDR8_COL, NUMERIC_COL, PATH_COL, PG_LSN_COL, PG_SNAPSHOT_COL, POINT_COL, POLYGON_COL, REAL_COL, SMALLINT_COL, SMALLSERIAL_COL, SERIAL_COL, TEXT_COL, TIME_COL, TIMESTAMP_COL, TIMESTAMPTZ_COL, TSQUERY_COL, TSVECTOR_COL, TXID_SNAPSHOT_COL, UUID_COL, XML_COL from test_schema.ALL_TYPE_TABLE"
+MYSQL_SELECT_QUERY = "select * from ALL_TYPES_TABLE"
 TABLE_NAME = "ALL_TYPE_TABLE_JDBC"
 
 
@@ -62,6 +68,12 @@ def postgres_jar_path(session):
     return stage_name + "/postgresql-42.7.7.jar"
 
 
+@pytest.fixture(scope="module")
+def mysql_jar_path(session):
+    stage_name = session.get_session_stage()
+    return stage_name + "/mysql-connector-j-9.5.0.jar"
+
+
 @pytest.fixture(scope="module")
 def postgres_udtf_configs(session, postgres_jar_path):
     return {
@@ -71,6 +83,15 @@ def postgres_udtf_configs(session, postgres_jar_path):
     }
 
 
+@pytest.fixture(scope="module")
+def mysql_udtf_configs(session, mysql_jar_path):
+    return {
+        "external_access_integration": MYSQL_TEST_EXTERNAL_ACCESS_INTEGRATION,
+        "secret": MYSQL_SECRET,
+        "imports": [mysql_jar_path],
+    }
+
+
 @pytest.fixture(scope="module")
 def udtf_configs(session, jar_path):
     return {
@@ -89,6 +110,9 @@ def setup(session, resources_path):
     session.file.put(
         resources_path + "/test_data_source_dir/postgresql-42.7.7.jar", stage_name
     )
+    session.file.put(
+        resources_path + "/test_data_source_dir/mysql-connector-j-9.5.0.jar", stage_name
+    )
     yield
 
 
@@ -340,3 +364,20 @@ def test_postgres_session_init_statement(
             query_timeout=1,
             session_init_statement=session_init_statement,
         ).collect()
+
+
+def test_connect_mysql(session, mysql_udtf_configs):
+    df = session.read.jdbc(
+        url=MYSQL_URL,
+        udtf_configs=mysql_udtf_configs,
+        query=MYSQL_SELECT_QUERY,
+    ).order_by("ID")
+    print(df.collect())
+
+
+def test_sql_server(session):
+    pass
+
+
+def test_databricks(session):
+    pass

tests/integ/datasource/test_mysql.py

@@ -26,6 +26,7 @@
     mysql_less_column_schema,
     mysql_unicode_schema,
     mysql_double_quoted_schema,
+    MYSQL_TEST_EXTERNAL_ACCESS_INTEGRATION,
 )
 from tests.utils import RUNNING_ON_JENKINS, Utils
 from tests.parameters import MYSQL_CONNECTION_PARAMETERS
@@ -54,7 +55,6 @@
 
 TEST_TABLE_NAME = "ALL_TYPES_TABLE"
 TEST_QUERY = "select * from ALL_TYPES_TABLE"
-MYSQL_TEST_EXTERNAL_ACCESS_INTEGRATION = "snowpark_dbapi_mysql_test_integration"
 
 
 def create_connection_mysql():

tests/resources/test_data_source_dir/mysql-connector-j-9.5.0.jar

@@ -18,6 +18,7 @@
 from tests.parameters import (
     ORACLEDB_CONNECTION_PARAMETERS,
     POSTGRES_CONNECTION_PARAMETERS,
+    MYSQL_CONNECTION_PARAMETERS,
 )
 
 host = ORACLEDB_CONNECTION_PARAMETERS["host"]
@@ -30,6 +31,12 @@
 POSTGRES_URL = f"jdbc:postgresql://{postgres_host}:{postgres_port}/{postgres_dbname}"
 POSTGRES_SECRET = "ADMIN.PUBLIC.SNOWPARK_DBAPI_POSTGRES_TEST_CRED"
 
+mysql_host = MYSQL_CONNECTION_PARAMETERS["host"]
+mysql_port = MYSQL_CONNECTION_PARAMETERS["port"]
+mysql_dbname = MYSQL_CONNECTION_PARAMETERS["database"]
+MYSQL_URL = f"jdbc:mysql://{mysql_host}:{mysql_port}/{mysql_dbname}"
+MYSQL_SECRET = "ADMIN.PUBLIC.SNOWPARK_DBAPI_MYSQL_TEST_CRED"
+
 URL = f"jdbc:oracle:thin:@//{host}:{port}/{service_name}"
 EXTERNAL_ACCESS_INTEGRATION = "SNOWPARK_DBAPI_ORACLEDB_TEST_INTEGRATION"
 SECRET = "ADMIN.PUBLIC.SNOWPARK_DBAPI_ORACLEDB_TEST_CRED"