49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"maps"
|
|
"net"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"logwisp/internal/session"
|
|
|
|
"github.com/lixenwraith/log"
|
|
)
|
|
|
|
func newTestHTTPSink(t *testing.T, overrides map[string]any) (*HTTPSink, *session.Manager) {
|
|
t.Helper()
|
|
manager := session.NewManager(time.Hour)
|
|
t.Cleanup(manager.Stop)
|
|
opts := map[string]any{"host": "127.0.0.1", "port": int64(8081), "write_timeout_ms": int64(5000)}
|
|
maps.Copy(opts, overrides)
|
|
created, err := NewHTTPSinkPlugin("stream", opts, log.NewLogger(), session.NewProxy(manager, "stream"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
h, ok := created.(*HTTPSink)
|
|
if !ok {
|
|
t.Fatalf("sink type = %T", created)
|
|
}
|
|
return h, manager
|
|
}
|
|
|
|
func serveTestHTTPSink(t *testing.T, h *HTTPSink) (*http.Client, string) {
|
|
t.Helper()
|
|
listener, err := net.Listen("tcp4", "127.0.0.1:0")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { _ = listener.Close() })
|
|
ctx, cancel := context.WithCancel(t.Context())
|
|
t.Cleanup(func() { cancel(); h.Stop() })
|
|
if err := h.serve(ctx, listener); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
client := &http.Client{Timeout: 3 * time.Second}
|
|
t.Cleanup(client.CloseIdleConnections)
|
|
return client, "http://" + listener.Addr().String()
|
|
}
|