mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-07 20:30:36 -05:00
3d9c4f513b
The list handling is surprisingly expensive: ``` ▶ go test -run="NONE" -bench="BenchmarkFrontmatterTags" -test.benchmem=true ./parser | prettybench PASS benchmark iter time/iter bytes alloc allocs --------- ---- --------- ----------- ------ BenchmarkFrontmatterTags/JSON:1-4 1000000 2039.00 ns/op 912 B/op 20 allocs/op BenchmarkFrontmatterTags/JSON:11-4 300000 5202.00 ns/op 1640 B/op 44 allocs/op BenchmarkFrontmatterTags/JSON:21-4 200000 7993.00 ns/op 2392 B/op 65 allocs/op BenchmarkFrontmatterTags/YAML:1-4 200000 9359.00 ns/op 5928 B/op 66 allocs/op BenchmarkFrontmatterTags/YAML:11-4 100000 21218.00 ns/op 8408 B/op 140 allocs/op BenchmarkFrontmatterTags/YAML:21-4 50000 32852.00 ns/op 10920 B/op 211 allocs/op BenchmarkFrontmatterTags/TOML:1-4 100000 21505.00 ns/op 9231 B/op 173 allocs/op BenchmarkFrontmatterTags/TOML:11-4 20000 82919.00 ns/op 19808 B/op 625 allocs/op BenchmarkFrontmatterTags/TOML:21-4 10000 141847.00 ns/op 31200 B/op 1106 allocs/op ok github.com/spf13/hugo/parser 17.890s ``` See #3464
389 lines
9.1 KiB
Go
389 lines
9.1 KiB
Go
// Copyright 2015 The Hugo Authors. All rights reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package parser
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestInterfaceToConfig(t *testing.T) {
|
|
cases := []struct {
|
|
input interface{}
|
|
mark byte
|
|
want []byte
|
|
isErr bool
|
|
}{
|
|
// TOML
|
|
{map[string]interface{}{}, TOMLLead[0], nil, false},
|
|
{
|
|
map[string]interface{}{"title": "test 1"},
|
|
TOMLLead[0],
|
|
[]byte("title = \"test 1\"\n"),
|
|
false,
|
|
},
|
|
|
|
// YAML
|
|
{map[string]interface{}{}, YAMLLead[0], []byte("{}\n"), false},
|
|
{
|
|
map[string]interface{}{"title": "test 1"},
|
|
YAMLLead[0],
|
|
[]byte("title: test 1\n"),
|
|
false,
|
|
},
|
|
|
|
// JSON
|
|
{map[string]interface{}{}, JSONLead[0], []byte("{}\n"), false},
|
|
{
|
|
map[string]interface{}{"title": "test 1"},
|
|
JSONLead[0],
|
|
[]byte("{\n \"title\": \"test 1\"\n}\n"),
|
|
false,
|
|
},
|
|
|
|
// Errors
|
|
{nil, TOMLLead[0], nil, true},
|
|
{map[string]interface{}{}, '$', nil, true},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
var buf bytes.Buffer
|
|
|
|
err := InterfaceToConfig(c.input, rune(c.mark), &buf)
|
|
if err != nil {
|
|
if c.isErr {
|
|
continue
|
|
}
|
|
t.Fatalf("[%d] unexpected error value: %v", i, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(buf.Bytes(), c.want) {
|
|
t.Errorf("[%d] not equal:\nwant %q,\n got %q", i, c.want, buf.Bytes())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestInterfaceToFrontMatter(t *testing.T) {
|
|
cases := []struct {
|
|
input interface{}
|
|
mark rune
|
|
want []byte
|
|
isErr bool
|
|
}{
|
|
// TOML
|
|
{map[string]interface{}{}, '+', []byte("+++\n\n+++\n"), false},
|
|
{
|
|
map[string]interface{}{"title": "test 1"},
|
|
'+',
|
|
[]byte("+++\ntitle = \"test 1\"\n\n+++\n"),
|
|
false,
|
|
},
|
|
|
|
// YAML
|
|
{map[string]interface{}{}, '-', []byte("---\n{}\n---\n"), false}, //
|
|
{
|
|
map[string]interface{}{"title": "test 1"},
|
|
'-',
|
|
[]byte("---\ntitle: test 1\n---\n"),
|
|
false,
|
|
},
|
|
|
|
// JSON
|
|
{map[string]interface{}{}, '{', []byte("{}\n"), false},
|
|
{
|
|
map[string]interface{}{"title": "test 1"},
|
|
'{',
|
|
[]byte("{\n \"title\": \"test 1\"\n}\n"),
|
|
false,
|
|
},
|
|
|
|
// Errors
|
|
{nil, '+', nil, true},
|
|
{map[string]interface{}{}, '$', nil, true},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
var buf bytes.Buffer
|
|
err := InterfaceToFrontMatter(c.input, c.mark, &buf)
|
|
if err != nil {
|
|
if c.isErr {
|
|
continue
|
|
}
|
|
t.Fatalf("[%d] unexpected error value: %v", i, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(buf.Bytes(), c.want) {
|
|
t.Errorf("[%d] not equal:\nwant %q,\n got %q", i, c.want, buf.Bytes())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHandleTOMLMetaData(t *testing.T) {
|
|
cases := []struct {
|
|
input []byte
|
|
want interface{}
|
|
isErr bool
|
|
}{
|
|
{nil, map[string]interface{}{}, false},
|
|
{[]byte("title = \"test 1\""), map[string]interface{}{"title": "test 1"}, false},
|
|
{[]byte("a = [1, 2, 3]"), map[string]interface{}{"a": []interface{}{int64(1), int64(2), int64(3)}}, false},
|
|
{[]byte("b = [\n[1, 2],\n[3, 4]\n]"), map[string]interface{}{"b": []interface{}{[]interface{}{int64(1), int64(2)}, []interface{}{int64(3), int64(4)}}}, false},
|
|
// errors
|
|
{[]byte("z = [\n[1, 2]\n[3, 4]\n]"), nil, true},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
res, err := HandleTOMLMetaData(c.input)
|
|
if err != nil {
|
|
if c.isErr {
|
|
continue
|
|
}
|
|
t.Fatalf("[%d] unexpected error value: %v", i, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(res, c.want) {
|
|
t.Errorf("[%d] not equal: given %q\nwant %#v,\n got %#v", i, c.input, c.want, res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHandleYAMLMetaData(t *testing.T) {
|
|
cases := []struct {
|
|
input []byte
|
|
want interface{}
|
|
isErr bool
|
|
}{
|
|
{nil, map[string]interface{}{}, false},
|
|
{[]byte("title: test 1"), map[string]interface{}{"title": "test 1"}, false},
|
|
{[]byte("a: Easy!\nb:\n c: 2\n d: [3, 4]"), map[string]interface{}{"a": "Easy!", "b": map[interface{}]interface{}{"c": 2, "d": []interface{}{3, 4}}}, false},
|
|
// errors
|
|
{[]byte("z = not toml"), nil, true},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
res, err := HandleYAMLMetaData(c.input)
|
|
if err != nil {
|
|
if c.isErr {
|
|
continue
|
|
}
|
|
t.Fatalf("[%d] unexpected error value: %v", i, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(res, c.want) {
|
|
t.Errorf("[%d] not equal: given %q\nwant %#v,\n got %#v", i, c.input, c.want, res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHandleJSONMetaData(t *testing.T) {
|
|
cases := []struct {
|
|
input []byte
|
|
want interface{}
|
|
isErr bool
|
|
}{
|
|
{nil, map[string]interface{}{}, false},
|
|
{[]byte("{\"title\": \"test 1\"}"), map[string]interface{}{"title": "test 1"}, false},
|
|
// errors
|
|
{[]byte("{noquotes}"), nil, true},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
res, err := HandleJSONMetaData(c.input)
|
|
if err != nil {
|
|
if c.isErr {
|
|
continue
|
|
}
|
|
t.Fatalf("[%d] unexpected error value: %v", i, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(res, c.want) {
|
|
t.Errorf("[%d] not equal: given %q\nwant %#v,\n got %#v", i, c.input, c.want, res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHandleOrgMetaData(t *testing.T) {
|
|
cases := []struct {
|
|
input []byte
|
|
want interface{}
|
|
isErr bool
|
|
}{
|
|
{nil, map[string]interface{}{}, false},
|
|
{[]byte("#+title: test 1\n"), map[string]interface{}{"title": "test 1"}, false},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
res, err := HandleOrgMetaData(c.input)
|
|
if err != nil {
|
|
if c.isErr {
|
|
continue
|
|
}
|
|
t.Fatalf("[%d] unexpected error value: %v", i, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(res, c.want) {
|
|
t.Errorf("[%d] not equal: given %q\nwant %#v,\n got %#v", i, c.input, c.want, res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFormatToLeadRune(t *testing.T) {
|
|
for i, this := range []struct {
|
|
kind string
|
|
expect rune
|
|
}{
|
|
{"yaml", '-'},
|
|
{"yml", '-'},
|
|
{"toml", '+'},
|
|
{"tml", '+'},
|
|
{"json", '{'},
|
|
{"js", '{'},
|
|
{"org", '#'},
|
|
{"unknown", '+'},
|
|
} {
|
|
result := FormatToLeadRune(this.kind)
|
|
|
|
if result != this.expect {
|
|
t.Errorf("[%d] got %q but expected %q", i, result, this.expect)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDetectFrontMatter(t *testing.T) {
|
|
cases := []struct {
|
|
mark rune
|
|
want *FrontmatterType
|
|
}{
|
|
// funcs are uncomparable, so we ignore FrontmatterType.Parse in these tests
|
|
{'-', &FrontmatterType{nil, []byte(YAMLDelim), []byte(YAMLDelim), false}},
|
|
{'+', &FrontmatterType{nil, []byte(TOMLDelim), []byte(TOMLDelim), false}},
|
|
{'{', &FrontmatterType{nil, []byte("{"), []byte("}"), true}},
|
|
{'#', &FrontmatterType{nil, []byte("#+"), []byte("\n"), false}},
|
|
{'$', nil},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
res := DetectFrontMatter(c.mark)
|
|
if res == nil {
|
|
if c.want == nil {
|
|
continue
|
|
}
|
|
|
|
t.Fatalf("want %v, got %v", *c.want, res)
|
|
}
|
|
|
|
if !reflect.DeepEqual(res.markstart, c.want.markstart) {
|
|
t.Errorf("markstart mismatch: want %v, got %v", c.want.markstart, res.markstart)
|
|
}
|
|
if !reflect.DeepEqual(res.markend, c.want.markend) {
|
|
t.Errorf("markend mismatch: want %v, got %v", c.want.markend, res.markend)
|
|
}
|
|
if !reflect.DeepEqual(res.includeMark, c.want.includeMark) {
|
|
t.Errorf("includeMark mismatch: want %v, got %v", c.want.includeMark, res.includeMark)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRemoveTOMLIdentifier(t *testing.T) {
|
|
cases := []struct {
|
|
input string
|
|
want string
|
|
}{
|
|
{"a = 1", "a = 1"},
|
|
{"a = 1\r\n", "a = 1\r\n"},
|
|
{"+++\r\na = 1\r\n+++\r\n", "a = 1\r\n"},
|
|
{"+++\na = 1\n+++\n", "a = 1\n"},
|
|
{"+++\nb = \"+++ oops +++\"\n+++\n", "b = \"+++ oops +++\"\n"},
|
|
{"+++\nc = \"\"\"+++\noops\n+++\n\"\"\"\"\n+++\n", "c = \"\"\"+++\noops\n+++\n\"\"\"\"\n"},
|
|
{"+++\nd = 1\n+++", "d = 1\n"},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
res := removeTOMLIdentifier([]byte(c.input))
|
|
if string(res) != c.want {
|
|
t.Errorf("[%d] given %q\nwant: %q\n got: %q", i, c.input, c.want, res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkFrontmatterTags(b *testing.B) {
|
|
|
|
for _, frontmatter := range []string{"JSON", "YAML", "TOML"} {
|
|
for i := 1; i < 30; i += 10 {
|
|
doBenchmarkFrontmatter(b, frontmatter, i)
|
|
}
|
|
}
|
|
}
|
|
|
|
func doBenchmarkFrontmatter(b *testing.B, fileformat string, numTags int) {
|
|
yamlTemplate := `---
|
|
name: "Tags"
|
|
tags:
|
|
%s
|
|
---
|
|
`
|
|
tomlTemplate := `+++
|
|
name = "Tags"
|
|
tags = %s
|
|
+++
|
|
`
|
|
|
|
jsonTemplate := `{
|
|
"name": "Tags",
|
|
"tags": [
|
|
%s
|
|
]
|
|
}`
|
|
name := fmt.Sprintf("%s:%d", fileformat, numTags)
|
|
b.Run(name, func(b *testing.B) {
|
|
tags := make([]string, numTags)
|
|
var (
|
|
tagsStr string
|
|
frontmatterTemplate string
|
|
)
|
|
for i := 0; i < numTags; i++ {
|
|
tags[i] = fmt.Sprintf("Hugo %d", i+1)
|
|
}
|
|
if fileformat == "TOML" {
|
|
frontmatterTemplate = tomlTemplate
|
|
tagsStr = strings.Replace(fmt.Sprintf("%q", tags), " ", ", ", -1)
|
|
} else if fileformat == "JSON" {
|
|
frontmatterTemplate = jsonTemplate
|
|
tagsStr = strings.Replace(fmt.Sprintf("%q", tags), " ", ", ", -1)
|
|
} else {
|
|
frontmatterTemplate = yamlTemplate
|
|
for _, tag := range tags {
|
|
tagsStr += "\n- " + tag
|
|
}
|
|
}
|
|
|
|
frontmatter := fmt.Sprintf(frontmatterTemplate, tagsStr)
|
|
|
|
p := page{frontmatter: []byte(frontmatter)}
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
meta, err := p.Metadata()
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
if meta == nil {
|
|
b.Fatal("Meta is nil")
|
|
}
|
|
}
|
|
})
|
|
}
|