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)
}