progress_test.go

package progress

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

func executeProgressRequest(path string) *httptest.ResponseRecorder {
	req := httptest.NewRequest(http.MethodGet, path, nil)
	rr := httptest.NewRecorder()
	Progress(rr, req)
	return rr
}

func executeProgressRequestWithMethod(method string, path string) *httptest.ResponseRecorder {
	req := httptest.NewRequest(method, path, nil)
	rr := httptest.NewRecorder()
	Progress(rr, req)
	return rr
}

func TestPickColorThresholds(t *testing.T) {
	if got := pickColor(10, "", "", ""); got != red {
		t.Fatalf("expected red for low percentage, got %q", got)
	}

	if got := pickColor(50, "", "", ""); got != yellow {
		t.Fatalf("expected yellow for medium percentage, got %q", got)
	}

	if got := pickColor(80, "", "", ""); got != green {
		t.Fatalf("expected green for high percentage, got %q", got)
	}
}

func TestPickColorWithCustomColors(t *testing.T) {
	if got := pickColor(10, "#00aa00", "#ffcc00", "#aa0000"); got != "#aa0000" {
		t.Fatalf("expected custom danger color, got %q", got)
	}

	if got := pickColor(50, "#00aa00", "#ffcc00", "#aa0000"); got != "#ffcc00" {
		t.Fatalf("expected custom warning color, got %q", got)
	}

	if got := pickColor(80, "#00aa00", "#ffcc00", "#aa0000"); got != "#00aa00" {
		t.Fatalf("expected custom success color, got %q", got)
	}
}

func TestProgressReturnsSVGForValidInput(t *testing.T) {
	rr := executeProgressRequest("/progress/76")

	if rr.Code != http.StatusOK {
		t.Fatalf("expected status 200, got %d", rr.Code)
	}

	if got := rr.Header().Get("Content-Type"); got != "image/svg+xml" {
		t.Fatalf("expected image/svg+xml content type, got %q", got)
	}

	if got := rr.Header().Get("Cache-Control"); got != cacheControlValue {
		t.Fatalf("expected cache control %q, got %q", cacheControlValue, got)
	}

	body := rr.Body.String()
	if !strings.Contains(body, "76%") {
		t.Fatalf("expected percentage text in SVG body, got %q", body)
	}
	if strings.Contains(body, "%%") {
		t.Fatalf("expected single percent symbol in SVG body, got %q", body)
	}

	if !strings.Contains(body, `width="68" height="20" fill="#5cb85c"`) {
		t.Fatalf("expected progress width/color in SVG body, got %q", body)
	}
}

func TestProgressSupportsFloatPercentages(t *testing.T) {
	rr := executeProgressRequest("/progress/76.5")
	if rr.Code != http.StatusOK {
		t.Fatalf("expected status 200, got %d", rr.Code)
	}

	body := rr.Body.String()
	if !strings.Contains(body, "76.5%") {
		t.Fatalf("expected decimal percentage in body, got %q", body)
	}
	if !strings.Contains(body, `width="68" height="20" fill="#5cb85c"`) {
		t.Fatalf("expected decimal percentage width/color in SVG body, got %q", body)
	}
}

func TestProgressSupportsDataBarMode(t *testing.T) {
	rr := executeProgressRequest("/progress/186?label=186&min=0&max=241&barColor=4472C4")
	if rr.Code != http.StatusOK {
		t.Fatalf("expected status 200, got %d", rr.Code)
	}

	body := rr.Body.String()
	if !strings.Contains(body, "186") {
		t.Fatalf("expected custom label in body, got %q", body)
	}
	if !strings.Contains(body, `width="69" height="20" fill="#4472c4"`) {
		t.Fatalf("expected scaled width/custom color in body, got %q", body)
	}
}

func TestProgressDataBarDefaultsLabelToRawValue(t *testing.T) {
	rr := executeProgressRequest("/progress/50?min=0&max=200")
	if rr.Code != http.StatusOK {
		t.Fatalf("expected status 200, got %d", rr.Code)
	}

	body := rr.Body.String()
	if !strings.Contains(body, "50\n") || strings.Contains(body, "50%") {
		t.Fatalf("expected raw value label in body, got %q", body)
	}
	if !strings.Contains(body, `width="22" height="20" fill="#d9534f"`) {
		t.Fatalf("expected scaled width and threshold color in body, got %q", body)
	}
}

func TestProgressSupportsHEAD(t *testing.T) {
	rr := executeProgressRequestWithMethod(http.MethodHead, "/progress/76")

	if rr.Code != http.StatusOK {
		t.Fatalf("expected status 200, got %d", rr.Code)
	}

	if got := rr.Header().Get("Content-Type"); got != "image/svg+xml" {
		t.Fatalf("expected image/svg+xml content type, got %q", got)
	}

	if got := rr.Header().Get("Cache-Control"); got != cacheControlValue {
		t.Fatalf("expected cache control %q, got %q", cacheControlValue, got)
	}

	if rr.Body.Len() != 0 {
		t.Fatalf("expected empty body for HEAD, got %q", rr.Body.String())
	}
}

