From f96c810ffbcff410dbd3a58a53bee2fd9f296da2 Mon Sep 17 00:00:00 2001
From: Tomas Babej <tomasbabej@gmail.com>
Date: Sun, 27 Dec 2015 12:41:36 +0100
Subject: [PATCH] tests: Expand test coverage

---
 tests/test_choose.py   | 40 +++++++++++++++++++-
 tests/test_selected.py | 85 ++++++++++++++++++++++++++++++++++++++++++
 tests/test_splits.py   | 24 ++++++++++++
 tests/test_viewport.py | 38 +++++++++++++++++++
 4 files changed, 186 insertions(+), 1 deletion(-)

diff --git a/tests/test_choose.py b/tests/test_choose.py
index bbaea1d..d98837d 100644
--- a/tests/test_choose.py
+++ b/tests/test_choose.py
@@ -179,5 +179,43 @@ class TestChooseTagCancelled(IntegrationTest):
             task.refresh()
 
         assert self.tasks[0]['tags'] == set(["home"])
-        assert self.tasks[1]['project'] == None
+        assert self.tasks[1]['tags'] == set()
+
+
+class TestChooseTagNoSelected(IntegrationTest):
+
+    viminput = """
+    * [ ] test task 1  #{uuid}
+    * [ ] test task 2  #{uuid}
+    """
+
+    vimoutput = """
+    Tag  Count
+    ---- -----
+    home     1
+    """
+
+    tasks = [
+        dict(description="test task 1", tags=["home"]),
+        dict(description="test task 2"),
+    ]
+
+    def execute(self):
+        self.client.normal('1gg')
+        sleep(0.5)
+
+        self.command("TaskWikiChooseTag")
+        sleep(0.5)
+
+        # No tak on the 5th row
+        self.client.normal('5gg')
+        sleep(0.5)
+        self.client.feedkeys("\\<CR>")
+        sleep(0.5)
+
+        for task in self.tasks:
+            task.refresh()
+
+        assert self.tasks[0]['tags'] == set(["home"])
+        assert self.tasks[1]['tags'] == set()
 
diff --git a/tests/test_selected.py b/tests/test_selected.py
index b93adb6..4d43cf8 100644
--- a/tests/test_selected.py
+++ b/tests/test_selected.py
@@ -259,6 +259,35 @@ class TestInfoAction(IntegrationTest):
         assert re.search(data2, output, re.MULTILINE)
 
 
