package handler

import (
	"io"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"

	"chis_osi/osi"
)

func newTestClient(t *testing.T, body string) (*osi.Client, *string) {
	t.Helper()
	var seenPath string
	osiServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		seenPath = r.URL.Path
		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(body))
	}))
	t.Cleanup(osiServer.Close)

	transport, err := osi.NewTransport(osi.TransportConfig{Timeout: time.Second})
	if err != nil {
		t.Fatalf("NewTransport: %v", err)
	}
	client := osi.NewClient(osi.ClientConfig{BaseURL: osiServer.URL, UserName: "u", Ask: "k", Transport: transport})
	return client, &seenPath
}

func TestHealthCheckLastReturnsRawResponse(t *testing.T) {
	client, seenPath := newTestClient(t, `{"code":"01","message":"操作成功","data":{"checkId":"CHK-1","idcard":"TEST-1"}}`)
	h := NewHealthCheckHandler(client)

	req := httptest.NewRequest(http.MethodGet, "/api/health-check/last?idCard=TEST-1", nil)
	rec := httptest.NewRecorder()
	h.Last(rec, req)

	if rec.Code != http.StatusOK {
		t.Fatalf("status = %d", rec.Code)
	}
	if *seenPath != "/osi/api/auto/jktjlscx/query" {
		t.Fatalf("osi path = %q", *seenPath)
	}
	body, _ := io.ReadAll(rec.Body)
	if !strings.Contains(string(body), `"CHK-1"`) {
		t.Fatalf("body missing content: %s", body)
	}
}

func TestHealthCheckLastRequiresIdentifier(t *testing.T) {
	h := NewHealthCheckHandler(nil) // 无标识符在触达 client 前返回 400
	req := httptest.NewRequest(http.MethodGet, "/api/health-check/last", nil)
	rec := httptest.NewRecorder()
	h.Last(rec, req)
	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status = %d, want 400", rec.Code)
	}
}

func TestHealthCheckAllReturnsArrayRaw(t *testing.T) {
	client, seenPath := newTestClient(t, `{"code":"01","message":"操作成功","data":[{"checkId":"CHK-1","idCard":"TEST-1"},{"checkId":null,"idCard":"TEST-1"}]}`)
	h := NewHealthCheckHandler(client)

	req := httptest.NewRequest(http.MethodGet, "/api/health-check/all?idCard=TEST-1", nil)
	rec := httptest.NewRecorder()
	h.All(rec, req)

	if rec.Code != http.StatusOK {
		t.Fatalf("status = %d", rec.Code)
	}
	if *seenPath != "/osi/api/auto/jktj/query" {
		t.Fatalf("osi path = %q", *seenPath)
	}
	body, _ := io.ReadAll(rec.Body)
	if !strings.Contains(string(body), `"CHK-1"`) {
		t.Fatalf("body missing records: %s", body)
	}
}

func TestHealthCheckAllRequiresIdentifier(t *testing.T) {
	h := NewHealthCheckHandler(nil)
	req := httptest.NewRequest(http.MethodGet, "/api/health-check/all", nil)
	rec := httptest.NewRecorder()
	h.All(rec, req)
	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status = %d, want 400", rec.Code)
	}
}

func TestHealthCheckListRequiresCheckYear(t *testing.T) {
	h := NewHealthCheckHandler(nil) // 缺 checkYear 在触达 client 前返回 400
	req := httptest.NewRequest(http.MethodGet, "/api/health-check/list?idCard=TEST-1", nil)
	rec := httptest.NewRecorder()
	h.List(rec, req)
	if rec.Code != http.StatusBadRequest {
		t.Fatalf("status = %d, want 400", rec.Code)
	}
}

func TestHealthCheckListReturnsRosterRaw(t *testing.T) {
	client, seenPath := newTestClient(t, `{"code":"01","message":"操作成功","data":[{"idCard":"TEST-1","checkType":"0"}]}`)
	h := NewHealthCheckHandler(client)

	req := httptest.NewRequest(http.MethodGet, "/api/health-check/list?checkYear=2025&idCard=TEST-1", nil)
	rec := httptest.NewRecorder()
	h.List(rec, req)

	if rec.Code != http.StatusOK {
		t.Fatalf("status = %d", rec.Code)
	}
	if *seenPath != "/osi/api/auto/jktjlist/query" {
		t.Fatalf("osi path = %q", *seenPath)
	}
	body, _ := io.ReadAll(rec.Body)
	if !strings.Contains(string(body), `"checkType"`) {
		t.Fatalf("body missing roster: %s", body)
	}
}