shithub: hugo

ref: b22364570be6c6add79af21a90de8c5f7a775813
dir: /transform/nav_test.go/

View raw version 
package transform

import (
	"bytes"
	"strings"
	"testing"
)

const HTML_WITH_NAV = `<!DOCTYPE html>
<html>
<head></head>
<body>
<nav>
	<ul class="nav navbar-nav">
		<li hugo-nav="section_1"><a href="#">Section 1</a></li>
		<li hugo-nav="section_2"><a href="#">Section 2</a></li>
	</ul>
</nav>
</body>
</html>
`
const EXPECTED_HTML_WITH_NAV_1 = `<!DOCTYPE html><html><head></head>
<body>
<nav>
	<ul class="nav navbar-nav">
		<li hugo-nav="section_1"><a href="#">Section 1</a></li>
		<li hugo-nav="section_2" class="active"><a href="#">Section 2</a></li>
	</ul>
</nav>


</body></html>`

func TestSetNav(t *testing.T) {
	trs := NavActive("section_2", "hugo-nav")
	chain := NewChain(trs...)
	out := new(bytes.Buffer)
	if err := chain.Apply(out, strings.NewReader(HTML_WITH_NAV)); err != nil {
		t.Errorf("Unexpected error in Apply() for NavActive: %s", err)
	}

	expected := EXPECTED_HTML_WITH_NAV_1
	if out.String() != expected {
		t.Errorf("NavActive.Apply output expected and got:\n%q\n%q", expected, out.String())
	}
}

func BenchmarkTransform(b *testing.B) {
	tr := NavActive("section_2", "hugo-nav")
	chain := NewChain(tr...)
	out := new(bytes.Buffer)
	for i := 0; i < b.N; i++ {
		if err := chain.Apply(out, strings.NewReader(HTML_WITH_NAV)); err != nil {
			b.Errorf("Unexpected error in Apply() for NavActive: %s", err)
		}
		out.Reset()
	}
}