From 49f933d9abe7ca6876bcbe1b2c0881be4f3c71a8 Mon Sep 17 00:00:00 2001
From: dreamhunter2333 <dreamhunter2333@gmail.com>
Date: Sun, 23 Aug 2026 22:02:31 +0800
Subject: [PATCH] test: initialize rate limit worker database

---
 e2e/scripts/docker-entrypoint.sh | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/e2e/scripts/docker-entrypoint.sh b/e2e/scripts/docker-entrypoint.sh
index a745319..48932d4 100755
--- a/e2e/scripts/docker-entrypoint.sh
+++ b/e2e/scripts/docker-entrypoint.sh
@@ -110,5 +110,12 @@ if [ -n "${WORKER_GZIP_URL:-}" ]; then
   echo "    Gzip worker database initialized"
 fi
 
+if [ -n "${WORKER_URL_RATE_LIMIT:-}" ]; then
+  echo "==> Initializing rate-limit worker database"
+  curl -sf -X POST "$WORKER_URL_RATE_LIMIT/admin/db_initialize" > /dev/null
+  curl -sf -X POST "$WORKER_URL_RATE_LIMIT/admin/db_migration" > /dev/null
+  echo "    Rate-limit worker database initialized"
+fi
+
 echo "==> Running Playwright tests"
 exec npx playwright test "$@"