bwilliams
c9f42b62ad
which runs a plain TCP server for receiving websocket hub clients, with server config, CSRF and rate limit middleware. - internal/websocket: Hub pushing messages to WebSocket clients (ServeHTTP, example.com/ws upgrade with optional CSRF check) and plain TCP clients (ServeTCP, newline-delimited JSON envelopes); ping/pong keepalive, per-client message rate limit, slow consumer handling, graceful Shutdown, debug logging of messages sent and received - internal/websocket/tcpserver: standalone TCP server for a hub, with accept back-off, a per-IP connection limit (server.websocket.connection_limit) and graceful shutdown - internal/cli/wsserve: `wsserve` command running the TCP server, with --listen/--host/--port overrides - internal/middleware: keyed token bucket rate limiter, real client IP from trusted proxies - internal/httpx: RFC 9457 problem responses, request IDs in context - internal/config: server section (host, port, shutdown timeout, rate limiter idle TTL, websocket); util.ByteSize settings (e.g. "32KiB"); lists of sections as single keys; empty lists stored as nil; isExplicitlySet for printLoadedConfig; JobSearchConfigSource renamed to JSCSourcesSource; TestSections checks every section's methods - internal/util: ByteSize, parsed and formatted with go-humanize
53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"slices"
|
|
"testing"
|
|
|
|
"git.avsdev.uk/Go/job-search/internal/cli/cmdtest"
|
|
)
|
|
|
|
// The commands are tested in their own packages under internal/cli. These
|
|
// tests check newRootCmd puts them together.
|
|
|
|
func TestCommandTree(t *testing.T) {
|
|
root := newRootCmd()
|
|
var names []string
|
|
for _, c := range root.Commands() {
|
|
names = append(names, c.Name())
|
|
}
|
|
for _, want := range []string{"config", "fetch", "migrate", "version", "wsserve"} {
|
|
if !slices.Contains(names, want) {
|
|
t.Errorf("job-search has no %q command; has %v", want, names)
|
|
}
|
|
}
|
|
for _, flag := range []string{"config", "debug", "quiet"} {
|
|
if root.PersistentFlags().Lookup(flag) == nil {
|
|
t.Errorf("job-search has no --%s flag", flag)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestVersion(t *testing.T) {
|
|
cmdtest.Setup(t, "")
|
|
cmdtest.Contains(t, cmdtest.MustRun(t, newRootCmd, "version"), "JobSearch - ")
|
|
}
|
|
|
|
// A command that loads the configuration runs through the real root: its
|
|
// flags and hooks, and the subcommand beneath it.
|
|
func TestRootRunsCommands(t *testing.T) {
|
|
cmdtest.Setup(t, "logging:\n level: info\n")
|
|
cmdtest.Contains(t, cmdtest.MustRun(t, newRootCmd, "config", "get", "logging.level"),
|
|
"info <string>")
|
|
cmdtest.Contains(t,
|
|
cmdtest.MustRun(t, newRootCmd, "config", "get", "logging.level", "--debug"),
|
|
"debug <string>")
|
|
}
|
|
|
|
func TestUnknownCommand(t *testing.T) {
|
|
cmdtest.Setup(t, "")
|
|
if _, err := cmdtest.Run(t, newRootCmd, "", "nope"); err == nil {
|
|
t.Error("an unknown command was accepted")
|
|
}
|
|
}
|