Support routes.

This commit is contained in:
2026-07-10 05:21:43 +00:00
parent 5c26c7530a
commit 43253ca120
8 changed files with 597 additions and 85 deletions

View File

@@ -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)