generating output markdown files works

[utils.git] / src / budget_sync / test / test_write_budget_markdown.py

diff --git a/src/budget_sync/test/test_write_budget_markdown.py b/src/budget_sync/test/test_write_budget_markdown.py
index 1691edc6f6d2bc23da72c6659ce65b38d0833ff7..b035a39c64b10da687ec67e5838596bada06f953 100644 (file)
--- a/src/budget_sync/test/test_write_budget_markdown.py
+++ b/src/budget_sync/test/test_write_budget_markdown.py
@@ -4,10 +4,22 @@ from budget_sync.test.mock_bug import MockBug
 from budget_sync.test.mock_path import MockPath, DIR
 from budget_sync.test.test_mock_path import make_filesystem_and_report_if_error
 from budget_sync.budget_graph import BudgetGraph
-from budget_sync.write_budget_markdown import write_budget_markdown
+from budget_sync.write_budget_markdown import (
+    write_budget_markdown, DisplayStatus, markdown_escape)
+from budget_sync.util import BugStatus
 
 
 class TestWriteBudgetMarkdown(unittest.TestCase):
+    maxDiff = None
+
+    def test_display_status(self):
+        for status in BugStatus:
+            DisplayStatus.from_status(status)
+
+    def test_markdown_escape(self):
+        self.assertEqual(markdown_escape("abc * def_k < &k"),
+                         r"abc \* def\_k &lt; &amp;k")
+
     def test(self):
         config = Config.from_str(
             """
@@ -37,10 +49,13 @@ class TestWriteBudgetMarkdown(unittest.TestCase):
                 "/": DIR,
                 "/output_dir": DIR,
                 '/output_dir/person1.mdwn': b'<!-- autogenerated by '
-                b'budget-sync -->\n# person1\n\n# Status Tracking\n',
+                b'budget-sync -->\n# person1\n\n# Status Tracking\n## Not yet '
+                b'started\n* [Bug #1](https://bugzilla.example.com/show_bug.c'
+                b'gi?id=1): \n',
                 '/output_dir/person2.mdwn': b'<!-- autogenerated by '
                 b'budget-sync -->\n# person2\n\n# Status Tracking\n',
             }, filesystem.files)
+    # TODO: add more test cases
 
 
 if __name__ == "__main__":