test/test_rack_hijack.rb in yahns-1.8.0 vs test/test_rack_hijack.rb in yahns-1.9.0

@@ -50,18 +50,30 @@
       logger(Logger.new(err.path))
       stderr_path err.path
     end
     pid = mkserver(cfg)
     res = Net::HTTP.start(host, port) { |h| h.get("/hijack_req") }
+
+    wait_for_msg = lambda do |n|
+      tries = 10000
+      begin
+        Thread.new { Thread.pass }.join # calls sched_yield() on MRI
+      end until File.readlines(err.path).grep(/DieIfUsed/).size >= n ||
+                (tries -= 1) < 0
+    end
     assert_equal "request.hijacked", res.body
     assert_equal 200, res.code.to_i
     assert_equal "1.0", res.http_version
 
+    wait_for_msg.call(1)
+
     res = Net::HTTP.start(host, port) { |h| h.get("/hijack_res") }
     assert_equal "response.hijacked", res.body
     assert_equal 200, res.code.to_i
     assert_equal "zzz", res["X-Test"]
     assert_equal "1.1", res.http_version
+
+    wait_for_msg.call(2)
 
     errs = File.readlines(err.path).grep(/DieIfUsed/)
     assert_equal([ "INFO #{pid} closed DieIfUsed 1\n",
                    "INFO #{pid} closed DieIfUsed 2\n" ], errs)