package transform import ( "bytes" "strings" "testing" ) const HTML_WITH_NAV = ` ` const EXPECTED_HTML_WITH_NAV_1 = ` ` func TestDegenerateNoSectionSet(t *testing.T) { var ( tr = new(NavActive) out = new(bytes.Buffer) ) if err := tr.Apply(out, strings.NewReader(HTML_WITH_NAV)); err != nil { t.Errorf("Unexpected error in NavActive.Apply: %s", err) } if out.String() != HTML_WITH_NAV { t.Errorf("NavActive.Apply should simply pass along the buffer unmodified.") } } func TestSetNav(t *testing.T) { tr := &NavActive{Section: "section_2"} out := new(bytes.Buffer) if err := tr.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()) } }