76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package proxy
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestHealthCheck(t *testing.T) {
|
|
upstream := &staticUpstream{
|
|
from: "",
|
|
Hosts: testPool(),
|
|
Policy: &Random{},
|
|
FailTimeout: 10 * time.Second,
|
|
MaxFails: 1,
|
|
}
|
|
upstream.healthCheck()
|
|
if upstream.Hosts[0].Down() {
|
|
t.Error("Expected first host in testpool to not fail healthcheck.")
|
|
}
|
|
if !upstream.Hosts[1].Down() {
|
|
t.Error("Expected second host in testpool to fail healthcheck.")
|
|
}
|
|
}
|
|
|
|
func TestSelect(t *testing.T) {
|
|
upstream := &staticUpstream{
|
|
from: "",
|
|
Hosts: testPool()[:3],
|
|
Policy: &Random{},
|
|
FailTimeout: 10 * time.Second,
|
|
MaxFails: 1,
|
|
}
|
|
upstream.Hosts[0].Unhealthy = true
|
|
upstream.Hosts[1].Unhealthy = true
|
|
upstream.Hosts[2].Unhealthy = true
|
|
if h := upstream.Select(); h != nil {
|
|
t.Error("Expected select to return nil as all host are down")
|
|
}
|
|
upstream.Hosts[2].Unhealthy = false
|
|
if h := upstream.Select(); h == nil {
|
|
t.Error("Expected select to not return nil")
|
|
}
|
|
}
|
|
|
|
func TestRegisterPolicy(t *testing.T) {
|
|
name := "custom"
|
|
customPolicy := &customPolicy{}
|
|
RegisterPolicy(name, func() Policy { return customPolicy })
|
|
if _, ok := supportedPolicies[name]; !ok {
|
|
t.Error("Expected supportedPolicies to have a custom policy.")
|
|
}
|
|
|
|
}
|
|
|
|
func TestAllowedPaths(t *testing.T) {
|
|
upstream := &staticUpstream{
|
|
from: "miek.nl.",
|
|
IgnoredSubDomains: []string{"download.", "static."}, // closing dot mandatory
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
expected bool
|
|
}{
|
|
{"miek.nl.", true},
|
|
{"download.miek.nl.", false},
|
|
{"static.miek.nl.", false},
|
|
{"blaat.miek.nl.", true},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
isAllowed := upstream.IsAllowedPath(test.name)
|
|
if test.expected != isAllowed {
|
|
t.Errorf("Test %d: expected %v found %v", i+1, test.expected, isAllowed)
|
|
}
|
|
}
|
|
}
|