test: mysql compatibility test

Author: ColinKYuen

.github/workflows/compatibility_test.yaml

@@ -19,8 +19,8 @@
 #include "rds_utils.h"
 
 #include "../dialect/dialect.h"
-#include "../dialect/dialect_aurora_postgres.h"
 #include "../dialect/dialect_aurora_mysql.h"
+#include "../dialect/dialect_aurora_postgres.h"
 
 std::shared_ptr<Dialect> InitDialect(std::map<std::string, std::string> conn_info)
 {

driver/util/init_plugin_helper.cpp

@@ -0,0 +1,4 @@
+CREATE DATABASE IF NOT EXISTS test_database;
+CREATE USER 'test_username'@'localhost' IDENTIFIED WITH caching_sha2_password BY 'test_password';
+GRANT ALL on *.* TO 'test_username'@'localhost';
+FLUSH PRIVILEGES;

scripts/init_local_mysql_query.sql

@@ -41,6 +41,7 @@ class BaseConnectionTest : public testing::Test {
     int test_port = TEST_UTILS::StrToInt(TEST_UTILS::GetEnvVar("TEST_PORT", "5432"));
     std::string test_region = TEST_UTILS::GetEnvVar("TEST_REGION", "us-west-1");
     std::string test_server = TEST_UTILS::GetEnvVar("TEST_SERVER");
+    std::string test_dialect = TEST_UTILS::GetEnvVar("TEST_DIALECT");
 
     std::string conn_str = "";
 

test/common/base_connection_test.h

@@ -102,20 +102,17 @@ target_link_libraries(${PROJECT_NAME} PRIVATE ${EXTERNAL_LIBRARIES})
 # ODBC Unix DSN ---------------------------------------------------------------------------------------------
 SET(ODBC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/../resources/)
 
-# Common Settings
+# DSN Settings
 SET(THREADING "0")
-SET(TEST_SERVER ${TEST_SERVER},${TEST_PORT})
-SET(TEST_DATABASE ${TEST_DATABASE})
-
-# Test Driver
-SET(TEST_DSN "wrapper-dsn")
-SET(TEST_DRIVER_NAME "aws-odbc-wrapper")
-SET(TEST_DRIVER_PATH ${TEST_DRIVER_PATH})
-
-# Base PG Driver
-SET(BASE_PG_DSN "pg-dsn")
-SET(BASE_PG_DRIVER_NAME "community-pg")
-SET(BASE_PG_DRIVER_PATH ${BASE_PG_DRIVER_PATH})
+
+# To be set via flags
+# -D TEST_DRIVER_PATH
+# -D PG_DRIVER_PATH
+# -D PG_SERVER
+# -D PG_DATABASE
+# -D MYSQL_DRIVER_PATH
+# -D MYSQL_SERVER
+# -D MYSQL_DATABASE
 
 CONFIGURE_FILE(
     ${ODBC_PATH}/odbcinst.ini.in

test/compatibility/CMakeLists.txt

@@ -62,6 +62,7 @@ class FailoverIntegrationTest : public BaseFailoverIntegrationTest {
             .withPWD(test_pwd)
             .withDatabase(test_db)
             .withEnableClusterFailover(true)
+            .withDatabaseDialect(test_dialect)
             .getString();
 
         // Check to see if cluster is available.

test/integration/failover_integration_test.cpp

@@ -46,6 +46,9 @@ class IamAuthenticationIntegrationTest : public BaseConnectionTest {
 
     void SetUp() override {
         BaseConnectionTest::SetUp();
+        if (test_dialect == "AURORA_MYSQL") {
+            GTEST_SKIP() << "MySQL ODBC Connector does not support IAM with session tokens due to limitations on connection key-value pairs.";
+        }
     }
 
     void TearDown() override {

test/integration/iam_authentication_integration_test.cpp

@@ -1,13 +1,21 @@
-[ODBC Data Sources]
-data_source_name = @TEST_DSN@
+[wrapper-pg-dsn]
+Driver          = @TEST_DRIVER_PATH@
+DATABASE        = @PG_DATABASE@
+SERVER          = @PG_SERVER@
+BASE_DRIVER     = @PG_DRIVER_PATH@
 
-[@TEST_DSN@]
+[wrapper-mysql-dsn]
 Driver          = @TEST_DRIVER_PATH@
-DATABASE        = @TEST_DATABASE@
-SERVER          = @TEST_SERVER@
-BASE_DRIVER     = @BASE_PG_DRIVER_PATH@
+DATABASE        = @MYSQL_DATABASE@
+SERVER          = @MYSQL_SERVER@
+BASE_DRIVER     = @MYSQL_DRIVER_PATH@
+
+[pg-dsn]
+Driver          = @PG_DRIVER_PATH@
+DATABASE        = @PG_DATABASE@
+SERVER          = @PG_SERVER@
 
-[@BASE_PG_DSN@]
-Driver          = @BASE_PG_DRIVER_PATH@
-DATABASE        = @TEST_DATABASE@
-SERVER          = @TEST_SERVER@
+[mysql-dsn]
+Driver          = @MYSQL_DRIVER_PATH@
+DATABASE        = @MYSQL_DATABASE@
+SERVER          = @MYSQL_SERVER@

test/resources/odbc.ini.in

@@ -1,11 +1,16 @@
 [ODBC Drivers]
 @TEST_DRIVER_NAME@ = Installed
-@BASE_PG_DRIVER_NAME@ = Installed
+community-pg = Installed
+community-mysql = Installed
 
 [@TEST_DRIVER_NAME@]
 Driver = @TEST_DRIVER_PATH@
 Threading = @THREADING@
 
-[@BASE_PG_DRIVER_NAME@]
-Driver = @BASE_PG_DRIVER_PATH@
+[community-pg]
+Driver = @PG_DRIVER_PATH@
+Threading = @THREADING@
+
+[community-mysql]
+Driver = @MYSQL_DRIVER_PATH@
 Threading = @THREADING@