package template

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestExtract_Basic(t *testing.T) {
	doc := `<root>
<!-- MD_CONTENT_START -->
<h1>Title</h1>
<p>Content here</p>
<!-- MD_CONTENT_END -->
</root>`

	result, err := Extract(doc, DefaultMarkerStart, DefaultMarkerEnd)
	require.NoError(t, err)
	assert.Equal(t, "<h1>Title</h1>\n<p>Content here</p>", result)
}

func TestExtract_CustomMarkers(t *testing.T) {
	doc := `<!-- BEGIN -->Hello<!-- END -->`
	result, err := Extract(doc, "<!-- BEGIN -->", "<!-- END -->")
	require.NoError(t, err)
	assert.Equal(t, "Hello", result)
}

func TestExtract_MissingStartMarker(t *testing.T) {
	doc := `<root><!-- MD_CONTENT_END --></root>`
	_, err := Extract(doc, DefaultMarkerStart, DefaultMarkerEnd)
	require.Error(t, err)
	assert.Contains(t, err.Error(), "start marker")
}

func TestExtract_MissingEndMarker(t *testing.T) {
	doc := `<root><!-- MD_CONTENT_START --></root>`
	_, err := Extract(doc, DefaultMarkerStart, DefaultMarkerEnd)
	require.Error(t, err)
	assert.Contains(t, err.Error(), "end marker")
}

func TestExtract_EmptyContent(t *testing.T) {
	doc := `<!-- MD_CONTENT_START --><!-- MD_CONTENT_END -->`
	result, err := Extract(doc, DefaultMarkerStart, DefaultMarkerEnd)
	require.NoError(t, err)
	assert.Empty(t, result)
}