+class TestInfoActionTriggeredByEnter(IntegrationTest):
+
+    viminput = """
+    * [ ] test task 1  #{uuid}
+    * [ ] test task 2  #{uuid}
+    """
+
+    tasks = [
+        dict(description="test task 1"),
+        dict(description="test task 2"),
+    ]
+
+    def execute(self):
+        self.client.type('1gg')
+        self.client.feedkeys("\\<CR>")
+        sleep(0.5)
+
+        assert self.command(":py print vim.current.buffer", silent=False).startswith("<buffer info")
+        output = '\n'.join(self.read_buffer())
+
+        header = r'\s*'.join(['Name', 'Value'])
+        data = r'\s*'.join(['Description', 'test task 1'])
+        data2 = r'\s*'.join(['Status', 'Pending'])
+
+        assert re.search(header, output, re.MULTILINE)
+        assert re.search(data, output, re.MULTILINE)
+        assert re.search(data2, output, re.MULTILINE)
+
+
 class TestInfoActionMoved(IntegrationTest):
 
     viminput = """
@@ -832,6 +861,35 @@ class TestDoneAction(IntegrationTest):
 
         assert self.tasks[1]['end'] == None
 
+class TestDoneNoSelected(IntegrationTest):
+
+    viminput = """
+    * [ ] test task 1  #{uuid}
+
+    * [ ] test task 2  #{uuid}
+    """
+
+    vimoutput = """
+    * [ ] test task 1  #{uuid}
+
+    * [ ] test task 2  #{uuid}
+    """
+
+    tasks = [
+        dict(description="test task 1"),
+        dict(description="test task 2"),
+    ]
+
+    def execute(self):
+        self.client.type('2gg')
+        self.command(
+            "TaskWikiDone",
+            regex="No tasks selected.",
+            lines=1)
+
+        assert self.tasks[0]['status'] == "pending"
+        assert self.tasks[1]['status'] == "pending"
+
 
 class TestDoneActionMoved(IntegrationTest):
 
@@ -911,3 +969,30 @@ class TestDoneActionRange(IntegrationTest):
 
         assert (now - self.tasks[1]['end']).total_seconds() < 5
         assert (self.tasks[1]['end'] - now).total_seconds() < 5
+
+
+class TestSortManually(IntegrationTest):
+
+    viminput = """
+    * [ ] test task 1  #{uuid}
+    * [ ] test task 2  #{uuid}
+    """
+
+    vimoutput = """
+    * [ ] test task 2  #{uuid}
+    * [ ] test task 1  #{uuid}
+    """
+
+    tasks = [
+        dict(description="test task 1"),
+        dict(description="test task 2"),
+    ]
+
+    def execute(self):
+        self.client.normal('1gg')
+        sleep(0.5)
+        self.client.normal('VG')
+        sleep(0.5)
+        self.client.feedkeys(":TaskWikiSort description-")
+        self.client.type("<Enter>")
+        sleep(0.5)
diff --git a/tests/test_splits.py b/tests/test_splits.py
index 9b583a3..6f7a59b 100644
--- a/tests/test_splits.py
+++ b/tests/test_splits.py
@@ -246,6 +246,30 @@ class TestTagsSimple(IntegrationTest):
         assert re.search(work, output, re.MULTILINE)
 
 
+class TestTagsSimpleFiltered(IntegrationTest):
+
+    tasks = [
+        dict(description="home task"),
+        dict(description="home chore task 1", tags=['chore']),
+        dict(description="home chore task 2", tags=['chore']),
+        dict(description="work task 1", tags=['work']),
+        dict(description="work task 2", tags=['work']),
+    ]
+
+    def execute(self):
+        self.command("TaskWikiTags +chore")
+        assert self.command(":py print vim.current.buffer", silent=False).startswith("<buffer tags")
+        output = '\n'.join(self.read_buffer())
+
+        header = r'\s*'.join(['Tag', 'Count'])
+        chores = r'\s*'.join(['chore', '2'])
+        work = r'\s*'.join(['work', '2'])
+
+        assert re.search(header, output, re.MULTILINE)
+        assert re.search(chores, output, re.MULTILINE)
+        assert not re.search(work, output, re.MULTILINE)
+
+
 class TestSplitReplacement(IntegrationTest):
 
     def execute(self):
diff --git a/tests/test_viewport.py b/tests/test_viewport.py
index 4abd079..6ef3127 100644
--- a/tests/test_viewport.py
+++ b/tests/test_viewport.py
@@ -227,6 +227,44 @@ class TestViewportInspection(IntegrationTest):
         assert self.command(":py print vim.current.buffer", regex="<buffer taskwiki.")
 
 
+class TestViewportInspectionWithVisibleTag(IntegrationTest):
+
+    viminput = """
+    === Work tasks | +work -VISIBLE ===
+    * [ ] tag work task  #{uuid}
+
+    === Home tasks | +home ===
+    * [ ] tag work task  #{uuid}
+    """
+
+    vimoutput = """
+    ViewPort inspection:
+    --------------------
+    Name: Work tasks
+    Filter used: -DELETED -PARENT +work
+    Defaults used: tags:['work']
+    Ordering used: due+,pri-,project+
+    Matching taskwarrior tasks: 0
+    Displayed tasks: 0
+    Tasks to be added:
+    Tasks to be deleted:
+    """
+
+    tasks = [
+        dict(description="tag work task", tags=['work', 'home']),
+    ]
+
+    def execute(self):
+        self.command("w", regex="written$", lines=1)
+        sleep(0.5)
+        self.client.feedkeys('1gg')
+        sleep(0.5)
+        self.client.feedkeys(r'\<CR>')
+        sleep(0.5)
+
+        assert self.command(":py print vim.current.buffer", regex="<buffer taskwiki.")
+
+
 class TestViewportsUnicodeTaskGeneration(IntegrationTest):
 
     viminput = """