You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.1 KiB
48 lines
1.1 KiB
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")
|
|
}
|
|
}
|