package httpstream

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

	"github.com/gin-gonic/gin"
)

func setupTestContext() (*gin.Context, *httptest.ResponseRecorder) {
	gin.SetMode(gin.TestMode)
	w := httptest.NewRecorder()
	c, _ := gin.CreateTestContext(w)
	c.Request, _ = http.NewRequest("POST", "/test", nil)
	return c, w
}

func TestWriteSSEHeader(t *testing.T) {
	c, w := setupTestContext()
	WriteSSEHeader(c)

	if w.Header().Get("Content-Type") != "text/event-stream" {
		t.Errorf("expected Content-Type text/event-stream, got %s", w.Header().Get("Content-Type"))
	}
	if w.Header().Get("Cache-Control") != "no-cache" {
		t.Errorf("expected Cache-Control no-cache, got %s", w.Header().Get("Cache-Control"))
	}
	if w.Header().Get("Connection") != "keep-alive" {
		t.Errorf("expected Connection keep-alive, got %s", w.Header().Get("Connection"))
	}
}

func TestWriteSSEEvent(t *testing.T) {
	c, w := setupTestContext()
	payload := map[string]string{"key": "value"}
	ok := WriteSSEEvent(c, "test.event", payload)

	if !ok {
		t.Error("WriteSSEEvent should return true")
	}
	body := w.Body.String()
	if body == "" {
		t.Error("expected non-empty response body")
	}
}

func TestWriteDone(t *testing.T) {
	c, w := setupTestContext()
	ok := WriteDone(c)

	if !ok {
		t.Error("WriteDone should return true")
	}
	body := w.Body.String()
	if body != "data: [DONE]\n\n" {
		t.Errorf("expected 'data: [DONE]\\n\\n', got %q", body)
	}
}

func TestWriteImageStreamHeader(t *testing.T) {
	c, w := setupTestContext()
	WriteImageStreamHeader(c)

	if w.Header().Get("Content-Type") != "text/event-stream" {
		t.Errorf("expected Content-Type text/event-stream, got %s", w.Header().Get("Content-Type"))
	}
}

func TestWriteImageStreamDone(t *testing.T) {
	c, w := setupTestContext()
	ok := WriteImageStreamDone(c)

	if !ok {
		t.Error("WriteImageStreamDone should return true")
	}
	body := w.Body.String()
	if body != "data: [DONE]\n\n" {
		t.Errorf("expected 'data: [DONE]\\n\\n', got %q", body)
	}
}

func TestWriteImageStreamError(t *testing.T) {
	c, w := setupTestContext()
	WriteImageStreamError(c, 0, 1, "test error")

	body := w.Body.String()
	if body == "" {
		t.Error("expected non-empty response body")
	}
}