70 lines
1.8 KiB
Go
70 lines
1.8 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"maps"
|
|
"net"
|
|
"net/http"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"logwisp/internal/authz"
|
|
"logwisp/internal/session"
|
|
"logwisp/internal/testutil"
|
|
|
|
"github.com/lixenwraith/auth"
|
|
"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()
|
|
}
|
|
|
|
// scramCredentials writes a credentials file holding viewer-01 under a cheap
|
|
// Argon2 profile
|
|
func scramCredentials(t *testing.T) string {
|
|
t.Helper()
|
|
cred, err := auth.NewCredential("viewer-01", "viewer-01-secret", auth.WithTime(1), auth.WithMemory(64), auth.WithThreads(1))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data, err := (&authz.Credentials{DecoyKey: make([]byte, 32), Users: []*auth.Credential{cred}}).Marshal()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
path := filepath.Join(t.TempDir(), "users.toml")
|
|
testutil.WriteFile(t, path, string(data))
|
|
return path
|
|
}
|