From 35d967360dc6833656c111ac6e18bf4543ef9de2 Mon Sep 17 00:00:00 2001
From: Xevion <xevion@xevion.dev>
Date: Sat, 9 Nov 2024 16:39:36 -0600
Subject: [PATCH] Integrate pytest into run.sh cleanly, use $@ args directly in
 COMMAND def

---
 backend/run.sh | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/backend/run.sh b/backend/run.sh
index 9577b93..a2d2dea 100755
--- a/backend/run.sh
+++ b/backend/run.sh
@@ -10,7 +10,14 @@ fi
 export ENVIRONMENT=${ENVIRONMENT:-development}
 export LOG_JSON_FORMAT=${LOG_JSON_FORMAT:-false}
 export LOG_LEVEL=${LOG_LEVEL:-debug}
-COMMAND='poetry run python3 -m linkpulse'
+COMMAND='poetry run python3 -m linkpulse $@'
+
+# If arguments start with 'poetry run pytest' or 'pytest' use args as is
+if [[ "$1" == "poetry" && "$2" == "run" && "$3" == "pytest" ]]; then
+    COMMAND=$@
+elif [[ "$1" == "pytest" ]]; then
+    COMMAND=$@
+fi
 
 # Check if Railway CLI is available
 RAILWAY_AVAILABLE=false
@@ -45,11 +52,11 @@ if $RAILWAY_AVAILABLE; then
 fi
 
 if $DATABASE_DEFINED; then
-    $COMMAND $@
+    $COMMAND
 else
     if $RAILWAY_AVAILABLE; then
         if $PROJECT_LINKED; then
-            DATABASE_URL="$(railway variables --service Postgres --environment development --json | jq .DATABASE_PUBLIC_URL -cMr)" $COMMAND $@
+            DATABASE_URL="$(railway variables --service Postgres --environment development --json | jq .DATABASE_PUBLIC_URL -cMr)" $COMMAND
         else
             echo "error: Railway project not linked."
             echo "Run 'railway link' to link the project."