summaryrefslogtreecommitdiffstats
path: root/hugolib/site_url_test.go
diff options
context:
space:
mode:
authorNoah Campbell <noahcampbell@gmail.com>2013-08-30 17:18:05 -0700
committerNoah Campbell <noahcampbell@gmail.com>2013-08-30 20:45:42 -0700
commitb14b61af37f0428545af0d191a27170434e4aad2 (patch)
treee85c3091019449c87ce4059675a892f554876a0d /hugolib/site_url_test.go
parentbc3c2290027359206f3569bcd51447443f7b40bb (diff)
Externalize the writing of content to a target
Introducing the target module in hugo. This provides the simple interface for writing content given a label (filename) and a io.Reader containing the content to be written. If site.Target is not set, it defaults back to the original behavior of writing to file system. In hugolib/site_url_test.go I have an InMemoryTarget for testing purposes and use it to see if the final output of a render matches.
Diffstat (limited to 'hugolib/site_url_test.go')
-rw-r--r--hugolib/site_url_test.go69
1 files changed, 69 insertions, 0 deletions
diff --git a/hugolib/site_url_test.go b/hugolib/site_url_test.go
new file mode 100644
index 000000000..6384f797b
--- /dev/null
+++ b/hugolib/site_url_test.go
@@ -0,0 +1,69 @@
+package hugolib
+
+import (
+ "bytes"
+ "io"
+ "path/filepath"
+ "strings"
+ "testing"
+)
+
+const SLUG_DOC_1 = "---\ntitle: slug doc 1\nslug: slug-doc-1\n---\nslug doc 1 content"
+const SLUG_DOC_2 = "---\ntitle: slug doc 2\nslug: slug-doc-2\n---\nslug doc 2 content"
+
+const INDEX_TEMPLATE = "{{ range .Data.Pages }}.{{ end }}"
+
+func must(err error) {
+ if err != nil {
+ panic(err)
+ }
+}
+
+func mustReturn(ret *Page, err error) *Page {
+ if err != nil {
+ panic(err)
+ }
+ return ret
+}
+
+type InMemoryTarget struct {
+ files map[string][]byte
+}
+
+func (t *InMemoryTarget) Publish(label string, reader io.Reader) (err error) {
+ if t.files == nil {
+ t.files = make(map[string][]byte)
+ }
+ bytes := new(bytes.Buffer)
+ bytes.ReadFrom(reader)
+ t.files[label] = bytes.Bytes()
+ return
+}
+
+func TestPageCount(t *testing.T) {
+ target := new(InMemoryTarget)
+ s := &Site{Target: target}
+ s.prepTemplates()
+ must(s.addTemplate("indexes/blue.html", INDEX_TEMPLATE))
+ s.Files = append(s.Files, "blue/doc1.md")
+ s.Files = append(s.Files, "blue/doc2.md")
+ s.Pages = append(s.Pages, mustReturn(ReadFrom(strings.NewReader(SLUG_DOC_1), filepath.FromSlash("content/blue/doc1.md"))))
+ s.Pages = append(s.Pages, mustReturn(ReadFrom(strings.NewReader(SLUG_DOC_2), filepath.FromSlash("content/blue/doc2.md"))))
+
+ if err := s.BuildSiteMeta(); err != nil {
+ t.Errorf("Unable to build site metadata: %s", err)
+ }
+
+ if err := s.RenderLists(); err != nil {
+ t.Errorf("Unable to render site lists: %s", err)
+ }
+
+ blueIndex := target.files["blue/index.html"]
+ if blueIndex == nil {
+ t.Errorf("No indexed rendered. %v", target.files)
+ }
+
+ if len(blueIndex) != 2 {
+ t.Errorf("Number of pages does not equal 2, got %d. %q", len(blueIndex), blueIndex)
+ }
+}