From d2bc64bee3190d5f1850bec45bf7f51375f13c79 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bj=C3=B8rn=20Erik=20Pedersen?=
 <bjorn.erik.pedersen@gmail.com>
Date: Tue, 11 Oct 2016 18:12:28 +0200
Subject: [PATCH] Add draft test

See #2549
---
 hugolib/page_test.go | 34 ++++++++++++++++++++++++++++++++--
 1 file changed, 32 insertions(+), 2 deletions(-)

diff --git a/hugolib/page_test.go b/hugolib/page_test.go
index 7cdc55898..a95670b28 100644
--- a/hugolib/page_test.go
+++ b/hugolib/page_test.go
@@ -1128,7 +1128,7 @@ func TestDraftAndPublishedFrontMatterError(t *testing.T) {
 	}
 }
 
-var pageWithPublishedFalse = `---
+var pagesWithPublishedFalse = `---
 title: okay
 published: false
 ---
@@ -1142,7 +1142,7 @@ some content
 `
 
 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)
 	}
@@ -1158,6 +1158,36 @@ func TestPublishedFrontMatter(t *testing.T) {
 	}
 }
 
+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)
+			}
+		}
+	}
+}
+
 func TestPageSimpleMethods(t *testing.T) {
 	for i, this := range []struct {
 		assertFunc func(p *Page) bool