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
28 lines
593 B
Go
28 lines
593 B
Go
package util
|
|
|
|
import "testing"
|
|
|
|
func TestByteSizeString(t *testing.T) {
|
|
tests := []struct {
|
|
b ByteSize
|
|
want string
|
|
}{
|
|
{0, "0B"},
|
|
{1500, "1500B"},
|
|
{1024, "1KiB"},
|
|
{1000, "1KB"},
|
|
{32768, "32KiB"},
|
|
{1000000, "1MB"},
|
|
{1024000, "1000KiB"},
|
|
{3 << 30, "3GiB"},
|
|
}
|
|
for _, tt := range tests {
|
|
if got := tt.b.String(); got != tt.want {
|
|
t.Errorf("ByteSize(%d).String() = %q; want %q", int64(tt.b), got, tt.want)
|
|
}
|
|
if back, err := ParseByteSize(tt.want); err != nil || back != tt.b {
|
|
t.Errorf("ParseByteSize(%q) = %d, %v; want %d", tt.want, back, err, tt.b)
|
|
}
|
|
}
|
|
}
|