From 3ce904669a0d24debc5849c9f1a2f3f5f8fc9f11 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bj=C3=B8rn=20Erik=20Pedersen?=
 <bjorn.erik.pedersen@gmail.com>
Date: Fri, 14 Oct 2016 16:41:55 +0200
Subject: [PATCH] Rewrite TestRSSOutput to use Afero

When we have a virtual filesystem, we might as well test as close to the iron as possible.
---
 hugolib/rss_test.go | 24 ++++++------------------
 1 file changed, 6 insertions(+), 18 deletions(-)

diff --git a/hugolib/rss_test.go b/hugolib/rss_test.go
index 007d6a17c..f0ab1c2c3 100644
--- 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))
+	assertFileContent(t, filepath.Join("public", rssURI), true, "<?xml", "rss version")
 
-	if err != nil {
-		t.Fatalf("Unable to locate: %s", rssURI)
-	}
-
-	rss := helpers.ReaderToBytes(file)
-	if !bytes.HasPrefix(rss, []byte("<?xml")) {
-		t.Errorf("rss feed should start with <?xml. %s", rss)
-	}
 }