shithub: hugo

Download patch

ref: d2bc64bee3190d5f1850bec45bf7f51375f13c79
parent: 5355ddf621de9b95b1fd65b4942f0cb77ddc9b19
author: Bjørn Erik Pedersen <[email protected]>
date: Tue Oct 11 14:12:28 EDT 2016

Add draft test

See #2549

--- a/hugolib/page_test.go
+++ b/hugolib/page_test.go
@@ -1128,7 +1128,7 @@
 	}
 }
 
-var pageWithPublishedFalse = `---
+var pagesWithPublishedFalse = `---
 title: okay
 published: false
 ---
@@ -1142,7 +1142,7 @@
 `
 
 func TestPublishedFrontMatter(t *testing.T) {
-	p, err := NewPageFrom(strings.NewReader(pageWithPublishedFalse), "content/post/broken.md")
+	p, err := NewPageFrom(strings.NewReader(pagesWithPublishedFalse), "content/post/broken.md")
 	if err != nil {
 		t.Fatalf("err during parse: %s", err)
 	}
@@ -1155,6 +1155,36 @@
 	}
 	if p.Draft {
 		t.Errorf("expected false, got %t", p.Draft)
+	}
+}
+
+var pagesDraftTemplate = []string{`---
+title: "okay"
+draft: %t
+---
+some content
+`,
+	`+++
+title = "okay"
+draft = %t
++++
+
+some content
+`,
+}
+
+func TestDraft(t *testing.T) {
+	for _, draft := range []bool{true, false} {
+		for i, templ := range pagesDraftTemplate {
+			pageContent := fmt.Sprintf(templ, draft)
+			p, err := NewPageFrom(strings.NewReader(pageContent), "content/post/broken.md")
+			if err != nil {
+				t.Fatalf("err during parse: %s", err)
+			}
+			if p.Draft != draft {
+				t.Errorf("[%d] expected %t, got %t", i, draft, p.Draft)
+			}
+		}
 	}
 }