completion: Add tab completion for TaskWikiMod

tests/test_completion.py

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+from taskwiki.completion import Completion
+from tests.base import IntegrationTest
+
+
+class FakeTW():
+    def __init__(self, projects=[], tags=[]):
+        self.projects = projects
+        self.tags = tags
+
+    def execute_command(self, args):
+        if args == ["_unique", "project"]:
+            return self.projects
+        elif args == ["_unique", "tag"]:
+            return self.tags
+        elif args == ["_columns"]:
+            return ["priority", "project", "due", "end"]
+        else:
+            assert False
+
+
+class TestCompletionUnit():
+    def test_attributes(self):
+        c = Completion(FakeTW())
+        assert c.modify("") == ["+", "-", "due:", "end:", "priority:", "project:"]
+        assert c.modify("pr") == ["priority:", "project:"]
+        assert c.modify("pri") == ["priority:"]
+
+    def test_projects(self):
+        c = Completion(FakeTW(projects=["aa", "ab", "c"]))
+        assert c.modify("proj:") == ["proj:aa", "proj:ab", "proj:c"]
+        assert c.modify("proj:a") == ["proj:aa", "proj:ab"]
+        assert c.modify("proj:ab") == ["proj:ab"]
+        assert c.modify("proj:abc") == []
+        assert c.modify("pr:") == []
+        assert c.modify("project:ab") == ["project:ab"]
+
+    def test_tags(self):
+        c = Completion(FakeTW(tags=["aa", "aa,ab", "c"]))
+        assert c.modify("+") == ["+aa", "+ab", "+c"]
+        assert c.modify("+a") == ["+aa", "+ab"]
+        assert c.modify("+ab") == ["+ab"]
+        assert c.modify("+abc") == []
+
+    def test_dates(self):
+        c = Completion(FakeTW())
+        assert c.modify("end:no") == ["end:now", "end:november"]
+        assert c.modify("sch:jan") == ["sch:january"]
+
+    def test_recur(self):
+        c = Completion(FakeTW())
+        assert c.modify("re:da") == ["re:daily", "re:day"]
+        assert c.modify("recur:q") == ["recur:quarterly"]
+
+
+class TestCompletionIntegration(IntegrationTest):
+    viminput = """
+    * [ ] test task 1  #{uuid}
+    * [ ] test task 2  #{uuid}
+    """
+
+    tasks = [
+        dict(description="test task 1", project="ABC"),
+        dict(description="test task 2", project="DEF"),
+    ]
+
+    def execute(self):
+        self.client.feedkeys(":TaskWikiMod\\<Enter>")
+        self.client.eval('feedkeys("pro\\<Tab>D\\<Tab>\\<Enter>", "t")')
+        self.client.eval('0')  # wait for command completion
+
+        for task in self.tasks:
+            task.refresh()
+
+        assert self.tasks[0]['project'] == "DEF"