func TestProgressRejectsUnsupportedMethod(t *testing.T) {
	rr := executeProgressRequestWithMethod(http.MethodPost, "/progress/50")

	if rr.Code != http.StatusMethodNotAllowed {
		t.Fatalf("expected status 405, got %d", rr.Code)
	}

	if got := rr.Header().Get("Allow"); got != "GET, HEAD" {
		t.Fatalf("expected Allow header for GET and HEAD, got %q", got)
	}
}

func TestProgressReturnsBadRequestForInvalidPercentage(t *testing.T) {
	rr := executeProgressRequest("/progress/not-a-number")

	if rr.Code != http.StatusBadRequest {
		t.Fatalf("expected status 400, got %d", rr.Code)
	}

	if !strings.Contains(rr.Body.String(), "percentage must be a number") {
		t.Fatalf("expected invalid percentage message, got %q", rr.Body.String())
	}
}

func TestProgressReturnsBadRequestForInvalidColor(t *testing.T) {
	rr := executeProgressRequest("/progress/50?successColor=nothex")

	if rr.Code != http.StatusBadRequest {
		t.Fatalf("expected status 400, got %d", rr.Code)
	}

	if !strings.Contains(rr.Body.String(), "successColor must be a 6-character hex value") {
		t.Fatalf("expected invalid color message, got %q", rr.Body.String())
	}
}

func TestProgressReturnsBadRequestForInvalidBarColor(t *testing.T) {
	rr := executeProgressRequest("/progress/50?barColor=nothex")

	if rr.Code != http.StatusBadRequest {
		t.Fatalf("expected status 400, got %d", rr.Code)
	}

	if !strings.Contains(rr.Body.String(), "barColor must be a 6-character hex value") {
		t.Fatalf("expected invalid bar color message, got %q", rr.Body.String())
	}
}

func TestProgressRequiresMinAndMaxTogether(t *testing.T) {
	rr := executeProgressRequest("/progress/50?min=0")
	if rr.Code != http.StatusBadRequest {
		t.Fatalf("expected status 400, got %d", rr.Code)
	}
	if !strings.Contains(rr.Body.String(), "min and max must be provided together") {
		t.Fatalf("expected min/max pair message, got %q", rr.Body.String())
	}
}

func TestProgressRejectsInvalidMinMaxValues(t *testing.T) {
	rr := executeProgressRequest("/progress/50?min=abc&max=100")
	if rr.Code != http.StatusBadRequest {
		t.Fatalf("expected status 400, got %d", rr.Code)
	}
	if !strings.Contains(rr.Body.String(), "min and max must be numeric values") {
		t.Fatalf("expected min/max numeric message, got %q", rr.Body.String())
	}
}

func TestProgressRejectsInvalidRangeOrder(t *testing.T) {
	rr := executeProgressRequest("/progress/50?min=10&max=10")
	if rr.Code != http.StatusBadRequest {
		t.Fatalf("expected status 400, got %d", rr.Code)
	}
	if !strings.Contains(rr.Body.String(), "max must be greater than min") {
		t.Fatalf("expected max greater than min message, got %q", rr.Body.String())
	}
}

func TestProgressRejectsTooLongLabel(t *testing.T) {
	tooLongLabel := strings.Repeat("x", 65)
	rr := executeProgressRequest("/progress/50?label=" + tooLongLabel)
	if rr.Code != http.StatusBadRequest {
		t.Fatalf("expected status 400, got %d", rr.Code)
	}
	if !strings.Contains(rr.Body.String(), "label is too long") {
		t.Fatalf("expected label length message, got %q", rr.Body.String())
	}
}

func TestProgressClampsPercentageToRange(t *testing.T) {
	high := executeProgressRequest("/progress/150")
	if high.Code != http.StatusOK {
		t.Fatalf("expected status 200, got %d", high.Code)
	}

	highBody := high.Body.String()
	if !strings.Contains(highBody, "100%") {
		t.Fatalf("expected clamped high percentage in body, got %q", highBody)
	}

	if !strings.Contains(highBody, `width="90" height="20" fill="#5cb85c"`) {
		t.Fatalf("expected full width for clamped high percentage, got %q", highBody)
	}

	low := executeProgressRequest("/progress/-10")
	if low.Code != http.StatusOK {
		t.Fatalf("expected status 200, got %d", low.Code)
	}

	lowBody := low.Body.String()
	if !strings.Contains(lowBody, "0%") {
		t.Fatalf("expected clamped low percentage in body, got %q", lowBody)
	}

	if !strings.Contains(lowBody, `width="0" height="20" fill="#d9534f"`) {
		t.Fatalf("expected empty width for clamped low percentage, got %q", lowBody)
	}
}