From 9a3a619791d34847e07c7dc7b952863d2a6d30c7 Mon Sep 17 00:00:00 2001
From: Tomas Babej <tomasbabej@gmail.com>
Date: Fri, 27 Mar 2015 13:36:44 +0100
Subject: [PATCH] tests: Add simple test for monthly ghistory

---
 tests/test_splits.py | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/tests/test_splits.py b/tests/test_splits.py
index 35a020c..55ecc2f 100644
--- a/tests/test_splits.py
+++ b/tests/test_splits.py
@@ -41,7 +41,7 @@ class TestCalendarSimple(IntegrationTest):
             assert any(day in line for line in output)
 
 
-class TestGhistorySimple(IntegrationTest):
+class TestGhistoryAnnualSimple(IntegrationTest):
 
     tasks = [
         dict(description="test task"),
@@ -65,3 +65,35 @@ class TestGhistorySimple(IntegrationTest):
 
         current_year = local_zone.localize(datetime.now()).year
         assert str(current_year) in '\n'.join(output)
+
+
+class TestGhistoryMonthlySimple(IntegrationTest):
+
+    tasks = [
+        dict(description="test task"),
+        dict(description="completed task 1", status="completed", end="now"),
+        dict(description="completed task 2", status="completed", end="now"),
+        dict(description="deleted task", status="deleted"),
+    ]
+
+    def execute(self):
+        self.command("TaskWikiGhistoryMonthly")
+        assert self.command(":py print vim.current.buffer", silent=False).startswith("<buffer ghistory.monthly")
+        output = self.read_buffer()
+
+        header_words = ("Year", "Month", "Number", "Added", "Completed", "Deleted")
+        for word in header_words:
+            assert word in output[0]
+
+        legend_words = ("Legend", "Added", "Completed", "Deleted")
+        for word in legend_words:
+            assert re.search(word, output[-1], re.IGNORECASE)
+
+        current_year = local_zone.localize(datetime.now()).year
+        current_month_number = local_zone.localize(datetime.now()).month
+        months = ["January", "February", "March", "April",
+                  "May", "June", "July", "August",
+                  "September", "October", "November", "December"]
+
+        assert str(current_year) in '\n'.join(output)
+        assert str(months[current_month_number - 1]) in '\n'.join(output)