Rewrite TestRSSOutput to use Afero

When we have a virtual filesystem, we might as well test as close to the iron as possible.

diff --git a/hugolib/rss_test.go b/hugolib/rss_test.go
index 007d6a17ce7a1cc8020e6873f42deb9ec01ca04e..f0ab1c2c365ff691118ff28bd10e9c2d5ccaf341 100644 (file)
--- a/hugolib/rss_test.go
+++ b/hugolib/rss_test.go
@@ -14,13 +14,9 @@
 package hugolib
 
 import (
-       "bytes"
        "path/filepath"
        "testing"
 
-       "github.com/spf13/hugo/helpers"
-       "github.com/spf13/hugo/hugofs"
-       "github.com/spf13/hugo/source"
        "github.com/spf13/viper"
 )
 
@@ -52,24 +48,16 @@ func TestRSSOutput(t *testing.T) {
        viper.Set("baseurl", "http://auth/bub/")
        viper.Set("RSSUri", rssURI)
 
-       hugofs.InitMemFs()
-       s := &Site{
-               Source:   &source.InMemorySource{ByteSource: weightedSources},
-               Language: helpers.NewDefaultLanguage(),
+       for _, s := range weightedSources {
+               writeSource(t, filepath.Join("content", s.Name), string(s.Content))
        }
 
-       if err := buildAndRenderSite(s, "rss.xml", rssTemplate); err != nil {
+       writeSource(t, filepath.Join("layouts", "rss.xml"), rssTemplate)
+
+       if err := buildAndRenderSite(newSiteDefaultLang()); err != nil {
                t.Fatalf("Failed to build site: %s", err)
        }
 
-       file, err := hugofs.Destination().Open(filepath.Join("public", rssURI))
-
-       if err != nil {
-               t.Fatalf("Unable to locate: %s", rssURI)
-       }
+       assertFileContent(t, filepath.Join("public", rssURI), true, "<?xml", "rss version")
 
-       rss := helpers.ReaderToBytes(file)
-       if !bytes.HasPrefix(rss, []byte("<?xml")) {
-               t.Errorf("rss feed should start with <?xml. %s", rss)
-       }
 }