From 8c74de6f6e05a8202a990983076432a1604f92ca Mon Sep 17 00:00:00 2001
From: chrislu <chris.lu@gmail.com>
Date: Wed, 10 Sep 2025 11:13:29 -0700
Subject: [PATCH] test(kafka): add integration smoke tests under test/kafka and
 server Addr() for dialing

---
 test/kafka/gateway_smoke_test.go | 50 ++++++++++++++++++++++++++++++++
 weed/mq/kafka/gateway/server.go  |  8 +++++
 2 files changed, 58 insertions(+)
 create mode 100644 test/kafka/gateway_smoke_test.go

diff --git a/test/kafka/gateway_smoke_test.go b/test/kafka/gateway_smoke_test.go
new file mode 100644
index 000000000..b8098b7a9
--- /dev/null
+++ b/test/kafka/gateway_smoke_test.go
@@ -0,0 +1,50 @@
+package kafka
+
+import (
+    "net"
+    "testing"
+    "time"
+
+    "github.com/seaweedfs/seaweedfs/weed/mq/kafka/gateway"
+)
+
+func TestGateway_StartAcceptsConnections(t *testing.T) {
+    srv := gateway.NewServer(gateway.Options{Listen: ":0"})
+    if err := srv.Start(); err != nil {
+        t.Fatalf("start gateway: %v", err)
+    }
+    addr := srv.Addr()
+    if addr == "" {
+        t.Fatalf("server Addr() empty")
+    }
+    conn, err := net.DialTimeout("tcp", addr, 2*time.Second)
+    if err != nil {
+        t.Fatalf("dial gateway: %v", err)
+    }
+    _ = conn.Close()
+    if err := srv.Close(); err != nil {
+        t.Fatalf("close gateway: %v", err)
+    }
+}
+
+func TestGateway_RefusesAfterClose(t *testing.T) {
+    srv := gateway.NewServer(gateway.Options{Listen: ":0"})
+    if err := srv.Start(); err != nil {
+        t.Fatalf("start gateway: %v", err)
+    }
+    addr := srv.Addr()
+    if addr == "" {
+        t.Fatalf("server Addr() empty")
+    }
+    if err := srv.Close(); err != nil {
+        t.Fatalf("close gateway: %v", err)
+    }
+    // give the OS a brief moment to release the port
+    time.Sleep(50 * time.Millisecond)
+    _, err := net.DialTimeout("tcp", addr, 300*time.Millisecond)
+    if err == nil {
+        t.Fatalf("expected dial to fail after close")
+    }
+}
+
+
diff --git a/weed/mq/kafka/gateway/server.go b/weed/mq/kafka/gateway/server.go
index c0bedb4a6..857993cc4 100644
--- a/weed/mq/kafka/gateway/server.go
+++ b/weed/mq/kafka/gateway/server.go
@@ -66,4 +66,12 @@ func (s *Server) Close() error {
     return nil
 }
 
+// Addr returns the bound address of the server listener, or empty if not started.
+func (s *Server) Addr() string {
+    if s.ln == nil {
+        return ""
+    }
+    return s.ln.Addr().String()
+}
+