64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package openapi
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"go.yaml.in/yaml/v4"
|
|
)
|
|
|
|
type document struct {
|
|
OpenAPI string `yaml:"openapi"`
|
|
Paths map[string]map[string]any `yaml:"paths"`
|
|
}
|
|
|
|
func TestDistributionContract(t *testing.T) {
|
|
spec := readDocument(t, "proxy-pool.yaml")
|
|
if spec.OpenAPI != "3.1.0" {
|
|
t.Fatalf("openapi version = %q, want 3.1.0", spec.OpenAPI)
|
|
}
|
|
extraction, ok := spec.Paths["/api/v1/proxies/extract"]
|
|
if !ok {
|
|
t.Fatal("exclusive extraction path is missing")
|
|
}
|
|
if _, ok := extraction["post"]; !ok {
|
|
t.Fatal("exclusive extraction must use POST")
|
|
}
|
|
for path := range spec.Paths {
|
|
if path == "/api/v1/leases" || path == "/api/v1/proxies/release" || path == "/api/v1/proxies/renew" {
|
|
t.Fatalf("lease/release path is forbidden: %s", path)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAdminContract(t *testing.T) {
|
|
spec := readDocument(t, "admin.yaml")
|
|
for _, path := range []string{
|
|
"/api/v1/status",
|
|
"/api/v1/upstreams/{name}/enable",
|
|
"/api/v1/upstreams/{name}/disable",
|
|
"/api/v1/routing/{name}/switch",
|
|
"/api/v1/config/reload",
|
|
} {
|
|
if _, ok := spec.Paths[path]; !ok {
|
|
t.Errorf("admin path is missing: %s", path)
|
|
}
|
|
}
|
|
}
|
|
|
|
func readDocument(t *testing.T, path string) document {
|
|
t.Helper()
|
|
content, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("read %s: %v", path, err)
|
|
}
|
|
var spec document
|
|
if err := yaml.Unmarshal(content, &spec); err != nil {
|
|
t.Fatalf("parse %s: %v", path, err)
|
|
}
|
|
if spec.OpenAPI != "3.1.0" {
|
|
t.Fatalf("%s openapi version = %q, want 3.1.0", path, spec.OpenAPI)
|
|
}
|
|
return spec
|
|
}
|