Update stage.yml

Author: zainforbjs

.github/workflows/stage.yml

@@ -36,38 +36,50 @@ jobs:
           docker build -t ${{ secrets.DOCKER_HUB_IP }}:${{ secrets.DOCKER_HUB_PORT}}/wheels-dev-stage .
           docker push ${{ secrets.DOCKER_HUB_IP }}:${{ secrets.DOCKER_HUB_PORT}}/wheels-dev-stage
 
-      - name: Run container and verify it starts
-        id: Run-Container-Test
+      - name: Run container and check health
+        id: Run-Container-Health-Test
         run: |
           IMAGE=${{ secrets.DOCKER_HUB_IP }}:${{ secrets.DOCKER_HUB_PORT }}/wheels-dev-stage
           CONTAINER_NAME=wheels-dev-stage-test
 
           echo "Running test container from image: $IMAGE"
           docker run -d --name $CONTAINER_NAME $IMAGE
 
-          echo "Waiting for container to reach 'running' state..."
+          echo "Waiting for container to become healthy..."
 
           TIMEOUT=60
           INTERVAL=5
           elapsed=0
 
           while [ $elapsed -lt $TIMEOUT ]; do
-            status=$(docker inspect -f '{{.State.Status}}' $CONTAINER_NAME)
-            if [ "$status" = "running" ]; then
-              echo "✅ Container is running."
+            health=$(docker inspect -f '{{.State.Health.Status}}' $CONTAINER_NAME 2>/dev/null || echo "no-health")
+            echo "Health status: $health"
+
+            if [ "$health" = "healthy" ]; then
+              echo "✅ Container is healthy."
               docker rm -f $CONTAINER_NAME
               exit 0
+            elif [ "$health" = "unhealthy" ]; then
+              echo "❌ Container is unhealthy."
+              docker logs $CONTAINER_NAME || true
+              docker rm -f $CONTAINER_NAME || true
+              exit 1
+            elif [ "$health" = "no-health" ]; then
+              echo "❌ HEALTHCHECK not defined in image. Cannot proceed."
+              docker rm -f $CONTAINER_NAME || true
+              exit 1
             fi
+
             sleep $INTERVAL
             elapsed=$((elapsed + INTERVAL))
           done
 
-          echo "❌ Container failed to reach 'running' state within $TIMEOUT seconds."
-          echo "::error::Image did not produce a healthy container."
+          echo "❌ Container health status stuck at 'starting' for more than $TIMEOUT seconds."
           docker logs $CONTAINER_NAME || true
           docker rm -f $CONTAINER_NAME || true
           exit 1
 
+
       - name: Deploy to Docker Swarm
         id: Deploy-to-Docker-Swarm
         run: |