package hugolib import ( "bytes" "testing" "github.com/spf13/afero" "github.com/spf13/hugo/helpers" "github.com/spf13/hugo/hugofs" "github.com/spf13/hugo/source" "github.com/spf13/viper" ) const SITEMAP_TEMPLATE = ` {{ range .Data.Pages }} {{ .Permalink }} {{ safeHTML ( .Date.Format "2006-01-02T15:04:05-07:00" ) }}{{ with .Sitemap.ChangeFreq }} {{ . }}{{ end }}{{ if ge .Sitemap.Priority 0.0 }} {{ .Sitemap.Priority }}{{ end }} {{ end }} ` func TestSitemapOutput(t *testing.T) { hugofs.DestinationFS = new(afero.MemMapFs) viper.Set("baseurl", "http://auth/bub/") s := &Site{ Source: &source.InMemorySource{ByteSource: WEIGHTED_SOURCES}, } s.initializeSiteInfo() s.prepTemplates() s.addTemplate("sitemap.xml", SITEMAP_TEMPLATE) 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.RenderHomePage(); err != nil { t.Fatalf("Unable to RenderHomePage: %s", err) } if err := s.RenderSitemap(); err != nil { t.Fatalf("Unable to RenderSitemap: %s", err) } sitemapFile, err := hugofs.DestinationFS.Open("sitemap.xml") if err != nil { t.Fatalf("Unable to locate: sitemap.xml") } sitemap := helpers.ReaderToBytes(sitemapFile) if !bytes.HasPrefix(sitemap, []byte("