Add MariaDB migration testing

.github/workflows/release.yml

@@ -66,6 +66,21 @@ jobs:
     - name: Test migrations across versions
       run: bash scripts/test-migrations.sh sqlite
 
+  mariadb-db-migration-testing:
+    runs-on: ubuntu-dind-runners
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v4.1.1
+
+    - name: Set up Python 3.9
+      uses: actions/setup-python@v4.8.0
+      with:
+        python-version: "3.9"
+
+    - name: Test migrations across versions
+      run: bash scripts/test-migrations.sh mariadb
+
   publish-python-package:
     needs: [setup-and-test, mlstacks-compatibility-check, sqlite-db-migration-testing, mysql-db-migration-testing]
     uses: ./.github/workflows/publish_to_pypi.yml

.github/workflows/testing_this_now.yml

@@ -0,0 +1,18 @@
+---
+name: TEST MariaDB DB migrations
+on:
+  pull_request:
+    types: [opened, synchronize, ready_for_review]
+    paths-ignore: [docs/**, docker/**, '*', '!pyproject.toml', '**.md']
+jobs:
+  mariadb-db-migration-testing:
+    runs-on: ubuntu-dind-runners
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4.1.1
+      - name: Set up Python 3.9
+        uses: actions/setup-python@v4.8.0
+        with:
+          python-version: '3.9'
+      - name: Test migrations across versions
+        run: bash scripts/test-migrations.sh mariadb

scripts/test-migrations.sh

@@ -41,9 +41,14 @@ if [ "$1" == "mysql" ]; then
     echo "===== Testing MySQL ====="
     # run a mysql instance in docker
     docker run --name mysql -d -p 3306:3306 -e MYSQL_ROOT_PASSWORD=password mysql:latest
-
     # mysql takes a while to start up
     sleep 30
+elif [ "$1" == "mariadb" ]; then
+    echo "===== Testing MariaDB ====="
+    # run a mariadb instance in docker
+    docker run --name mariadb -d -p 3306:3306 -e MYSQL_ROOT_PASSWORD=password mariadb:10.6
+    # mariadb takes a while to start up
+    sleep 30
 fi
 
 # List of versions to test
@@ -75,6 +80,8 @@ do
 
     if [ "$1" == "mysql" ]; then
         zenml connect --url mysql://127.0.0.1/zenml --username root --password password
+    elif [ "$1" == "mariadb" ]; then
+        zenml connect --url mysql://127.0.0.1/zenml --username root --password password
     fi
 
     # Run the tests for this version
@@ -100,13 +107,18 @@ pip3 install importlib_metadata
 
 if [ "$1" == "mysql" ]; then
     zenml connect --url mysql://127.0.0.1/zenml --username root --password password
+elif [ "$1" == "mariadb" ]; then
+    zenml connect --url mysql://127.0.0.1/zenml --username root --password password
 fi
 
 run_tests_for_version current_branch_mysql
 
 if [ "$1" == "mysql" ]; then
     zenml disconnect
     docker rm -f mysql
+elif [ "$1" == "mariadb" ]; then
+    zenml disconnect
+    docker rm -f mariadb
 fi
 
 deactivate