2017-03-06 05:20:30 -05:00
|
|
|
// Copyright 2017-present 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 output
|
|
|
|
|
|
|
|
import (
|
2018-01-13 11:21:42 -05:00
|
|
|
"fmt"
|
|
|
|
"reflect"
|
|
|
|
"strings"
|
2017-03-06 05:20:30 -05:00
|
|
|
"testing"
|
|
|
|
|
2017-06-13 12:42:45 -04:00
|
|
|
"github.com/gohugoio/hugo/media"
|
2017-03-07 11:26:22 -05:00
|
|
|
|
2017-03-06 05:20:30 -05:00
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestLayout(t *testing.T) {
|
|
|
|
|
2017-06-20 02:45:52 -04:00
|
|
|
noExtNoDelimMediaType := media.TextType
|
|
|
|
noExtNoDelimMediaType.Suffix = ""
|
|
|
|
noExtNoDelimMediaType.Delimiter = ""
|
|
|
|
|
|
|
|
noExtMediaType := media.TextType
|
|
|
|
noExtMediaType.Suffix = ""
|
|
|
|
|
|
|
|
var (
|
|
|
|
ampType = Format{
|
|
|
|
Name: "AMP",
|
|
|
|
MediaType: media.HTMLType,
|
|
|
|
BaseName: "index",
|
|
|
|
}
|
|
|
|
|
2018-01-13 11:21:42 -05:00
|
|
|
htmlFormat = HTMLFormat
|
|
|
|
|
2017-06-20 02:45:52 -04:00
|
|
|
noExtDelimFormat = Format{
|
|
|
|
Name: "NEM",
|
|
|
|
MediaType: noExtNoDelimMediaType,
|
|
|
|
BaseName: "_redirects",
|
|
|
|
}
|
|
|
|
noExt = Format{
|
|
|
|
Name: "NEX",
|
|
|
|
MediaType: noExtMediaType,
|
|
|
|
BaseName: "next",
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
2017-03-07 11:26:22 -05:00
|
|
|
for _, this := range []struct {
|
|
|
|
name string
|
2017-03-16 03:58:50 -04:00
|
|
|
d LayoutDescriptor
|
2017-03-06 13:54:46 -05:00
|
|
|
hasTheme bool
|
2017-03-06 07:18:33 -05:00
|
|
|
layoutOverride string
|
2017-03-16 03:32:14 -04:00
|
|
|
tp Format
|
2017-03-06 07:18:33 -05:00
|
|
|
expect []string
|
2018-01-13 11:21:42 -05:00
|
|
|
expectCount int
|
2017-03-06 05:20:30 -05:00
|
|
|
}{
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Home", LayoutDescriptor{Kind: "home"}, true, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"index.amp.html", "theme/index.amp.html", "home.amp.html", "theme/home.amp.html", "list.amp.html", "theme/list.amp.html", "index.html", "theme/index.html", "home.html", "theme/home.html", "list.html", "theme/list.html", "_default/index.amp.html"}, 24},
|
|
|
|
{"Home, HTML", LayoutDescriptor{Kind: "home"}, true, "", htmlFormat,
|
|
|
|
// We will eventually get to index.html. This looks stuttery, but makes the lookup logic easy to understand.
|
|
|
|
[]string{"index.html.html", "theme/index.html.html", "home.html.html"}, 24},
|
2017-07-02 04:46:28 -04:00
|
|
|
{"Home, french language", LayoutDescriptor{Kind: "home", Lang: "fr"}, true, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"index.fr.amp.html", "theme/index.fr.amp.html"},
|
|
|
|
48},
|
2017-06-20 02:45:52 -04:00
|
|
|
{"Home, no ext or delim", LayoutDescriptor{Kind: "home"}, true, "", noExtDelimFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"index.nem", "theme/index.nem", "home.nem", "theme/home.nem", "list.nem"}, 12},
|
2017-06-20 02:45:52 -04:00
|
|
|
{"Home, no ext", LayoutDescriptor{Kind: "home"}, true, "", noExt,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"index.nex", "theme/index.nex", "home.nex", "theme/home.nex", "list.nex"}, 12},
|
2017-06-20 02:45:52 -04:00
|
|
|
{"Page, no ext or delim", LayoutDescriptor{Kind: "page"}, true, "", noExtDelimFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"_default/single.nem", "theme/_default/single.nem"}, 2},
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Section", LayoutDescriptor{Kind: "section", Section: "sect1"}, false, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"sect1/sect1.amp.html", "sect1/section.amp.html", "sect1/list.amp.html", "sect1/sect1.html", "sect1/section.html", "sect1/list.html", "section/sect1.amp.html", "section/section.amp.html"}, 18},
|
|
|
|
{"Section with layout", LayoutDescriptor{Kind: "section", Section: "sect1", Layout: "mylayout"}, false, "", ampType,
|
|
|
|
[]string{"sect1/mylayout.amp.html", "sect1/sect1.amp.html", "sect1/section.amp.html", "sect1/list.amp.html", "sect1/mylayout.html", "sect1/sect1.html"}, 24},
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Taxonomy", LayoutDescriptor{Kind: "taxonomy", Section: "tag"}, false, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"taxonomy/tag.amp.html", "taxonomy/taxonomy.amp.html", "taxonomy/list.amp.html", "taxonomy/tag.html", "taxonomy/taxonomy.html"}, 18},
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Taxonomy term", LayoutDescriptor{Kind: "taxonomyTerm", Section: "categories"}, false, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"taxonomy/categories.terms.amp.html", "taxonomy/terms.amp.html", "taxonomy/list.amp.html", "taxonomy/categories.terms.html", "taxonomy/terms.html"}, 18},
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Page", LayoutDescriptor{Kind: "page"}, true, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"_default/single.amp.html", "theme/_default/single.amp.html", "_default/single.html", "theme/_default/single.html"}, 4},
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Page with layout", LayoutDescriptor{Kind: "page", Layout: "mylayout"}, false, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"_default/mylayout.amp.html", "_default/single.amp.html", "_default/mylayout.html", "_default/single.html"}, 4},
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Page with layout and type", LayoutDescriptor{Kind: "page", Layout: "mylayout", Type: "myttype"}, false, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"myttype/mylayout.amp.html", "myttype/single.amp.html", "myttype/mylayout.html"}, 8},
|
2017-03-16 03:58:50 -04:00
|
|
|
{"Page with layout and type with subtype", LayoutDescriptor{Kind: "page", Layout: "mylayout", Type: "myttype/mysubtype"}, false, "", ampType,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"myttype/mysubtype/mylayout.amp.html", "myttype/mysubtype/single.amp.html", "myttype/mysubtype/mylayout.html"}, 8},
|
2017-03-25 12:46:09 -04:00
|
|
|
// RSS
|
2017-03-25 14:36:50 -04:00
|
|
|
{"RSS Home with theme", LayoutDescriptor{Kind: "home"}, true, "", RSSFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"index.rss.xml", "theme/index.rss.xml", "home.rss.xml", "theme/home.rss.xml", "rss.xml"}, 29},
|
2017-03-25 14:36:50 -04:00
|
|
|
{"RSS Section", LayoutDescriptor{Kind: "section", Section: "sect1"}, false, "", RSSFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"sect1/sect1.rss.xml", "sect1/section.rss.xml", "sect1/rss.xml", "sect1/list.rss.xml", "sect1/sect1.xml", "sect1/section.xml"}, 22},
|
2017-03-25 14:36:50 -04:00
|
|
|
{"RSS Taxonomy", LayoutDescriptor{Kind: "taxonomy", Section: "tag"}, false, "", RSSFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"taxonomy/tag.rss.xml", "taxonomy/taxonomy.rss.xml", "taxonomy/rss.xml", "taxonomy/list.rss.xml", "taxonomy/tag.xml", "taxonomy/taxonomy.xml"}, 22},
|
2017-03-25 14:36:50 -04:00
|
|
|
{"RSS Taxonomy term", LayoutDescriptor{Kind: "taxonomyTerm", Section: "tag"}, false, "", RSSFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"taxonomy/tag.terms.rss.xml", "taxonomy/terms.rss.xml", "taxonomy/rss.xml", "taxonomy/list.rss.xml", "taxonomy/tag.terms.xml"}, 22},
|
2017-03-27 14:43:49 -04:00
|
|
|
{"Home plain text", LayoutDescriptor{Kind: "home"}, true, "", JSONFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"_text/index.json.json", "_text/theme/index.json.json", "_text/home.json.json", "_text/theme/home.json.json"}, 24},
|
2017-03-27 14:43:49 -04:00
|
|
|
{"Page plain text", LayoutDescriptor{Kind: "page"}, true, "", JSONFormat,
|
2018-01-13 11:21:42 -05:00
|
|
|
[]string{"_text/_default/single.json.json", "_text/theme/_default/single.json.json", "_text/_default/single.json", "_text/theme/_default/single.json"}, 4},
|
|
|
|
{"Reserved section, shortcodes", LayoutDescriptor{Kind: "section", Section: "shortcodes", Type: "shortcodes"}, true, "", ampType,
|
|
|
|
[]string{"section/shortcodes.amp.html", "theme/section/shortcodes.amp.html"}, 24},
|
|
|
|
{"Reserved section, partials", LayoutDescriptor{Kind: "section", Section: "partials", Type: "partials"}, true, "", ampType,
|
|
|
|
[]string{"section/partials.amp.html", "theme/section/partials.amp.html"}, 24},
|
2017-03-06 05:20:30 -05:00
|
|
|
} {
|
2017-03-07 11:26:22 -05:00
|
|
|
t.Run(this.name, func(t *testing.T) {
|
|
|
|
l := NewLayoutHandler(this.hasTheme)
|
|
|
|
|
2018-01-13 11:21:42 -05:00
|
|
|
layouts, err := l.For(this.d, this.tp)
|
2017-03-07 11:26:22 -05:00
|
|
|
|
2017-03-26 13:34:30 -04:00
|
|
|
require.NoError(t, err)
|
2017-03-07 11:26:22 -05:00
|
|
|
require.NotNil(t, layouts)
|
2018-01-13 11:21:42 -05:00
|
|
|
require.True(t, len(layouts) >= len(this.expect), fmt.Sprint(layouts))
|
2017-03-07 11:26:22 -05:00
|
|
|
// Not checking the complete list for now ...
|
2018-01-13 11:21:42 -05:00
|
|
|
got := layouts[:len(this.expect)]
|
|
|
|
if len(layouts) != this.expectCount || !reflect.DeepEqual(got, this.expect) {
|
|
|
|
formatted := strings.Replace(fmt.Sprintf("%v", layouts), "[", "\"", 1)
|
|
|
|
formatted = strings.Replace(formatted, "]", "\"", 1)
|
|
|
|
formatted = strings.Replace(formatted, " ", "\", \"", -1)
|
|
|
|
|
|
|
|
t.Fatalf("Got %d/%d:\n%v\nExpected:\n%v\nAll:\n%v\nFormatted:\n%s", len(layouts), this.expectCount, got, this.expect, layouts, formatted)
|
|
|
|
|
|
|
|
}
|
2017-03-06 13:54:46 -05:00
|
|
|
|
2017-03-07 11:26:22 -05:00
|
|
|
if !this.hasTheme {
|
|
|
|
for _, layout := range layouts {
|
|
|
|
require.NotContains(t, layout, "theme")
|
|
|
|
}
|
2017-03-06 13:54:46 -05:00
|
|
|
}
|
2017-03-07 11:26:22 -05:00
|
|
|
})
|
2017-03-06 05:20:30 -05:00
|
|
|
}
|
2017-03-18 11:46:10 -04:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
func BenchmarkLayout(b *testing.B) {
|
|
|
|
descriptor := LayoutDescriptor{Kind: "taxonomyTerm", Section: "categories"}
|
|
|
|
l := NewLayoutHandler(false)
|
|
|
|
|
|
|
|
for i := 0; i < b.N; i++ {
|
2018-01-13 11:21:42 -05:00
|
|
|
layouts, err := l.For(descriptor, HTMLFormat)
|
2017-03-26 13:34:30 -04:00
|
|
|
require.NoError(b, err)
|
2017-03-18 11:46:10 -04:00
|
|
|
require.NotEmpty(b, layouts)
|
|
|
|
}
|
2017-03-06 05:20:30 -05:00
|
|
|
}
|