Support routes.
This commit is contained in:
@@ -463,6 +463,94 @@ func TestList_EmptyDir(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestRegisterWithL7Fields(t *testing.T) {
|
||||
mgr := NewManager(t.TempDir())
|
||||
|
||||
svc := Service{
|
||||
Domain: "keila.cloud.payne.io",
|
||||
Source: "wild-cloud",
|
||||
Routes: []Route{{
|
||||
Paths: []string{"/.well-known/matrix", "/_matrix"},
|
||||
Backend: Backend{
|
||||
Address: "192.168.8.80:80",
|
||||
Type: BackendHTTP,
|
||||
},
|
||||
Headers: &HeaderConfig{
|
||||
Response: map[string]string{
|
||||
"Access-Control-Allow-Origin": "https://payne.io",
|
||||
},
|
||||
Request: map[string]string{
|
||||
"X-Forwarded-Proto": "https",
|
||||
},
|
||||
},
|
||||
IPAllow: []string{"10.0.0.0/8", "192.168.0.0/16"},
|
||||
}},
|
||||
}
|
||||
|
||||
if err := mgr.Register(svc); err != nil {
|
||||
t.Fatalf("Register failed: %v", err)
|
||||
}
|
||||
|
||||
got, err := mgr.Get("keila.cloud.payne.io")
|
||||
if err != nil {
|
||||
t.Fatalf("Get failed: %v", err)
|
||||
}
|
||||
|
||||
if len(got.Routes) != 1 {
|
||||
t.Fatalf("expected 1 route, got %d", len(got.Routes))
|
||||
}
|
||||
r := got.Routes[0]
|
||||
if len(r.Paths) != 2 || r.Paths[0] != "/.well-known/matrix" {
|
||||
t.Errorf("expected 2 paths, got %v", r.Paths)
|
||||
}
|
||||
if r.Headers == nil {
|
||||
t.Fatal("expected headers, got nil")
|
||||
}
|
||||
if r.Headers.Response["Access-Control-Allow-Origin"] != "https://payne.io" {
|
||||
t.Errorf("expected response header, got %v", r.Headers.Response)
|
||||
}
|
||||
if r.Headers.Request["X-Forwarded-Proto"] != "https" {
|
||||
t.Errorf("expected request header, got %v", r.Headers.Request)
|
||||
}
|
||||
if len(r.IPAllow) != 2 || r.IPAllow[0] != "10.0.0.0/8" {
|
||||
t.Errorf("expected 2 ipAllow CIDRs, got %v", r.IPAllow)
|
||||
}
|
||||
}
|
||||
|
||||
func TestUpdateL7Fields(t *testing.T) {
|
||||
mgr := NewManager(t.TempDir())
|
||||
|
||||
mgr.Register(Service{
|
||||
Domain: "updatable.example.com",
|
||||
Source: "test",
|
||||
Routes: []Route{{
|
||||
Paths: []string{"/api", "/webhook"},
|
||||
Backend: Backend{Address: "127.0.0.1:8080", Type: BackendHTTP},
|
||||
IPAllow: []string{"10.0.0.0/8"},
|
||||
Headers: &HeaderConfig{
|
||||
Response: map[string]string{
|
||||
"X-Custom": "value",
|
||||
},
|
||||
},
|
||||
}},
|
||||
})
|
||||
|
||||
got, _ := mgr.Get("updatable.example.com")
|
||||
if len(got.Routes) != 1 {
|
||||
t.Fatalf("expected 1 route, got %d", len(got.Routes))
|
||||
}
|
||||
r := got.Routes[0]
|
||||
if len(r.Paths) != 2 || r.Paths[0] != "/api" {
|
||||
t.Errorf("expected paths [/api /webhook], got %v", r.Paths)
|
||||
}
|
||||
if len(r.IPAllow) != 1 || r.IPAllow[0] != "10.0.0.0/8" {
|
||||
t.Errorf("expected ipAllow [10.0.0.0/8], got %v", r.IPAllow)
|
||||
}
|
||||
if r.Headers == nil || r.Headers.Response["X-Custom"] != "value" {
|
||||
t.Errorf("expected response header X-Custom=value, got %v", r.Headers)
|
||||
}
|
||||
}
|
||||
|
||||
func TestList_IgnoresNonYAML(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
mgr := NewManager(dir)
|
||||
|
||||
Reference in New Issue
Block a user