seaweedfs/test/kafka/gateway_smoke_test.go

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")
	}
}