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
}