Graham McIntire
aef770ea83
Cover WebSocket session lifecycle (result channels, SNMP batch flushing, heartbeats, restart), MikroTik TLS TOFU verification mismatch, worker pool exhaustion, plaintext WebSocket rejection, HTTP/TCP/DNS check execution, IPv4 fallback, and customer network edge cases. Make heartbeat intervals and toWebSocketURL exit injectable for testing.
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestEnvOrDefault(t *testing.T) {
|
|
key := "TOWEROPS_TEST_ENV_OR_DEFAULT"
|
|
|
|
// Unset case
|
|
_ = os.Unsetenv(key)
|
|
if got := envOrDefault(key, "fallback"); got != "fallback" {
|
|
t.Errorf("unset: got %q, want %q", got, "fallback")
|
|
}
|
|
|
|
// Set case
|
|
t.Setenv(key, "custom")
|
|
if got := envOrDefault(key, "fallback"); got != "custom" {
|
|
t.Errorf("set: got %q, want %q", got, "custom")
|
|
}
|
|
}
|
|
|
|
func TestSanitizeURL(t *testing.T) {
|
|
tests := []struct {
|
|
input, want string
|
|
}{
|
|
{"wss://towerops.net/socket", "wss://towerops.net/socket"},
|
|
{"wss://towerops.net/socket?token=secret", "wss://towerops.net/socket?***"},
|
|
{"wss://towerops.net/socket?token=secret&key=abc", "wss://towerops.net/socket?***"},
|
|
{"://invalid url", "[invalid URL]"},
|
|
{"", ""},
|
|
}
|
|
for _, tt := range tests {
|
|
got := sanitizeURL(tt.input)
|
|
if got != tt.want {
|
|
t.Errorf("sanitizeURL(%q) = %q, want %q", tt.input, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsFlagSet(t *testing.T) {
|
|
// In tests, flags are not set via flag.Parse on the default flag set
|
|
// so isFlagSet should return false for any arbitrary name.
|
|
if isFlagSet("nonexistent-flag-xyz") {
|
|
t.Error("expected isFlagSet to return false for unset flag")
|
|
}
|
|
}
|
|
|
|
func TestToWebSocketURL(t *testing.T) {
|
|
// Enable insecure for testing plaintext conversions
|
|
origInsecure := insecureFlag
|
|
defer func() { insecureFlag = origInsecure }()
|
|
insecureFlag = true
|
|
|
|
tests := []struct {
|
|
input, want string
|
|
}{
|
|
{"http://localhost:4000", "ws://localhost:4000"},
|
|
{"https://towerops.net", "wss://towerops.net"},
|
|
{"ws://localhost:4000", "ws://localhost:4000"},
|
|
{"wss://towerops.net", "wss://towerops.net"},
|
|
{"towerops.net", "wss://towerops.net"},
|
|
{"localhost:4000", "wss://localhost:4000"},
|
|
}
|
|
for _, tt := range tests {
|
|
got := toWebSocketURL(tt.input)
|
|
if got != tt.want {
|
|
t.Errorf("toWebSocketURL(%q) = %q, want %q", tt.input, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestToWebSocketURLRejectsPlaintext(t *testing.T) {
|
|
origInsecure := insecureFlag
|
|
origExit := osExit
|
|
defer func() {
|
|
insecureFlag = origInsecure
|
|
osExit = origExit
|
|
}()
|
|
|
|
insecureFlag = false
|
|
exitCode := -1
|
|
osExit = func(code int) { exitCode = code }
|
|
|
|
tests := []struct {
|
|
name string
|
|
input string
|
|
}{
|
|
{"ws:// scheme", "ws://localhost:4000"},
|
|
{"http:// converts to ws://", "http://localhost:4000"},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
exitCode = -1
|
|
toWebSocketURL(tt.input)
|
|
if exitCode != 1 {
|
|
t.Errorf("expected osExit(1), got %d", exitCode)
|
|
}
|
|
})
|
|
}
|
|
}
|