63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
package routing
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
func TestRuleSetUsesFirstMatchingRule(t *testing.T) {
|
|
rules, err := Compile([]Rule{
|
|
{Name: "specific", Match: Match{HostRegex: `(^|\.)jd\.com$`}, Upstreams: []string{"jd"}},
|
|
{Name: "default", Match: Match{HostRegex: `.*`}, Action: ActionReject},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("Compile(): %v", err)
|
|
}
|
|
|
|
got, ok := rules.Match(Request{Host: "api.jd.com", Method: "GET", Path: "/"})
|
|
if !ok || got.Name != "specific" {
|
|
t.Fatalf("Match() = %q, %v; want specific, true", got.Name, ok)
|
|
}
|
|
}
|
|
|
|
func TestSequentialSwitchesOnceAtThreshold(t *testing.T) {
|
|
sequence, err := NewSequential([]string{"a", "b", "c"}, 5)
|
|
if err != nil {
|
|
t.Fatalf("NewSequential(): %v", err)
|
|
}
|
|
for range 4 {
|
|
sequence.ObserveEmpty("a")
|
|
}
|
|
if got := sequence.Current(); got != "a" {
|
|
t.Fatalf("Current() = %q before threshold, want a", got)
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
for range 100 {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
sequence.ObserveEmpty("a")
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
|
|
if got := sequence.Current(); got != "b" {
|
|
t.Fatalf("Current() = %q after concurrent threshold, want b", got)
|
|
}
|
|
}
|
|
|
|
func TestSequentialValidFetchResetsEmptyCount(t *testing.T) {
|
|
sequence, _ := NewSequential([]string{"a", "b"}, 5)
|
|
for range 4 {
|
|
sequence.ObserveEmpty("a")
|
|
}
|
|
sequence.ObserveValid("a")
|
|
for range 4 {
|
|
sequence.ObserveEmpty("a")
|
|
}
|
|
if got := sequence.Current(); got != "a" {
|
|
t.Fatalf("Current() = %q, want a after reset", got)
|
|
}
|
|
}
|