From fac43df2dab74d0a487b19fecb12c739da283d8f Mon Sep 17 00:00:00 2001
From: Gio <sethg@ipi.org>
Date: Fri, 16 Apr 2021 19:38:24 -0500
Subject: [PATCH] Test cases for hidden articles and drafts

---
 pelican/tests/content/article_draft.md  |  5 +++++
 pelican/tests/content/article_hidden.md |  5 +++++
 pelican/tests/test_generators.py        | 14 ++++++++++++++
 3 files changed, 24 insertions(+)
 create mode 100644 pelican/tests/content/article_draft.md
 create mode 100644 pelican/tests/content/article_hidden.md

diff --git a/pelican/tests/content/article_draft.md b/pelican/tests/content/article_draft.md
new file mode 100644
index 00000000..d2235553
--- /dev/null
+++ b/pelican/tests/content/article_draft.md
@@ -0,0 +1,5 @@
+Title: Draft article
+Date: 2012-10-31
+Status: draft
+
+This is some content.
diff --git a/pelican/tests/content/article_hidden.md b/pelican/tests/content/article_hidden.md
new file mode 100644
index 00000000..d449f701
--- /dev/null
+++ b/pelican/tests/content/article_hidden.md
@@ -0,0 +1,5 @@
+Title: Hidden article
+Date: 2012-10-31
+Status: hidden
+
+This is some unlisted content.
diff --git a/pelican/tests/test_generators.py b/pelican/tests/test_generators.py
index 332d7e78..5e7f21c9 100644
--- a/pelican/tests/test_generators.py
+++ b/pelican/tests/test_generators.py
@@ -181,6 +181,8 @@ class TestArticlesGenerator(unittest.TestCase):
             path=CONTENT_DIR, theme=settings['THEME'], output_path=None)
         cls.generator.generate_context()
         cls.articles = cls.distill_articles(cls.generator.articles)
+        cls.drafts = cls.distill_articles(cls.generator.drafts)
+        cls.hidden_articles = cls.distill_articles(cls.generator.hidden_articles)
 
     def setUp(self):
         self.temp_cache = mkdtemp(prefix='pelican_cache.')
@@ -283,6 +285,18 @@ class TestArticlesGenerator(unittest.TestCase):
         ]
         self.assertEqual(sorted(articles_expected), sorted(self.articles))
 
+    def test_articles_draft(self):
+        draft_articles_expected = [
+            ['Draft article', 'draft', 'Default', 'article'],
+        ]
+        self.assertEqual(sorted(draft_articles_expected), sorted(self.drafts))
+
+    def test_articles_hidden(self):
+        hidden_articles_expected = [
+            ['Hidden article', 'hidden', 'Default', 'article'],
+        ]
+        self.assertEqual(sorted(hidden_articles_expected), sorted(self.hidden_articles))
+
     def test_generate_categories(self):
         # test for name
         # categories are grouped by slug; if two categories have the same slug