shithub: hugo

Info  •  Files  •  Log  •  Branches

refactor and clean up site tests

--- /dev/null

+++ b/hugolib/helpers_for_test.go

@@ -1,0 +1,80 @@

+package hugolib

+

+import (

+	"testing"

+

+	"github.com/spf13/afero"

+	"github.com/spf13/hugo/hugofs"

+	"github.com/spf13/hugo/source"

+	"github.com/spf13/hugo/tpl"

+	"github.com/spf13/viper"

+)

+

+func siteFromByteSources(bs []source.ByteSource) *Site {

+

+	viper.SetDefault("baseurl", "http://auth/bub")

+	hugofs.DestinationFS = new(afero.MemMapFs)

+

+	s := &Site{

+		Source: &source.InMemorySource{ByteSource: bs},

+	}

+	s.initializeSiteInfo()

+	return s

+}

+

+func buildSiteFromByteSources(bs []source.ByteSource, t *testing.T) *Site {

+	s := siteFromByteSources(bs)

+

+	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)

+	}

+

+	return s

+}

+

+func setHugoDefaultTaxonomies() {

+	taxonomies := make(map[string]string)

+	taxonomies["tag"] = "tags"

+	taxonomies["category"] = "categories"

+

+	viper.Set("taxonomies", taxonomies)

+}

+

+func createAndRenderPages(t *testing.T, s *Site) {

+	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)

+	}

+}

+

+func testRenderPages(t *testing.T, s *Site) {

+	if err := s.RenderPages(); err != nil {

+		t.Fatalf("Unable to render pages. %s", err)

+	}

+}

+

+func templatePrep(s *Site) {

+	s.Tmpl = tpl.New()

+	s.Tmpl.LoadTemplates(s.absLayoutDir())

+	if s.hasTheme() {

+		s.Tmpl.LoadTemplatesWithPrefix(s.absThemeDir()+"/layouts", "theme")

+	}

+}

+

+func pageMust(p *Page, err error) *Page {

+	if err != nil {

+		panic(err)

+	}

+	return p

+}

--- a/hugolib/site_test.go

+++ b/hugolib/site_test.go

@@ -15,7 +15,6 @@

 	"github.com/spf13/hugo/hugofs"

 	"github.com/spf13/hugo/source"

 	"github.com/spf13/hugo/target"

-	"github.com/spf13/hugo/tpl"

 	"github.com/spf13/viper"

 	"github.com/stretchr/testify/assert"

 )

