package controller

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestDetectionHTMLRequiresCompletedDocument(t *testing.T) {
	raw := []byte(`{"status":"completed","output":[{"type":"message","content":[{"type":"output_text","text":"<html><body>pelican</body></html>"}]}]}`)
	assert.Equal(t, "<html><body>pelican</body></html>", detectionHTML(raw, "application/json"))
	assert.Empty(t, detectionHTML([]byte(`{"status":"incomplete","output":[]}`), "application/json"))
	sse := []byte("data: {\"type\":\"response.completed\",\"response\":" + string(raw) + "}\n\n")
	assert.Equal(t, detectionHTML(raw, "application/json"), detectionHTML(sse, "text/event-stream"))
}