hugo/hugolib/page_toc_test.go

package hugolib

import (
	"testing"
)

func TestTableOfContents(t *testing.T) {
	text := `
Blah blah blah blah blah.

## AA

Blah blah blah blah blah.

### AAA

Blah blah blah blah blah.

## BB

Blah blah blah blah blah.

### BBB

Blah blah blah blah blah.
`

	markdown := RemoveSummaryDivider([]byte(text))
	toc := string(getTableOfContents(markdown))

	expected := `<nav>
<ul>
<li>
<ul>
<li><a href="#toc_0">AA</a>
<ul>
<li><a href="#toc_1">AAA</a></li>
</ul></li>
<li><a href="#toc_2">BB</a>
<ul>
<li><a href="#toc_3">BBB</a></li>
</ul></li>
</ul></li>
</ul>
</nav>
`

	if toc != expected {
		t.Errorf("Expected table of contents: %s, got: %s", expected, toc)
	}
}