package main

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

func TestCommandBoundary(t *testing.T) {
	for _, tc := range []struct {
		name, method, body string
		allow              bool
		authErr            error
		status, calls      int
	}{
		{"viewer", "POST", `{"command":"stop"}`, false, nil, 403, 0},
		{"auth failure", "POST", `{"command":"stop"}`, false, errors.New("offline"), 503, 0},
		{"query refresh", "GET", `{"command":"stop"}`, true, nil, 405, 0},
		{"empty", "POST", `{"command":" "}`, true, nil, 400, 0},
		{"newline", "POST", `{"command":"list\nstop"}`, true, nil, 400, 0},
		{"nul", "POST", `{"command":"list\u0000stop"}`, true, nil, 400, 0},
		{"oversized", "POST", `{"command":"` + strings.Repeat("x", 1001) + `"}`, true, nil, 400, 0},
		{"admin", "POST", `{"command":"list"}`, true, nil, 200, 1},
	} {
		t.Run(tc.name, func(t *testing.T) {
			calls := 0
			c := &console{allowed: func(*http.Request) (bool, error) { return tc.allow, tc.authErr }, execute: func(context.Context, string) (string, error) { calls++; return "Nobody online", nil }}
			w := httptest.NewRecorder()
			c.routes().ServeHTTP(w, httptest.NewRequest(tc.method, "/command", strings.NewReader(tc.body)))
			if w.Code != tc.status || calls != tc.calls {
				t.Fatalf("status=%d calls=%d", w.Code, calls)
			}
		})
	}
}

func TestNoRetry(t *testing.T) {
	calls := 0
	c := &console{allowed: func(*http.Request) (bool, error) { return true, nil }, execute: func(context.Context, string) (string, error) { calls++; return "", errors.New("lost response") }}
	w := httptest.NewRecorder()
	c.routes().ServeHTTP(w, httptest.NewRequest("POST", "/command", strings.NewReader(`{"command":"say once"}`)))
	if w.Code != 502 || calls != 1 {
		t.Fatalf("status=%d calls=%d", w.Code, calls)
	}
}

func TestHealth(t *testing.T) {
	for _, tc := range []struct {
		name          string
		allow         bool
		err           error
		status, calls int
	}{
		{"viewer", false, nil, 403, 0},
		{"admin", true, nil, 200, 1},
		{"unavailable RCON", true, errors.New("offline"), 502, 1},
	} {
		t.Run(tc.name, func(t *testing.T) {
			checks, commands := 0, 0
			c := &console{
				allowed: func(*http.Request) (bool, error) { return tc.allow, nil },
				check:   func(context.Context) error { checks++; return tc.err },
				execute: func(context.Context, string) (string, error) { commands++; return "", nil },
			}
			w := httptest.NewRecorder()
			c.routes().ServeHTTP(w, httptest.NewRequest("GET", "/health", nil))
			if w.Code != tc.status || checks != tc.calls || commands != 0 {
				t.Fatalf("status=%d checks=%d commands=%d", w.Code, checks, commands)
			}
		})
	}
}