62 lines
2.1 KiB
Go
62 lines
2.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestServiceConfigKeepsConnectivityDisabledWithoutExplicitAuthorization(t *testing.T) {
|
|
configured, err := serviceConfigFromEnvironment(testEnvironment(nil))
|
|
if err != nil {
|
|
t.Fatalf("configure disabled connectivity: %v", err)
|
|
}
|
|
if configured.config.AllowConnectivityChecks || configured.config.Probe != nil {
|
|
t.Fatal("connectivity probe must not be created without explicit authorization")
|
|
}
|
|
}
|
|
|
|
func TestServiceConfigRequiresAllExplicitConnectivityLimits(t *testing.T) {
|
|
_, err := serviceConfigFromEnvironment(testEnvironment(map[string]string{
|
|
"CHORUS_ADMIN_ALLOW_CONNECTIVITY_PROBES": "1",
|
|
}))
|
|
if err == nil || !strings.Contains(err.Error(), "CHORUS_ADMIN_CONNECTIVITY_COOLDOWN_SECONDS") {
|
|
t.Fatalf("missing cooldown error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestServiceConfigRejectsInvalidProviderAllowedPorts(t *testing.T) {
|
|
_, err := serviceConfigFromEnvironment(testEnvironment(map[string]string{
|
|
"CHORUS_ADMIN_ALLOW_CONNECTIVITY_PROBES": "1",
|
|
"CHORUS_ADMIN_CONNECTIVITY_COOLDOWN_SECONDS": "60",
|
|
"CHORUS_PROVIDER_HTTP_TIMEOUT_SECONDS": "10",
|
|
"CHORUS_PROVIDER_MAX_RESPONSE_BYTES": "1024",
|
|
"CHORUS_PROVIDER_ALLOWED_PORTS": "80,,443",
|
|
}))
|
|
if err == nil || !strings.Contains(err.Error(), "CHORUS_PROVIDER_ALLOWED_PORTS") {
|
|
t.Fatalf("invalid provider ports error = %v", err)
|
|
}
|
|
}
|
|
|
|
func TestServiceConfigAcceptsCustomProviderAllowedPort(t *testing.T) {
|
|
configured, err := serviceConfigFromEnvironment(testEnvironment(map[string]string{
|
|
"CHORUS_ADMIN_ALLOW_CONNECTIVITY_PROBES": "1",
|
|
"CHORUS_ADMIN_CONNECTIVITY_COOLDOWN_SECONDS": "60",
|
|
"CHORUS_PROVIDER_HTTP_TIMEOUT_SECONDS": "10",
|
|
"CHORUS_PROVIDER_MAX_RESPONSE_BYTES": "1024",
|
|
"CHORUS_PROVIDER_ALLOWED_PORTS": "80,443,8080",
|
|
}))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !configured.config.AllowConnectivityChecks || configured.config.Probe == nil {
|
|
t.Fatal("connectivity probe was not configured")
|
|
}
|
|
}
|
|
|
|
func testEnvironment(values map[string]string) func(string) (string, bool) {
|
|
return func(key string) (string, bool) {
|
|
value, ok := values[key]
|
|
return value, ok
|
|
}
|
|
}
|