mirror of
https://github.com/tbabej/taskwiki.git
synced 2025-08-19 15:53:07 +02:00
123 lines
3.8 KiB
Python
123 lines
3.8 KiB
Python
import os
|
|
import tempfile
|
|
import vimrunner
|
|
|
|
from tasklib.task import TaskWarrior, Task
|
|
|
|
server = vimrunner.Server()
|
|
|
|
class IntegrationTest(object):
|
|
|
|
input = None
|
|
output = None
|
|
|
|
def add_plugin(self, name):
|
|
plugin_base = os.path.expanduser('~/.vim/bundle/')
|
|
plugin_path = os.path.join(plugin_base, name)
|
|
self.client.add_plugin(plugin_path)
|
|
|
|
def write_buffer(self, lines, position=0):
|
|
result = self.client.write_buffer(position + 1, lines)
|
|
assert result == u"0"
|
|
|
|
def read_buffer(self, start=0, end=1000):
|
|
return self.client.read_buffer(
|
|
unicode(start+1),
|
|
unicode(end+1)
|
|
).splitlines()
|
|
|
|
|
|
def generate_data(self):
|
|
self.dir = tempfile.mkdtemp(dir='/tmp/')
|
|
self.tw = TaskWarrior(data_location=self.dir)
|
|
self.tasks = [
|
|
Task(self.tw, description="project random task 1", project="Random"),
|
|
Task(self.tw, description="project random task 2", project="Random"),
|
|
Task(self.tw, description="tag home task 1", tags=["home"]),
|
|
Task(self.tw, description="tag work task 1", tags=["work"]),
|
|
Task(self.tw, description="today task 1", due="now"),
|
|
]
|
|
|
|
for task in self.tasks:
|
|
task.save()
|
|
|
|
def setup(self):
|
|
self.generate_data()
|
|
self.client = server.start_gvim()
|
|
self.add_plugin('taskwiki')
|
|
self.add_plugin('vimwiki')
|
|
self.command('let g:taskwiki_data_location="{0}"'.format(self.dir))
|
|
self.client.edit(os.path.join(self.dir, 'testwiki.txt'))
|
|
self.command('set filetype=vimwiki')
|
|
|
|
def teardown(self):
|
|
self.client.quit()
|
|
|
|
def command(self, command):
|
|
return self.client.command(command)
|
|
|
|
def check_sanity(self):
|
|
"""
|
|
Makes sanity checks upon the vim instance.
|
|
"""
|
|
|
|
# Assert all the important files were loaded
|
|
scriptnames = self.client.command('scriptnames').splitlines()
|
|
expected_loaded_files = [
|
|
'vimwiki/autoload/vimwiki/base.vim',
|
|
'vimwiki/ftplugin/vimwiki.vim',
|
|
'vimwiki/autoload/vimwiki/u.vim',
|
|
'vimwiki/syntax/omnipresent_syntax.vim',
|
|
'vimwiki/syntax/vimwiki.vim',
|
|
'taskwiki/ftplugin/vimwiki.vim',
|
|
]
|
|
|
|
# Do a partial match for each line from scriptnames
|
|
for scriptfile in expected_loaded_files:
|
|
assert any([scriptfile in line for line in scriptnames])
|
|
|
|
# Assert only note about Bram being maintainer is in messages
|
|
bramline = u'Messages maintainer: Bram Moolenaar <Bram@vim.org>'
|
|
assert self.client.command('messages') == bramline
|
|
|
|
# Assert that TW and cache objects exist
|
|
tw_class = self.client.command('py print(tw.__class__.__name__)')
|
|
cache_class = self.client.command('py print(cache.__class__.__name__)')
|
|
|
|
assert tw_class == 'TaskWarrior'
|
|
assert cache_class == 'TaskCache'
|
|
|
|
def test_execute(self):
|
|
|
|
# First, run sanity checks
|
|
self.check_sanity()
|
|
|
|
# Then load the input
|
|
if self.input:
|
|
self.write_buffer(input)
|
|
|
|
# Do the stuff
|
|
self.execute()
|
|
|
|
# Check expected output
|
|
if self.output:
|
|
assert self.read_buffer() == self.output
|
|
|
|
class TestBurndown(IntegrationTest):
|
|
|
|
def execute(self):
|
|
self.command("TaskWikiBurndownDaily")
|
|
assert self.command(":py print vim.current.buffer").startswith("<buffer burndown.daily")
|
|
assert "Daily Burndown" in self.read_buffer()[0]
|
|
|
|
class TestViewports(IntegrationTest):
|
|
|
|
def execute(self):
|
|
lines = ["=== Work tasks | +work ==="]
|
|
self.write_buffer(lines)
|
|
self.command("w")
|
|
assert self.read_buffer() == [
|
|
"=== Work tasks | +work ===",
|
|
"* [ ] tag work task 1 #{0}".format(self.tasks[3]['uuid'])
|
|
]
|
|
|