mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-07 20:30:36 -05:00
Adding more test cases
This commit is contained in:
parent
e67db666c8
commit
e26b43f6d9
3 changed files with 78 additions and 16 deletions
|
@ -1,8 +1,8 @@
|
||||||
package hugolib
|
package hugolib
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"os"
|
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -12,7 +12,7 @@ func BenchmarkParsePage(b *testing.B) {
|
||||||
sample.ReadFrom(f)
|
sample.ReadFrom(f)
|
||||||
b.ResetTimer()
|
b.ResetTimer()
|
||||||
for i := 0; i < b.N; i++ {
|
for i := 0; i < b.N; i++ {
|
||||||
ReadFrom(sample, "bench")
|
ReadFrom(sample, "bench")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -303,25 +303,28 @@ func (page *Page) GetParam(key string) interface{} {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO return error on last line instead of nil
|
var ErrDetectingFrontMatter = errors.New("unable to detect front matter")
|
||||||
|
var ErrMatchingStartingFrontMatterDelimiter = errors.New("unable to match beginning front matter delimiter")
|
||||||
|
var ErrMatchingEndingFrontMatterDelimiter = errors.New("unable to match ending front matter delimiter")
|
||||||
|
|
||||||
func (page *Page) parseFrontMatter(data *bufio.Reader) (err error) {
|
func (page *Page) parseFrontMatter(data *bufio.Reader) (err error) {
|
||||||
|
|
||||||
if err = checkEmpty(data); err != nil {
|
if err = checkEmpty(data); err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
var mark rune
|
var mark rune
|
||||||
if mark, err = chompWhitespace(data); err != nil {
|
if mark, err = chompWhitespace(data); err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
f := page.detectFrontMatter(mark)
|
f := page.detectFrontMatter(mark)
|
||||||
if f == nil {
|
if f == nil {
|
||||||
return errors.New("unable to match beginning front matter delimiter")
|
return ErrDetectingFrontMatter
|
||||||
}
|
}
|
||||||
|
|
||||||
if found, err := beginFrontMatter(data, f); err != nil || !found {
|
if found, err := beginFrontMatter(data, f); err != nil || !found {
|
||||||
return errors.New("unable to match beginning front matter delimiter")
|
return ErrMatchingStartingFrontMatterDelimiter
|
||||||
}
|
}
|
||||||
|
|
||||||
var frontmatter = new(bytes.Buffer)
|
var frontmatter = new(bytes.Buffer)
|
||||||
|
@ -329,24 +332,29 @@ func (page *Page) parseFrontMatter(data *bufio.Reader) (err error) {
|
||||||
line, _, err := data.ReadLine()
|
line, _, err := data.ReadLine()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == io.EOF {
|
if err == io.EOF {
|
||||||
return errors.New("unable to match ending front matter delimiter")
|
return ErrMatchingEndingFrontMatterDelimiter
|
||||||
}
|
}
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
if bytes.Equal(line, f.markend) {
|
if bytes.Equal(line, f.markend) {
|
||||||
|
if f.includeMark {
|
||||||
|
frontmatter.Write(line)
|
||||||
|
}
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
frontmatter.Write(line)
|
frontmatter.Write(line)
|
||||||
frontmatter.Write([]byte{'\n'})
|
frontmatter.Write([]byte{'\n'})
|
||||||
}
|
}
|
||||||
|
|
||||||
metadata, err := f.parse(frontmatter.Bytes())
|
metadata, err := f.parse(frontmatter.Bytes())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if err = page.update(metadata); err != nil {
|
if err = page.update(metadata); err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
return
|
return
|
||||||
|
@ -365,24 +373,31 @@ func checkEmpty(data *bufio.Reader) (err error) {
|
||||||
type frontmatterType struct {
|
type frontmatterType struct {
|
||||||
markstart, markend []byte
|
markstart, markend []byte
|
||||||
parse func([]byte) (interface{}, error)
|
parse func([]byte) (interface{}, error)
|
||||||
|
includeMark bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func (page *Page) detectFrontMatter(mark rune) (f *frontmatterType) {
|
func (page *Page) detectFrontMatter(mark rune) (f *frontmatterType) {
|
||||||
switch mark {
|
switch mark {
|
||||||
case '-':
|
case '-':
|
||||||
return &frontmatterType{[]byte{'-', '-', '-'}, []byte{'-', '-', '-'}, page.handleYamlMetaData}
|
return &frontmatterType{[]byte{'-', '-', '-'}, []byte{'-', '-', '-'}, page.handleYamlMetaData, false}
|
||||||
case '+':
|
case '+':
|
||||||
return &frontmatterType{[]byte{'+', '+', '+'}, []byte{'+', '+', '+'}, page.handleTomlMetaData}
|
return &frontmatterType{[]byte{'+', '+', '+'}, []byte{'+', '+', '+'}, page.handleTomlMetaData, false}
|
||||||
case '{':
|
case '{':
|
||||||
return &frontmatterType{[]byte{'{'}, []byte{'}'}, page.handleJsonMetaData}
|
return &frontmatterType{[]byte{'{'}, []byte{'}'}, page.handleJsonMetaData, true}
|
||||||
default:
|
default:
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func beginFrontMatter(data *bufio.Reader, f *frontmatterType) (bool, error) {
|
func beginFrontMatter(data *bufio.Reader, f *frontmatterType) (bool, error) {
|
||||||
peek := make([]byte, 3)
|
var err error
|
||||||
_, err := data.Read(peek)
|
var peek []byte
|
||||||
|
if f.includeMark {
|
||||||
|
peek, err = data.Peek(len(f.markstart))
|
||||||
|
} else {
|
||||||
|
peek = make([]byte, len(f.markstart))
|
||||||
|
_, err = data.Read(peek)
|
||||||
|
}
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
}
|
}
|
||||||
|
|
|
@ -39,6 +39,36 @@ title: Leading WS
|
||||||
Leading
|
Leading
|
||||||
`
|
`
|
||||||
|
|
||||||
|
var SIMPLE_PAGE_JSON = `
|
||||||
|
{
|
||||||
|
"title": "spf13-vim 3.0 release and new website",
|
||||||
|
"description": "spf13-vim is a cross platform distribution of vim plugins and resources for Vim.",
|
||||||
|
"tags": [ ".vimrc", "plugins", "spf13-vim", "vim" ],
|
||||||
|
"date": "2012-04-06",
|
||||||
|
"categories": [
|
||||||
|
"Development",
|
||||||
|
"VIM"
|
||||||
|
],
|
||||||
|
"slug": "spf13-vim-3-0-release-and-new-website"
|
||||||
|
}
|
||||||
|
|
||||||
|
Content of the file goes Here
|
||||||
|
`
|
||||||
|
|
||||||
|
var SIMPLE_PAGE_JSON_MULTIPLE = `
|
||||||
|
{
|
||||||
|
"title": "foobar",
|
||||||
|
"customData": { "foo": "bar" },
|
||||||
|
"date": "2012-08-06"
|
||||||
|
}
|
||||||
|
Some text
|
||||||
|
`
|
||||||
|
|
||||||
|
var SIMPLE_PAGE_JSON_COMPACT = `
|
||||||
|
{"title":"foobar","customData":{"foo":"bar"},"date":"2012-08-06"}
|
||||||
|
Text
|
||||||
|
`
|
||||||
|
|
||||||
func checkError(t *testing.T, err error, expected string) {
|
func checkError(t *testing.T, err error, expected string) {
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Fatalf("err is nil")
|
t.Fatalf("err is nil")
|
||||||
|
@ -101,6 +131,23 @@ func TestCreateNewPage(t *testing.T) {
|
||||||
checkPageLayout(t, p, "page/single.html")
|
checkPageLayout(t, p, "page/single.html")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestCreatePage(t *testing.T) {
|
||||||
|
var tests = []struct {
|
||||||
|
r io.Reader
|
||||||
|
}{
|
||||||
|
{strings.NewReader(SIMPLE_PAGE_JSON)},
|
||||||
|
{strings.NewReader(SIMPLE_PAGE_JSON_MULTIPLE)},
|
||||||
|
//{strings.NewReader(SIMPLE_PAGE_JSON_COMPACT)},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
_, err := ReadFrom(test.r, "page")
|
||||||
|
if err != nil {
|
||||||
|
t.Errorf("Unable to parse page: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestDegenerateInvalidFrontMatterShortDelim(t *testing.T) {
|
func TestDegenerateInvalidFrontMatterShortDelim(t *testing.T) {
|
||||||
var tests = []struct {
|
var tests = []struct {
|
||||||
r io.Reader
|
r io.Reader
|
||||||
|
@ -108,7 +155,7 @@ func TestDegenerateInvalidFrontMatterShortDelim(t *testing.T) {
|
||||||
}{
|
}{
|
||||||
{strings.NewReader(INVALID_FRONT_MATTER_SHORT_DELIM), "unable to match beginning front matter delimiter"},
|
{strings.NewReader(INVALID_FRONT_MATTER_SHORT_DELIM), "unable to match beginning front matter delimiter"},
|
||||||
{strings.NewReader(INVALID_FRONT_MATTER_SHORT_DELIM_ENDING), "unable to match ending front matter delimiter"},
|
{strings.NewReader(INVALID_FRONT_MATTER_SHORT_DELIM_ENDING), "unable to match ending front matter delimiter"},
|
||||||
{strings.NewReader(INVALID_FRONT_MATTER_MISSING), "unable to match beginning front matter delimiter"},
|
{strings.NewReader(INVALID_FRONT_MATTER_MISSING), "unable to detect front matter"},
|
||||||
}
|
}
|
||||||
for _, test := range tests {
|
for _, test := range tests {
|
||||||
_, err := ReadFrom(test.r, "invalid/front/matter/short/delim")
|
_, err := ReadFrom(test.r, "invalid/front/matter/short/delim")
|
||||||
|
|
Loading…
Reference in a new issue