package router

import (
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestProxy_AddsAndRemoves(t *testing.T) {
	Proxy("app.example.com", "localhost:5000")
	defer RemoveProxy("app.example.com")

	p := Proxies()
	if p["app.example.com"] != "localhost:5000" {
		t.Errorf("expected localhost:5000, got %q", p["app.example.com"])
	}

	RemoveProxy("app.example.com")
	p = Proxies()
	if _, ok := p["app.example.com"]; ok {
		t.Error("expected proxy to be removed")
	}
}

func TestProxy_CaseInsensitive(t *testing.T) {
	Proxy("APP.Example.COM", "backend:3000")
	defer RemoveProxy("app.example.com")

	p := Proxies()
	if p["app.example.com"] != "backend:3000" {
		t.Errorf("expected case-insensitive storage, got %v", p)
	}
}

func TestProxies_ReturnsCopy(t *testing.T) {
	Proxy("test.example.com", "backend:4000")
	defer RemoveProxy("test.example.com")

	p := Proxies()
	p["test.example.com"] = "modified"

	p2 := Proxies()
	if p2["test.example.com"] != "backend:4000" {
		t.Error("Proxies() should return a copy, not a reference")
	}
}

func TestHandler_FallsThrough(t *testing.T) {
	fallback := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
		w.Write([]byte("fallback"))
	})

	h := Handler(fallback)
	rec := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "http://unknown.example.com/", nil)
	h.ServeHTTP(rec, req)

	if rec.Code != http.StatusOK {
		t.Errorf("expected 200, got %d", rec.Code)
	}
	if rec.Body.String() != "fallback" {
		t.Errorf("expected fallback body, got %q", rec.Body.String())
	}
}