.

lolgab · Jan 13, 2025 · d10922e · d10922e
1 parent b44373e
commit d10922e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/snunit-mill-plugin/test/src/snunit/plugin/SNUnitMillPluginTests.scala b/snunit-mill-plugin/test/src/snunit/plugin/SNUnitMillPluginTests.scala
@@ -4,6 +4,7 @@ import mill._
 import mill.scalalib._
 import mill.testkit.{TestBaseModule, UnitTester}
 import utest._
+import java.util.concurrent.atomic.AtomicBoolean
 
 object SNUnitMillPluginTests extends TestSuite {
   def tests: Tests = Tests {
@@ -17,9 +18,13 @@ object SNUnitMillPluginTests extends TestSuite {
       val resourceFolder = os.Path(sys.env("MILL_TEST_RESOURCE_DIR").split(";").head)
 
       UnitTester(build, resourceFolder / "simple").scoped { eval =>
-        scala.concurrent.ExecutionContext.global.execute(() => eval(build.snunitRunNGINXUnit()))
+        val ended = new AtomicBoolean(false)
+        scala.concurrent.ExecutionContext.global.execute { () =>
+          eval(build.snunitRunNGINXUnit())
+          ended.set(true)
+        }
         var started = false
-        while (!started) {
+        while (!started && !ended.get()) {
           try {
             val response = requests.get("http://127.0.0.1:8080", check = false).text()
             started = true