@@ -50,35 +49,6 @@

 `

 )

-func createAndRenderPages(t *testing.T, s *Site) {

-	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)

-	}

-}

-

-func templatePrep(s *Site) {

-	s.Tmpl = tpl.New()

-	s.Tmpl.LoadTemplates(s.absLayoutDir())

-	if s.hasTheme() {

-		s.Tmpl.LoadTemplatesWithPrefix(s.absThemeDir()+"/layouts", "theme")

-	}

-}

-

-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()

@@ -227,23 +197,8 @@

 		{filepath.FromSlash("sect/doc4.md"), []byte("---\ntitle: doc4\ndraft: false\npublishdate: \"2012-05-29\"\n---\n# doc4\n*some content*")},

 	}

-	siteSetup := func() *Site {

-		s := &Site{

-			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()

+	s := buildSiteFromByteSources(sources, t)

 	if len(s.Pages) != 1 {

 		t.Fatal("Draft or Future dated content published unexpectedly")

 	}

@@ -250,7 +205,8 @@

 	// only publishDate in the past should be rendered

 	viper.Set("BuildDrafts", true)

-	s = siteSetup()

+

+	s = buildSiteFromByteSources(sources, t)

 	if len(s.Pages) != 2 {

 		t.Fatal("Future Dated Posts published unexpectedly")

 	}

@@ -258,7 +214,8 @@

 	//  drafts should not be rendered, but all dates should

 	viper.Set("BuildDrafts", false)

 	viper.Set("BuildFuture", true)

-	s = siteSetup()

+

+	s = buildSiteFromByteSources(sources, t)

 	if len(s.Pages) != 2 {

 		t.Fatal("Draft posts published unexpectedly")

 	}

@@ -266,14 +223,11 @@

 	// all 4 should be included

 	viper.Set("BuildDrafts", true)

 	viper.Set("BuildFuture", true)

-	s = siteSetup()

+

+	s = buildSiteFromByteSources(sources, t)

 	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)

 }

 // Issue #957

@@ -385,12 +339,8 @@

 		{filepath.FromSlash("sect/doc1.html"), []byte("---\nmarkup: markdown\n---\n# title\nsome *content*")},

 	}

-	s := &Site{

-		Source:  &source.InMemorySource{ByteSource: sources},

-		Targets: targetList{Page: &target.PagePub{UglyURLs: uglyURLs}},

-	}

+	s := buildSiteFromByteSources(sources, t)

-	s.initializeSiteInfo()

 	templatePrep(s)

 	must(s.addTemplate("index.html", "Home Sweet Home"))

@@ -401,7 +351,8 @@

 	must(s.addTemplate("rss.xml", "<root>RSS</root>"))

 	must(s.addTemplate("sitemap.xml", "<root>SITEMAP</root>"))

-	createAndRenderPages(t, s)

+	testRenderPages(t, s)

+

 	s.RenderHomePage()

 	s.RenderSitemap()

@@ -527,7 +478,6 @@

 	viper.Reset()

 	defer viper.Reset()

-	hugofs.DestinationFS = new(afero.MemMapFs)

 	sources := []source.ByteSource{

 		{filepath.FromSlash("sect/doc1.html"), []byte("---\nmarkup: markdown\n---\n# title\nsome *content*")},

 		{filepath.FromSlash("sect/doc2.html"), []byte("<!doctype html><html><body>more content</body></html>")},

@@ -542,13 +492,10 @@

 	viper.Set("DefaultExtension", "html")

 	viper.Set("verbose", true)

 	viper.Set("CanonifyURLs", true)

-	viper.Set("baseurl", "http://auth/bub")

-	s := &Site{

-		Source:  &source.InMemorySource{ByteSource: sources},

-		Targets: targetList{Page: &target.PagePub{UglyURLs: true}},

-	}

+	viper.Set("UglyURLs", true)

-	s.initializeSiteInfo()

+	s := buildSiteFromByteSources(sources, t)

+

 	templatePrep(s)

 	must(s.addTemplate("_default/single.html", "{{.Content}}"))

@@ -555,7 +502,7 @@

 	must(s.addTemplate("head", "<head><script src=\"script.js\"></script></head>"))

 	must(s.addTemplate("head_abs", "<head><script src=\"/script.js\"></script></head>"))

-	createAndRenderPages(t, s)

+	testRenderPages(t, s)

 	tests := []struct {

 		doc      string

@@ -596,26 +543,18 @@

 		{filepath.FromSlash("sect/doc1.html"), []byte("<!doctype html><html><head></head><body><a href=\"#frag1\">link</a></body></html>")},

 		{filepath.FromSlash("content/blue/doc2.html"), []byte("---\nf: t\n---\n<!doctype html><html><body>more content</body></html>")},

 	}

+

+	viper.Set("UglyURLs", true)

+

 	for _, canonify := range []bool{true, false} {

+		s := buildSiteFromByteSources(sources, t)

 		viper.Set("CanonifyURLs", canonify)

-		viper.Set("BaseURL", "http://auth/bub")

-		s := &Site{

-			Source:  &source.InMemorySource{ByteSource: sources},

-			Targets: targetList{Page: &target.PagePub{UglyURLs: true}},

-		}

+

 		t.Logf("Rendering with BaseURL %q and CanonifyURLs set %v", viper.GetString("baseURL"), canonify)

-		s.initializeSiteInfo()

+

 		templatePrep(s)

 		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)

 		}

@@ -697,22 +636,8 @@

 	viper.Reset()

 	defer viper.Reset()

-	hugofs.DestinationFS = new(afero.MemMapFs)

+	s := buildSiteFromByteSources(WEIGHTED_SOURCES, t)

-	viper.Set("baseurl", "http://auth/bub")

-	s := &Site{

-		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)

 	}

@@ -771,22 +696,8 @@

 		}

 	}()

-	hugofs.DestinationFS = new(afero.MemMapFs)

+	s := buildSiteFromByteSources(GROUPED_SOURCES, t)

-	viper.Set("baseurl", "http://auth/bub")

-	s := &Site{

-		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)

@@ -951,31 +862,15 @@

 	viper.Reset()

 	defer viper.Reset()

-	hugofs.DestinationFS = new(afero.MemMapFs)

 	sources := []source.ByteSource{

 		{filepath.FromSlash("sect/doc1.md"), PAGE_WITH_WEIGHTED_TAXONOMIES_1},

 		{filepath.FromSlash("sect/doc2.md"), PAGE_WITH_WEIGHTED_TAXONOMIES_2},

 		{filepath.FromSlash("sect/doc3.md"), PAGE_WITH_WEIGHTED_TAXONOMIES_3},

 	}

-	taxonomies := make(map[string]string)

-	taxonomies["tag"] = "tags"

-	taxonomies["category"] = "categories"

+	setHugoDefaultTaxonomies()

-	viper.Set("baseurl", "http://auth/bub")

-	viper.Set("taxonomies", taxonomies)

-	s := &Site{

-		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)

-	}

+	s := buildSiteFromByteSources(sources, t)

 	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)