mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-21 20:46:30 -05:00
5e28606b84
`GroupBy` is modified to allow it to receive a method name argument for example `Type` as its first argument. It is only allowed to call with a method which takes no arguments and returns a result or a pair of a result and an error. The functions discussed at #443 are also added - `ByPublishDate`: Order contents by `PublishDate` front matter variable - `GroupByPublishDate(format, order)`: Group contents by `PublishDate` front matter variable formatted in string like `GroupByDate` - `GroupByParam(key, order)`: Group contents by `Param` front matter variable specified by `key` argument - `GroupByParamDate(key, format, order)`: Group contents by `Param` front matter variable specified by `key` argument and formatted in string like `GroupByDate`. It's effective against `time.Time` type front matter variable
739 lines
23 KiB
Go
739 lines
23 KiB
Go
package hugolib
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"html/template"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/spf13/hugo/helpers"
|
|
"github.com/spf13/hugo/source"
|
|
"github.com/spf13/hugo/target"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
const (
|
|
TEMPLATE_TITLE = "{{ .Title }}"
|
|
PAGE_SIMPLE_TITLE = `---
|
|
title: simple template
|
|
---
|
|
content`
|
|
|
|
TEMPLATE_MISSING_FUNC = "{{ .Title | funcdoesnotexists }}"
|
|
TEMPLATE_FUNC = "{{ .Title | urlize }}"
|
|
TEMPLATE_CONTENT = "{{ .Content }}"
|
|
TEMPLATE_DATE = "{{ .Date }}"
|
|
INVALID_TEMPLATE_FORMAT_DATE = "{{ .Date.Format time.RFC3339 }}"
|
|
TEMPLATE_WITH_URL_REL = "<a href=\"foobar.jpg\">Going</a>"
|
|
TEMPLATE_WITH_URL_ABS = "<a href=\"/foobar.jpg\">Going</a>"
|
|
PAGE_URL_SPECIFIED = `---
|
|
title: simple template
|
|
url: "mycategory/my-whatever-content/"
|
|
---
|
|
content`
|
|
|
|
PAGE_WITH_MD = `---
|
|
title: page with md
|
|
---
|
|
# heading 1
|
|
text
|
|
## heading 2
|
|
more text
|
|
`
|
|
)
|
|
|
|
func pageMust(p *Page, err error) *Page {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return p
|
|
}
|
|
|
|
func TestDegenerateRenderThingMissingTemplate(t *testing.T) {
|
|
p, _ := NewPageFrom(strings.NewReader(PAGE_SIMPLE_TITLE), "content/a/file.md")
|
|
p.Convert()
|
|
s := new(Site)
|
|
s.prepTemplates()
|
|
err := s.renderThing(p, "foobar", nil)
|
|
if err == nil {
|
|
t.Errorf("Expected err to be returned when missing the template.")
|
|
}
|
|
}
|
|
|
|
func TestAddInvalidTemplate(t *testing.T) {
|
|
s := new(Site)
|
|
s.prepTemplates()
|
|
err := s.addTemplate("missing", TEMPLATE_MISSING_FUNC)
|
|
if err == nil {
|
|
t.Fatalf("Expecting the template to return an error")
|
|
}
|
|
}
|
|
|
|
type nopCloser struct {
|
|
io.Writer
|
|
}
|
|
|
|
func (nopCloser) Close() error { return nil }
|
|
|
|
func NopCloser(w io.Writer) io.WriteCloser {
|
|
return nopCloser{w}
|
|
}
|
|
|
|
func matchRender(t *testing.T, s *Site, p *Page, tmplName string, expected string) {
|
|
content := new(bytes.Buffer)
|
|
err := s.renderThing(p, tmplName, NopCloser(content))
|
|
if err != nil {
|
|
t.Fatalf("Unable to render template.")
|
|
}
|
|
|
|
if string(content.Bytes()) != expected {
|
|
t.Fatalf("Content did not match expected: %s. got: %s", expected, content)
|
|
}
|
|
}
|
|
|
|
func TestRenderThing(t *testing.T) {
|
|
tests := []struct {
|
|
content string
|
|
template string
|
|
expected string
|
|
}{
|
|
{PAGE_SIMPLE_TITLE, TEMPLATE_TITLE, "simple template"},
|
|
{PAGE_SIMPLE_TITLE, TEMPLATE_FUNC, "simple-template"},
|
|
{PAGE_WITH_MD, TEMPLATE_CONTENT, "\n\n<h1 id=\"toc_0\">heading 1</h1>\n\n<p>text</p>\n\n<h2 id=\"toc_1\">heading 2</h2>\n\n<p>more text</p>\n"},
|
|
{SIMPLE_PAGE_RFC3339_DATE, TEMPLATE_DATE, "2013-05-17 16:59:30 +0000 UTC"},
|
|
}
|
|
|
|
s := new(Site)
|
|
s.prepTemplates()
|
|
|
|
for i, test := range tests {
|
|
p, err := NewPageFrom(strings.NewReader(test.content), "content/a/file.md")
|
|
p.Convert()
|
|
if err != nil {
|
|
t.Fatalf("Error parsing buffer: %s", err)
|
|
}
|
|
templateName := fmt.Sprintf("foobar%d", i)
|
|
err = s.addTemplate(templateName, test.template)
|
|
if err != nil {
|
|
t.Fatalf("Unable to add template")
|
|
}
|
|
|
|
p.Content = template.HTML(p.Content)
|
|
html := new(bytes.Buffer)
|
|
err = s.renderThing(p, templateName, NopCloser(html))
|
|
if err != nil {
|
|
t.Errorf("Unable to render html: %s", err)
|
|
}
|
|
|
|
if string(html.Bytes()) != test.expected {
|
|
t.Errorf("Content does not match.\nExpected\n\t'%q'\ngot\n\t'%q'", test.expected, html)
|
|
}
|
|
}
|
|
}
|
|
|
|
func HTML(in string) string {
|
|
return in
|
|
}
|
|
|
|
func TestRenderThingOrDefault(t *testing.T) {
|
|
tests := []struct {
|
|
content string
|
|
missing bool
|
|
template string
|
|
expected string
|
|
}{
|
|
{PAGE_SIMPLE_TITLE, true, TEMPLATE_TITLE, HTML("simple template")},
|
|
{PAGE_SIMPLE_TITLE, true, TEMPLATE_FUNC, HTML("simple-template")},
|
|
{PAGE_SIMPLE_TITLE, false, TEMPLATE_TITLE, HTML("simple template")},
|
|
{PAGE_SIMPLE_TITLE, false, TEMPLATE_FUNC, HTML("simple-template")},
|
|
}
|
|
|
|
files := make(map[string][]byte)
|
|
target := &target.InMemoryTarget{Files: files}
|
|
s := &Site{
|
|
Target: target,
|
|
}
|
|
s.prepTemplates()
|
|
|
|
for i, test := range tests {
|
|
p, err := NewPageFrom(strings.NewReader(PAGE_SIMPLE_TITLE), "content/a/file.md")
|
|
if err != nil {
|
|
t.Fatalf("Error parsing buffer: %s", err)
|
|
}
|
|
templateName := fmt.Sprintf("default%d", i)
|
|
err = s.addTemplate(templateName, test.template)
|
|
if err != nil {
|
|
t.Fatalf("Unable to add template")
|
|
}
|
|
|
|
var err2 error
|
|
if test.missing {
|
|
err2 = s.render("name", p, "out", "missing", templateName)
|
|
} else {
|
|
err2 = s.render("name", p, "out", templateName, "missing_default")
|
|
}
|
|
|
|
if err2 != nil {
|
|
t.Errorf("Unable to render html: %s", err)
|
|
}
|
|
|
|
if string(files["out"]) != test.expected {
|
|
t.Errorf("Content does not match. Expected '%s', got '%s'", test.expected, files["out"])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestTargetPath(t *testing.T) {
|
|
tests := []struct {
|
|
doc string
|
|
content string
|
|
expectedOutFile string
|
|
expectedSection string
|
|
}{
|
|
{"content/a/file.md", PAGE_URL_SPECIFIED, "mycategory/my-whatever-content/index.html", "a"},
|
|
{"content/x/y/deepfile.md", SIMPLE_PAGE, "x/y/deepfile.html", "x/y"},
|
|
{"content/x/y/z/deeperfile.md", SIMPLE_PAGE, "x/y/z/deeperfile.html", "x/y/z"},
|
|
{"content/b/file.md", SIMPLE_PAGE, "b/file.html", "b"},
|
|
{"a/file.md", SIMPLE_PAGE, "a/file.html", "a"},
|
|
{"file.md", SIMPLE_PAGE, "file.html", ""},
|
|
}
|
|
|
|
if true {
|
|
return
|
|
}
|
|
|
|
for _, test := range tests {
|
|
p := pageMust(NewPageFrom(strings.NewReader(test.content), helpers.AbsPathify(test.doc)))
|
|
|
|
expected := test.expectedOutFile
|
|
|
|
if p.TargetPath() != expected {
|
|
t.Errorf("%s => OutFile expected: '%s', got: '%s'", test.doc, expected, p.TargetPath())
|
|
}
|
|
|
|
if p.Section() != test.expectedSection {
|
|
t.Errorf("%s => p.Section expected: %s, got: %s", test.doc, test.expectedSection, p.Section)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDraftAndFutureRender(t *testing.T) {
|
|
files := make(map[string][]byte)
|
|
target := &target.InMemoryTarget{Files: files}
|
|
sources := []source.ByteSource{
|
|
{"sect/doc1.md", []byte("---\ntitle: doc1\ndraft: true\npublishdate: \"2414-05-29\"\n---\n# doc1\n*some content*")},
|
|
{"sect/doc2.md", []byte("---\ntitle: doc2\ndraft: true\npublishdate: \"2012-05-29\"\n---\n# doc2\n*some content*")},
|
|
{"sect/doc3.md", []byte("---\ntitle: doc3\ndraft: false\npublishdate: \"2414-05-29\"\n---\n# doc3\n*some content*")},
|
|
{"sect/doc4.md", []byte("---\ntitle: doc4\ndraft: false\npublishdate: \"2012-05-29\"\n---\n# doc4\n*some content*")},
|
|
}
|
|
|
|
siteSetup := func() *Site {
|
|
s := &Site{
|
|
Target: target,
|
|
Source: &source.InMemorySource{ByteSource: sources},
|
|
}
|
|
|
|
s.initializeSiteInfo()
|
|
|
|
if err := s.CreatePages(); err != nil {
|
|
t.Fatalf("Unable to create pages: %s", err)
|
|
}
|
|
return s
|
|
}
|
|
|
|
viper.Set("baseurl", "http://auth/bub")
|
|
|
|
// Testing Defaults.. Only draft:true and publishDate in the past should be rendered
|
|
s := siteSetup()
|
|
if len(s.Pages) != 1 {
|
|
t.Fatal("Draft or Future dated content published unexpectedly")
|
|
}
|
|
|
|
// only publishDate in the past should be rendered
|
|
viper.Set("BuildDrafts", true)
|
|
s = siteSetup()
|
|
if len(s.Pages) != 2 {
|
|
t.Fatal("Future Dated Posts published unexpectedly")
|
|
}
|
|
|
|
// drafts should not be rendered, but all dates should
|
|
viper.Set("BuildDrafts", false)
|
|
viper.Set("BuildFuture", true)
|
|
s = siteSetup()
|
|
if len(s.Pages) != 2 {
|
|
t.Fatal("Draft posts published unexpectedly")
|
|
}
|
|
|
|
// all 4 should be included
|
|
viper.Set("BuildDrafts", true)
|
|
viper.Set("BuildFuture", true)
|
|
s = siteSetup()
|
|
if len(s.Pages) != 4 {
|
|
t.Fatal("Drafts or Future posts not included as expected")
|
|
}
|
|
|
|
//setting defaults back
|
|
viper.Set("BuildDrafts", false)
|
|
viper.Set("BuildFuture", false)
|
|
}
|
|
|
|
func TestSkipRender(t *testing.T) {
|
|
files := make(map[string][]byte)
|
|
target := &target.InMemoryTarget{Files: files}
|
|
sources := []source.ByteSource{
|
|
{"sect/doc1.html", []byte("---\nmarkup: markdown\n---\n# title\nsome *content*")},
|
|
{"sect/doc2.html", []byte("<!doctype html><html><body>more content</body></html>")},
|
|
{"sect/doc3.md", []byte("# doc3\n*some* content")},
|
|
{"sect/doc4.md", []byte("---\ntitle: doc4\n---\n# doc4\n*some content*")},
|
|
{"sect/doc5.html", []byte("<!doctype html><html>{{ template \"head\" }}<body>body5</body></html>")},
|
|
{"sect/doc6.html", []byte("<!doctype html><html>{{ template \"head_abs\" }}<body>body5</body></html>")},
|
|
{"doc7.html", []byte("<html><body>doc7 content</body></html>")},
|
|
{"sect/doc8.html", []byte("---\nmarkup: md\n---\n# title\nsome *content*")},
|
|
}
|
|
|
|
viper.Set("verbose", true)
|
|
viper.Set("CanonifyUrls", true)
|
|
viper.Set("baseurl", "http://auth/bub")
|
|
s := &Site{
|
|
Target: target,
|
|
Source: &source.InMemorySource{ByteSource: sources},
|
|
}
|
|
|
|
s.initializeSiteInfo()
|
|
s.prepTemplates()
|
|
|
|
must(s.addTemplate("_default/single.html", "{{.Content}}"))
|
|
must(s.addTemplate("head", "<head><script src=\"script.js\"></script></head>"))
|
|
must(s.addTemplate("head_abs", "<head><script src=\"/script.js\"></script></head>"))
|
|
|
|
if err := s.CreatePages(); err != nil {
|
|
t.Fatalf("Unable to create pages: %s", err)
|
|
}
|
|
|
|
if err := s.BuildSiteMeta(); err != nil {
|
|
t.Fatalf("Unable to build site metadata: %s", err)
|
|
}
|
|
|
|
if err := s.RenderPages(); err != nil {
|
|
t.Fatalf("Unable to render pages. %s", err)
|
|
}
|
|
|
|
tests := []struct {
|
|
doc string
|
|
expected string
|
|
}{
|
|
{"sect/doc1.html", "\n\n<h1 id=\"toc_0\">title</h1>\n\n<p>some <em>content</em></p>\n"},
|
|
{"sect/doc2.html", "<!doctype html><html><body>more content</body></html>"},
|
|
{"sect/doc3.html", "\n\n<h1 id=\"toc_0\">doc3</h1>\n\n<p><em>some</em> content</p>\n"},
|
|
{"sect/doc4.html", "\n\n<h1 id=\"toc_0\">doc4</h1>\n\n<p><em>some content</em></p>\n"},
|
|
{"sect/doc5.html", "<!doctype html><html><head><script src=\"script.js\"></script></head><body>body5</body></html>"},
|
|
{"sect/doc6.html", "<!doctype html><html><head><script src=\"http://auth/bub/script.js\"></script></head><body>body5</body></html>"},
|
|
{"doc7.html", "<html><body>doc7 content</body></html>"},
|
|
{"sect/doc8.html", "\n\n<h1 id=\"toc_0\">title</h1>\n\n<p>some <em>content</em></p>\n"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
content, ok := target.Files[test.doc]
|
|
if !ok {
|
|
t.Fatalf("Did not find %s in target. %v", test.doc, target.Files)
|
|
}
|
|
|
|
if !bytes.Equal(content, []byte(test.expected)) {
|
|
t.Errorf("%s content expected:\n%q\ngot:\n%q", test.doc, test.expected, string(content))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAbsUrlify(t *testing.T) {
|
|
files := make(map[string][]byte)
|
|
target := &target.InMemoryTarget{Files: files}
|
|
sources := []source.ByteSource{
|
|
{"sect/doc1.html", []byte("<!doctype html><html><head></head><body><a href=\"#frag1\">link</a></body></html>")},
|
|
{"content/blue/doc2.html", []byte("---\nf: t\n---\n<!doctype html><html><body>more content</body></html>")},
|
|
}
|
|
for _, canonify := range []bool{true, false} {
|
|
viper.Set("CanonifyUrls", canonify)
|
|
viper.Set("BaseUrl", "http://auth/bub")
|
|
s := &Site{
|
|
Target: target,
|
|
Source: &source.InMemorySource{ByteSource: sources},
|
|
}
|
|
t.Logf("Rendering with BaseUrl %q and CanonifyUrls set %v", viper.GetString("baseUrl"), canonify)
|
|
s.initializeSiteInfo()
|
|
s.prepTemplates()
|
|
must(s.addTemplate("blue/single.html", TEMPLATE_WITH_URL_ABS))
|
|
|
|
if err := s.CreatePages(); err != nil {
|
|
t.Fatalf("Unable to create pages: %s", err)
|
|
}
|
|
|
|
if err := s.BuildSiteMeta(); err != nil {
|
|
t.Fatalf("Unable to build site metadata: %s", err)
|
|
}
|
|
|
|
if err := s.RenderPages(); err != nil {
|
|
t.Fatalf("Unable to render pages. %s", err)
|
|
}
|
|
|
|
tests := []struct {
|
|
file, expected string
|
|
}{
|
|
{"content/blue/doc2.html", "<a href=\"http://auth/bub/foobar.jpg\">Going</a>"},
|
|
{"sect/doc1.html", "<!doctype html><html><head></head><body><a href=\"#frag1\">link</a></body></html>"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
content, ok := target.Files[test.file]
|
|
if !ok {
|
|
t.Fatalf("Unable to locate rendered content: %s", test.file)
|
|
}
|
|
|
|
expected := test.expected
|
|
if !canonify {
|
|
expected = strings.Replace(expected, viper.GetString("baseurl"), "", -1)
|
|
}
|
|
if string(content) != expected {
|
|
t.Errorf("AbsUrlify content expected:\n%q\ngot\n%q", expected, string(content))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var WEIGHTED_PAGE_1 = []byte(`+++
|
|
weight = "2"
|
|
title = "One"
|
|
my_param = "foo"
|
|
my_date = 1979-05-27T07:32:00Z
|
|
+++
|
|
Front Matter with Ordered Pages`)
|
|
|
|
var WEIGHTED_PAGE_2 = []byte(`+++
|
|
weight = "6"
|
|
title = "Two"
|
|
publishdate = "2012-03-05"
|
|
my_param = "foo"
|
|
+++
|
|
Front Matter with Ordered Pages 2`)
|
|
|
|
var WEIGHTED_PAGE_3 = []byte(`+++
|
|
weight = "4"
|
|
title = "Three"
|
|
date = "2012-04-06"
|
|
publishdate = "2012-04-06"
|
|
my_param = "bar"
|
|
only_one = "yes"
|
|
my_date = 2010-05-27T07:32:00Z
|
|
+++
|
|
Front Matter with Ordered Pages 3`)
|
|
|
|
var WEIGHTED_PAGE_4 = []byte(`+++
|
|
weight = "4"
|
|
title = "Four"
|
|
date = "2012-01-01"
|
|
publishdate = "2012-01-01"
|
|
my_param = "baz"
|
|
my_date = 2010-05-27T07:32:00Z
|
|
+++
|
|
Front Matter with Ordered Pages 4. This is longer content`)
|
|
|
|
var WEIGHTED_SOURCES = []source.ByteSource{
|
|
{"sect/doc1.md", WEIGHTED_PAGE_1},
|
|
{"sect/doc2.md", WEIGHTED_PAGE_2},
|
|
{"sect/doc3.md", WEIGHTED_PAGE_3},
|
|
{"sect/doc4.md", WEIGHTED_PAGE_4},
|
|
}
|
|
|
|
func TestOrderedPages(t *testing.T) {
|
|
files := make(map[string][]byte)
|
|
target := &target.InMemoryTarget{Files: files}
|
|
|
|
viper.Set("baseurl", "http://auth/bub")
|
|
s := &Site{
|
|
Target: target,
|
|
Source: &source.InMemorySource{ByteSource: WEIGHTED_SOURCES},
|
|
}
|
|
s.initializeSiteInfo()
|
|
|
|
if err := s.CreatePages(); err != nil {
|
|
t.Fatalf("Unable to create pages: %s", err)
|
|
}
|
|
|
|
if err := s.BuildSiteMeta(); err != nil {
|
|
t.Fatalf("Unable to build site metadata: %s", err)
|
|
}
|
|
|
|
if s.Sections["sect"][0].Weight != 2 || s.Sections["sect"][3].Weight != 6 {
|
|
t.Errorf("Pages in unexpected order. First should be '%d', got '%d'", 2, s.Sections["sect"][0].Weight)
|
|
}
|
|
|
|
if s.Sections["sect"][1].Page.Title != "Three" || s.Sections["sect"][2].Page.Title != "Four" {
|
|
t.Errorf("Pages in unexpected order. Second should be '%s', got '%s'", "Three", s.Sections["sect"][1].Page.Title)
|
|
}
|
|
|
|
bydate := s.Pages.ByDate()
|
|
|
|
if bydate[0].Title != "One" {
|
|
t.Errorf("Pages in unexpected order. First should be '%s', got '%s'", "One", bydate[0].Title)
|
|
}
|
|
|
|
rev := bydate.Reverse()
|
|
if rev[0].Title != "Three" {
|
|
t.Errorf("Pages in unexpected order. First should be '%s', got '%s'", "Three", rev[0].Title)
|
|
}
|
|
|
|
bypubdate := s.Pages.ByPublishDate()
|
|
|
|
if bypubdate[0].Title != "One" {
|
|
t.Errorf("Pages in unexpected order. First should be '%s', got '%s'", "One", bypubdate[0].Title)
|
|
}
|
|
|
|
rbypubdate := bypubdate.Reverse()
|
|
if rbypubdate[0].Title != "Three" {
|
|
t.Errorf("Pages in unexpected order. First should be '%s', got '%s'", "Three", rbypubdate[0].Title)
|
|
}
|
|
|
|
bylength := s.Pages.ByLength()
|
|
if bylength[0].Title != "One" {
|
|
t.Errorf("Pages in unexpected order. First should be '%s', got '%s'", "One", bylength[0].Title)
|
|
}
|
|
|
|
rbylength := bylength.Reverse()
|
|
if rbylength[0].Title != "Four" {
|
|
t.Errorf("Pages in unexpected order. First should be '%s', got '%s'", "Four", rbylength[0].Title)
|
|
}
|
|
}
|
|
|
|
var GROUPED_SOURCES = []source.ByteSource{
|
|
{"sect1/doc1.md", WEIGHTED_PAGE_1},
|
|
{"sect1/doc2.md", WEIGHTED_PAGE_2},
|
|
{"sect2/doc3.md", WEIGHTED_PAGE_3},
|
|
{"sect3/doc4.md", WEIGHTED_PAGE_4},
|
|
}
|
|
|
|
func TestGroupedPages(t *testing.T) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
fmt.Println("Recovered in f", r)
|
|
}
|
|
}()
|
|
files := make(map[string][]byte)
|
|
target := &target.InMemoryTarget{Files: files}
|
|
|
|
viper.Set("baseurl", "http://auth/bub")
|
|
s := &Site{
|
|
Target: target,
|
|
Source: &source.InMemorySource{ByteSource: GROUPED_SOURCES},
|
|
}
|
|
s.initializeSiteInfo()
|
|
|
|
if err := s.CreatePages(); err != nil {
|
|
t.Fatalf("Unable to create pages: %s", err)
|
|
}
|
|
|
|
if err := s.BuildSiteMeta(); err != nil {
|
|
t.Fatalf("Unable to build site metadata: %s", err)
|
|
}
|
|
|
|
rbysection, err := s.Pages.GroupBy("Section", "desc")
|
|
if err != nil {
|
|
t.Fatalf("Unable to make PageGroup array: %s", err)
|
|
}
|
|
if rbysection[0].Key != "sect3" {
|
|
t.Errorf("PageGroup array in unexpected order. First group key should be '%s', got '%s'", "sect3", rbysection[0].Key)
|
|
}
|
|
if rbysection[1].Key != "sect2" {
|
|
t.Errorf("PageGroup array in unexpected order. Second group key should be '%s', got '%s'", "sect2", rbysection[1].Key)
|
|
}
|
|
if rbysection[2].Key != "sect1" {
|
|
t.Errorf("PageGroup array in unexpected order. Third group key should be '%s', got '%s'", "sect1", rbysection[2].Key)
|
|
}
|
|
if rbysection[0].Pages[0].Title != "Four" {
|
|
t.Errorf("PageGroup has an unexpected page. First group's pages should have '%s', got '%s'", "Four", rbysection[0].Pages[0].Title)
|
|
}
|
|
if len(rbysection[2].Pages) != 2 {
|
|
t.Errorf("PageGroup has unexpected number of pages. Third group should have '%d' pages, got '%d' pages", 2, len(rbysection[2].Pages))
|
|
}
|
|
|
|
bytype, err := s.Pages.GroupBy("Type", "asc")
|
|
if err != nil {
|
|
t.Fatalf("Unable to make PageGroup array: %s", err)
|
|
}
|
|
if bytype[0].Key != "sect1" {
|
|
t.Errorf("PageGroup array in unexpected order. First group key should be '%s', got '%s'", "sect1", bytype[0].Key)
|
|
}
|
|
if bytype[1].Key != "sect2" {
|
|
t.Errorf("PageGroup array in unexpected order. Second group key should be '%s', got '%s'", "sect2", bytype[1].Key)
|
|
}
|
|
if bytype[2].Key != "sect3" {
|
|
t.Errorf("PageGroup array in unexpected order. Third group key should be '%s', got '%s'", "sect3", bytype[2].Key)
|
|
}
|
|
if bytype[2].Pages[0].Title != "Four" {
|
|
t.Errorf("PageGroup has an unexpected page. Third group's data should have '%s', got '%s'", "Four", bytype[0].Pages[0].Title)
|
|
}
|
|
if len(bytype[0].Pages) != 2 {
|
|
t.Errorf("PageGroup has unexpected number of pages. First group should have '%d' pages, got '%d' pages", 2, len(bytype[2].Pages))
|
|
}
|
|
|
|
bydate, err := s.Pages.GroupByDate("2006-01", "asc")
|
|
if err != nil {
|
|
t.Fatalf("Unable to make PageGroup array: %s", err)
|
|
}
|
|
if bydate[0].Key != "0001-01" {
|
|
t.Errorf("PageGroup array in unexpected order. First group key should be '%s', got '%s'", "0001-01", bydate[0].Key)
|
|
}
|
|
if bydate[1].Key != "2012-01" {
|
|
t.Errorf("PageGroup array in unexpected order. Second group key should be '%s', got '%s'", "2012-01", bydate[1].Key)
|
|
}
|
|
if bydate[2].Key != "2012-04" {
|
|
t.Errorf("PageGroup array in unexpected order. Third group key should be '%s', got '%s'", "2012-04", bydate[2].Key)
|
|
}
|
|
if bydate[2].Pages[0].Title != "Three" {
|
|
t.Errorf("PageGroup has an unexpected page. Third group's pages should have '%s', got '%s'", "Three", bydate[2].Pages[0].Title)
|
|
}
|
|
if len(bydate[0].Pages) != 2 {
|
|
t.Errorf("PageGroup has unexpected number of pages. First group should have '%d' pages, got '%d' pages", 2, len(bydate[2].Pages))
|
|
}
|
|
|
|
bypubdate, err := s.Pages.GroupByPublishDate("2006")
|
|
if err != nil {
|
|
t.Fatalf("Unable to make PageGroup array: %s", err)
|
|
}
|
|
if bypubdate[0].Key != "2012" {
|
|
t.Errorf("PageGroup array in unexpected order. First group key should be '%s', got '%s'", "2012", bypubdate[0].Key)
|
|
}
|
|
if bypubdate[1].Key != "0001" {
|
|
t.Errorf("PageGroup array in unexpected order. Second group key should be '%s', got '%s'", "0001", bypubdate[1].Key)
|
|
}
|
|
if bypubdate[0].Pages[0].Title != "Three" {
|
|
t.Errorf("PageGroup has an unexpected page. Third group's pages should have '%s', got '%s'", "Three", bypubdate[0].Pages[0].Title)
|
|
}
|
|
if len(bypubdate[0].Pages) != 3 {
|
|
t.Errorf("PageGroup has unexpected number of pages. First group should have '%d' pages, got '%d' pages", 3, len(bypubdate[0].Pages))
|
|
}
|
|
|
|
byparam, err := s.Pages.GroupByParam("my_param", "desc")
|
|
if err != nil {
|
|
t.Fatalf("Unable to make PageGroup array: %s", err)
|
|
}
|
|
if byparam[0].Key != "foo" {
|
|
t.Errorf("PageGroup array in unexpected order. First group key should be '%s', got '%s'", "foo", byparam[0].Key)
|
|
}
|
|
if byparam[1].Key != "baz" {
|
|
t.Errorf("PageGroup array in unexpected order. Second group key should be '%s', got '%s'", "baz", byparam[1].Key)
|
|
}
|
|
if byparam[2].Key != "bar" {
|
|
t.Errorf("PageGroup array in unexpected order. Third group key should be '%s', got '%s'", "bar", byparam[2].Key)
|
|
}
|
|
if byparam[2].Pages[0].Title != "Three" {
|
|
t.Errorf("PageGroup has an unexpected page. Third group's pages should have '%s', got '%s'", "Three", byparam[2].Pages[0].Title)
|
|
}
|
|
if len(byparam[0].Pages) != 2 {
|
|
t.Errorf("PageGroup has unexpected number of pages. First group should have '%d' pages, got '%d' pages", 2, len(byparam[0].Pages))
|
|
}
|
|
|
|
_, err = s.Pages.GroupByParam("not_exist")
|
|
if err == nil {
|
|
t.Errorf("GroupByParam didn't return an expected error")
|
|
}
|
|
|
|
byOnlyOneParam, err := s.Pages.GroupByParam("only_one")
|
|
if err != nil {
|
|
t.Fatalf("Unable to make PageGroup array: %s", err)
|
|
}
|
|
if len(byOnlyOneParam) != 1 {
|
|
t.Errorf("PageGroup array has unexpected elements. Group length should be '%d', got '%d'", 1, len(byOnlyOneParam))
|
|
}
|
|
if byOnlyOneParam[0].Key != "yes" {
|
|
t.Errorf("PageGroup array in unexpected order. First group key should be '%s', got '%s'", "yes", byOnlyOneParam[0].Key)
|
|
}
|
|
|
|
byParamDate, err := s.Pages.GroupByParamDate("my_date", "2006-01")
|
|
if err != nil {
|
|
t.Fatalf("Unable to make PageGroup array: %s", err)
|
|
}
|
|
if byParamDate[0].Key != "2010-05" {
|
|
t.Errorf("PageGroup array in unexpected order. First group key should be '%s', got '%s'", "2010-05", byParamDate[0].Key)
|
|
}
|
|
if byParamDate[1].Key != "1979-05" {
|
|
t.Errorf("PageGroup array in unexpected order. Second group key should be '%s', got '%s'", "1979-05", byParamDate[1].Key)
|
|
}
|
|
if byParamDate[1].Pages[0].Title != "One" {
|
|
t.Errorf("PageGroup has an unexpected page. Second group's pages should have '%s', got '%s'", "One", byParamDate[1].Pages[0].Title)
|
|
}
|
|
if len(byParamDate[0].Pages) != 2 {
|
|
t.Errorf("PageGroup has unexpected number of pages. First group should have '%d' pages, got '%d' pages", 2, len(byParamDate[2].Pages))
|
|
}
|
|
}
|
|
|
|
var PAGE_WITH_WEIGHTED_TAXONOMIES_2 = []byte(`+++
|
|
tags = [ "a", "b", "c" ]
|
|
tags_weight = 22
|
|
categories = ["d"]
|
|
title = "foo"
|
|
categories_weight = 44
|
|
+++
|
|
Front Matter with weighted tags and categories`)
|
|
|
|
var PAGE_WITH_WEIGHTED_TAXONOMIES_1 = []byte(`+++
|
|
tags = "a"
|
|
tags_weight = 33
|
|
title = "bar"
|
|
categories = [ "d", "e" ]
|
|
categories_weight = 11
|
|
alias = "spf13"
|
|
date = 1979-05-27T07:32:00Z
|
|
+++
|
|
Front Matter with weighted tags and categories`)
|
|
|
|
var PAGE_WITH_WEIGHTED_TAXONOMIES_3 = []byte(`+++
|
|
title = "bza"
|
|
categories = [ "e" ]
|
|
categories_weight = 11
|
|
alias = "spf13"
|
|
date = 2010-05-27T07:32:00Z
|
|
+++
|
|
Front Matter with weighted tags and categories`)
|
|
|
|
func TestWeightedTaxonomies(t *testing.T) {
|
|
files := make(map[string][]byte)
|
|
target := &target.InMemoryTarget{Files: files}
|
|
sources := []source.ByteSource{
|
|
{"sect/doc1.md", PAGE_WITH_WEIGHTED_TAXONOMIES_1},
|
|
{"sect/doc2.md", PAGE_WITH_WEIGHTED_TAXONOMIES_2},
|
|
{"sect/doc3.md", PAGE_WITH_WEIGHTED_TAXONOMIES_3},
|
|
}
|
|
taxonomies := make(map[string]string)
|
|
|
|
taxonomies["tag"] = "tags"
|
|
taxonomies["category"] = "categories"
|
|
|
|
viper.Set("baseurl", "http://auth/bub")
|
|
viper.Set("taxonomies", taxonomies)
|
|
s := &Site{
|
|
Target: target,
|
|
Source: &source.InMemorySource{ByteSource: sources},
|
|
}
|
|
s.initializeSiteInfo()
|
|
|
|
if err := s.CreatePages(); err != nil {
|
|
t.Fatalf("Unable to create pages: %s", err)
|
|
}
|
|
|
|
if err := s.BuildSiteMeta(); err != nil {
|
|
t.Fatalf("Unable to build site metadata: %s", err)
|
|
}
|
|
|
|
if s.Taxonomies["tags"]["a"][0].Page.Title != "foo" {
|
|
t.Errorf("Pages in unexpected order, 'foo' expected first, got '%v'", s.Taxonomies["tags"]["a"][0].Page.Title)
|
|
}
|
|
|
|
if s.Taxonomies["categories"]["d"][0].Page.Title != "bar" {
|
|
t.Errorf("Pages in unexpected order, 'bar' expected first, got '%v'", s.Taxonomies["categories"]["d"][0].Page.Title)
|
|
}
|
|
|
|
if s.Taxonomies["categories"]["e"][0].Page.Title != "bza" {
|
|
t.Errorf("Pages in unexpected order, 'bza' expected first, got '%v'", s.Taxonomies["categories"]["e"][0].Page.Title)
|
|
}
|
|
}
|