hugolib, output: Add theme logic to LayoutHandler

1 files changed, 32 insertions, 14 deletions

diff --git a/output/layout_test.go b/output/layout_test.go
index 5b95e01d8..333216e17 100644
--- a/output/layout_test.go
+++ b/output/layout_test.go
@@ -14,6 +14,7 @@
 package output
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -43,28 +44,45 @@ func (l testLayoutIdentifier) PageSection() string {
 }
 
 func TestLayout(t *testing.T) {
-	l := &LayoutHandler{}
 
-	for _, this := range []struct {
+	for i, this := range []struct {
 		li             testLayoutIdentifier
+		hasTheme       bool
 		layoutOverride string
 		tp             Type
 		expect         []string
 	}{
-		{testLayoutIdentifier{"home", "", "", ""}, "", HTMLType, []string{"index.html", "_default/list.html", "theme/index.html", "theme/_default/list.html"}},
-		{testLayoutIdentifier{"section", "sect1", "", ""}, "", HTMLType, []string{"section/sect1.html", "sect1/list.html"}},
-		{testLayoutIdentifier{"taxonomy", "tag", "", ""}, "", HTMLType, []string{"taxonomy/tag.html", "indexes/tag.html"}},
-		{testLayoutIdentifier{"taxonomyTerm", "categories", "", ""}, "", HTMLType, []string{"taxonomy/categories.terms.html", "_default/terms.html"}},
-		{testLayoutIdentifier{"page", "", "", ""}, "", HTMLType, []string{"_default/single.html", "theme/_default/single.html"}},
-		{testLayoutIdentifier{"page", "", "mylayout", ""}, "", HTMLType, []string{"_default/mylayout.html"}},
-		{testLayoutIdentifier{"page", "", "mylayout", "myttype"}, "", HTMLType, []string{"myttype/mylayout.html", "_default/mylayout.html"}},
-		{testLayoutIdentifier{"page", "", "mylayout", "myttype/mysubtype"}, "", HTMLType, []string{"myttype/mysubtype/mylayout.html", "myttype/mylayout.html", "_default/mylayout.html"}},
-		{testLayoutIdentifier{"page", "", "mylayout", "myttype"}, "myotherlayout", HTMLType, []string{"myttype/myotherlayout.html", "_default/myotherlayout.html"}},
+		{testLayoutIdentifier{"home", "", "", ""}, true, "", HTMLType,
+			[]string{"index.html", "_default/list.html", "theme/index.html", "theme/_default/list.html"}},
+		{testLayoutIdentifier{"section", "sect1", "", ""}, false, "", HTMLType,
+			[]string{"section/sect1.html", "sect1/list.html"}},
+		{testLayoutIdentifier{"taxonomy", "tag", "", ""}, false, "", HTMLType,
+			[]string{"taxonomy/tag.html", "indexes/tag.html"}},
+		{testLayoutIdentifier{"taxonomyTerm", "categories", "", ""}, false, "", HTMLType,
+			[]string{"taxonomy/categories.terms.html", "_default/terms.html"}},
+		{testLayoutIdentifier{"page", "", "", ""}, true, "", HTMLType,
+			[]string{"_default/single.html", "theme/_default/single.html"}},
+		{testLayoutIdentifier{"page", "", "mylayout", ""}, false, "", HTMLType,
+			[]string{"_default/mylayout.html"}},
+		{testLayoutIdentifier{"page", "", "mylayout", "myttype"}, false, "", HTMLType,
+			[]string{"myttype/mylayout.html", "_default/mylayout.html"}},
+		{testLayoutIdentifier{"page", "", "mylayout", "myttype/mysubtype"}, false, "", HTMLType,
+			[]string{"myttype/mysubtype/mylayout.html", "myttype/mylayout.html", "_default/mylayout.html"}},
+		{testLayoutIdentifier{"page", "", "mylayout", "myttype"}, false, "myotherlayout", HTMLType,
+			[]string{"myttype/myotherlayout.html", "_default/myotherlayout.html"}},
 	} {
+		l := NewLayoutHandler(this.hasTheme)
+		logMsg := fmt.Sprintf("Test %d", i)
 		layouts := l.For(this.li, this.layoutOverride, this.tp)
-		require.NotNil(t, layouts)
-		require.True(t, len(layouts) >= len(this.expect))
+		require.NotNil(t, layouts, logMsg)
+		require.True(t, len(layouts) >= len(this.expect), logMsg)
 		// Not checking the complete list for now ...
-		require.Equal(t, this.expect, layouts[:len(this.expect)])
+		require.Equal(t, this.expect, layouts[:len(this.expect)], logMsg)
+
+		if !this.hasTheme {
+			for _, layout := range layouts {
+				require.NotContains(t, layout, "theme", logMsg)
+			}
+		}
 	}
 }