package hugolib import ( "bytes" "strings" "testing" "github.com/spf13/hugo/source" "github.com/spf13/hugo/target" ) const ALIAS_DOC_1 = "---\ntitle: alias doc\naliases:\n - \"alias1/\"\n - \"alias-2/\"\n---\naliases\n" var fakeSource = []source.ByteSource{ { Name: "foo/bar/file.md", Content: []byte(SIMPLE_PAGE), }, { Name: "alias/test/file1.md", Content: []byte(ALIAS_DOC_1), }, { Name: "section/somecontent.html", Content: []byte(RENDER_NO_FRONT_MATTER), }, } func stringInSlice(a string, list []string) bool { for _, b := range list { if b == a { return true } } return false } func checkShowPlanExpected(t *testing.T, s *Site, expected string) { out := new(bytes.Buffer) if err := s.ShowPlan(out); err != nil { t.Fatalf("ShowPlan unexpectedly returned an error: %s", err) } got := out.String() gotList := strings.Split(got, "\n") expectedList := strings.Split(expected, "\n") for _, x := range gotList { if !stringInSlice(x, expectedList) { t.Errorf("%v %v %v %v", "\nShowPlan expected:\n", expected, "\ngot:\n", got) } } for _, x := range expectedList { if !stringInSlice(x, gotList) { t.Errorf("%v %v %v %v", "\nShowPlan expected:\n", expected, "\ngot:\n", got) } } } func TestDegenerateNoFiles(t *testing.T) { checkShowPlanExpected(t, new(Site), "No source files provided.\n") } func TestDegenerateNoTarget(t *testing.T) { s := &Site{ Source: &source.InMemorySource{ByteSource: fakeSource}, } must(s.CreatePages()) expected := "foo/bar/file.md (renderer: markdown)\n canonical => !no target specified!\n\n" + "alias/test/file1.md (renderer: markdown)\n canonical => !no target specified!\n\n" + "section/somecontent.html (renderer: n/a)\n canonical => !no target specified!\n\n" checkShowPlanExpected(t, s, expected) } func TestFileTarget(t *testing.T) { s := &Site{ Source: &source.InMemorySource{ByteSource: fakeSource}, } s.AliasTarget() s.PageTarget() must(s.CreatePages()) expected := "foo/bar/file.md (renderer: markdown)\n canonical => foo/bar/file/index.html\n\n" + "alias/test/file1.md (renderer: markdown)\n" + " canonical => alias/test/file1/index.html\n" + " alias1/ => alias1/index.html\n" + " alias-2/ => alias-2/index.html\n\n" + "section/somecontent.html (renderer: n/a)\n canonical => section/somecontent/index.html\n\n" checkShowPlanExpected(t, s, expected) } func TestPageTargetUgly(t *testing.T) { s := &Site{ Targets: targetList{Page: &target.PagePub{UglyUrls: true}}, Source: &source.InMemorySource{ByteSource: fakeSource}, } s.AliasTarget() s.CreatePages() expected := "foo/bar/file.md (renderer: markdown)\n canonical => foo/bar/file.html\n\n" + "alias/test/file1.md (renderer: markdown)\n" + " canonical => alias/test/file1.html\n" + " alias1/ => alias1/index.html\n" + " alias-2/ => alias-2/index.html\n\n" + "section/somecontent.html (renderer: n/a)\n canonical => section/somecontent.html\n\n" checkShowPlanExpected(t, s, expected) } func TestFileTargetPublishDir(t *testing.T) { s := &Site{ Targets: targetList{ Page: &target.PagePub{PublishDir: "../public"}, Alias: &target.HTMLRedirectAlias{PublishDir: "../public"}, }, Source: &source.InMemorySource{ByteSource: fakeSource}, } must(s.CreatePages()) expected := "foo/bar/file.md (renderer: markdown)\n canonical => ../public/foo/bar/file/index.html\n\n" + "alias/test/file1.md (renderer: markdown)\n" + " canonical => ../public/alias/test/file1/index.html\n" + " alias1/ => ../public/alias1/index.html\n" + " alias-2/ => ../public/alias-2/index.html\n\n" + "section/somecontent.html (renderer: n/a)\n canonical => ../public/section/somecontent/index.html\n\n" checkShowPlanExpected(t, s, expected) }