From 1cc513e3283dc64421ceba27b4f9b92daccf32d5 Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Mon, 15 Dec 2025 10:07:04 +0800
Subject: [PATCH] Add Environment Setup Step in CI Workflows

- Introduced a new step in both `pr-test.yml` and `unit-test.yml` workflows to set up environment variables for database configuration.
- Enhanced the environment setup to conditionally configure database connection strings based on the selected database driver, improving test reliability and flexibility.
- Created necessary directories for SQLite database storage when not using MySQL, ensuring proper environment preparation for tests.
---
 .github/workflows/pr-test.yml   | 12 ++++++++++++
 .github/workflows/unit-test.yml | 12 ++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/.github/workflows/pr-test.yml b/.github/workflows/pr-test.yml
index 402df344..0b2ea7b7 100644
--- a/.github/workflows/pr-test.yml
+++ b/.github/workflows/pr-test.yml
@@ -461,6 +461,18 @@ jobs:
           user: "xiang"
           password: ${{ secrets.UNIT_PASS }}
 
+      - name: Setup ENV
+        env:
+          PASSWORD: ${{ secrets.UNIT_PASS }}
+        run: |
+          echo "YAO_DB_DRIVER=$DB_DRIVER" >> $GITHUB_ENV
+          if [ "$DB_DRIVER" = "mysql" ]; then
+            echo "YAO_DB_PRIMARY=$DB_USER:$PASSWORD@$DB_HOST" >> $GITHUB_ENV
+          else
+            echo "YAO_DB_PRIMARY=${{ github.WORKSPACE }}/../app/db/yao.db" >> $GITHUB_ENV
+            mkdir -p ${{ github.WORKSPACE }}/../app/db
+          fi
+
       - name: Run Benchmark & Memory Leak Tests
         run: |
           make benchmark
diff --git a/.github/workflows/unit-test.yml b/.github/workflows/unit-test.yml
index e6c6af64..be645cd3 100644
--- a/.github/workflows/unit-test.yml
+++ b/.github/workflows/unit-test.yml
@@ -372,6 +372,18 @@ jobs:
           user: "xiang"
           password: ${{ secrets.UNIT_PASS }}
 
+      - name: Setup ENV
+        env:
+          PASSWORD: ${{ secrets.UNIT_PASS }}
+        run: |
+          echo "YAO_DB_DRIVER=$DB_DRIVER" >> $GITHUB_ENV
+          if [ "$DB_DRIVER" = "mysql" ]; then
+            echo "YAO_DB_PRIMARY=$DB_USER:$PASSWORD@$DB_HOST" >> $GITHUB_ENV
+          else
+            echo "YAO_DB_PRIMARY=${{ github.WORKSPACE }}/../app/db/yao.db" >> $GITHUB_ENV
+            mkdir -p ${{ github.WORKSPACE }}/../app/db
+          fi
+
       - name: Run Benchmark & Memory Leak Tests
         run: |
           make benchmark