From 9bf26f8c9e7e3f87b17f138d827a967eff60c224 Mon Sep 17 00:00:00 2001
From: Caleb Xu <calebcenter@live.com>
Date: Wed, 7 Feb 2024 14:36:22 +0100
Subject: [PATCH] emp: use TCPServer instead of WEBrick for test

Related:
* https://github.com/Homebrew/homebrew-core/pull/161976
---
 Formula/e/emp.rb | 33 ++++++++++++++++++++-------------
 1 file changed, 20 insertions(+), 13 deletions(-)

diff --git a/Formula/e/emp.rb b/Formula/e/emp.rb
index 5f9aa98b47aa5..41714b9c7184c 100644
--- a/Formula/e/emp.rb
+++ b/Formula/e/emp.rb
@@ -31,14 +31,15 @@ def install
     (buildpath/"src/github.com/remind101/").mkpath
     ln_s buildpath, buildpath/"src/github.com/remind101/empire"
 
-    system "go", "build", "-o", bin/"emp", "./src/github.com/remind101/empire/cmd/emp"
+    system "go", "build", *std_go_args, "./src/github.com/remind101/empire/cmd/emp"
   end
 
   test do
-    require "webrick"
+    port = free_port
 
-    server = WEBrick::HTTPServer.new Port: 8035
-    server.mount_proc "/apps/foo/releases" do |_req, res|
+    # Mock an API server response to test the CLI
+    fork do
+      server = TCPServer.new(port)
       resp = {
         "created_at"  => "2015-10-12T0:00:00.00000000-00:00",
         "description" => "my awesome release",
@@ -49,17 +50,23 @@ def install
         },
         "version"     => 1,
       }
-      res.body = JSON.generate([resp])
+      body = JSON.generate([resp])
+
+      loop do
+        socket = server.accept
+        socket.write "HTTP/1.1 200 OK\r\n" \
+                     "Content-Type: application/json; charset=utf-8\r\n" \
+                     "Content-Length: #{body.bytesize}\r\n" \
+                     "\r\n"
+        socket.write body
+        socket.close
+      end
     end
 
-    Thread.new { server.start }
+    sleep 1
 
-    begin
-      ENV["EMPIRE_API_URL"] = "http://127.0.0.1:8035"
-      assert_match(/v1  zab  Oct 1(1|2|3)  2015  my awesome release/,
-        shell_output("#{bin}/emp releases -a foo").strip)
-    ensure
-      server.shutdown
-    end
+    ENV["EMPIRE_API_URL"] = "http://127.0.0.1:#{port}"
+    assert_match(/v1  zab  Oct 1(1|2|3)  2015  my awesome release/,
+      shell_output("#{bin}/emp releases -a foo").strip)
   end
 end