Wait for test-server ready before running tests

clj-commons · Mar 16, 2024 · 1f99736 · 1f99736
1 parent 4a86cd0
commit 1f99736
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/script/test_server.clj b/script/test_server.clj
@@ -1,6 +1,7 @@
 (ns test-server
   (:require [babashka.cli :as cli]
             [babashka.http-server :as server]
+            [babashka.http-client :as client]
             [lread.status-line :as status]))
 
 (def cli-spec {:help {:desc "This usage help" :alias :h}
@@ -31,5 +32,19 @@
     (if (:help opts)
       (usage-help)
       (do
-        (status/line :detail "Static dir: %s" (:dir opts))
-        (server/exec opts)))))
+        (status/line :detail "Test server static dir: %s" (:dir opts))
+        (let [server (future (server/exec opts))
+              deadline (+ (System/currentTimeMillis) 15000)
+              test-url (str "http://localhost:" (:port opts)) ]
+          (loop []
+            (let [resp (try (client/get test-url)
+                            (catch Throwable _ :not-ready))]
+              (when (= :not-ready resp)
+                (if (< (System/currentTimeMillis) deadline)
+                  (do
+                    (println "- waiting for test-server to be ready" test-url)
+                    (Thread/sleep 1000)
+                    (recur))
+                  (throw (ex-info "Timed out waiting for ready test server" {}))))))
+          (println "Test server ready")
+          @server)))))