tests: Make existing tests use the new infrastructure

This commit is contained in:
Tomas Babej 2015-03-26 20:55:51 +01:00
parent e630aa4fe2
commit 4d7ea13d7f

View file

@ -103,14 +103,16 @@ class IntegrationTest(object):
if self.output: if self.output:
assert self.read_buffer() == self.output assert self.read_buffer() == self.output
def test_focus_burndown_daily(self): class TestBurndown(IntegrationTest):
def execute(self):
self.command("TaskWikiBurndownDaily") self.command("TaskWikiBurndownDaily")
assert self.command(":py print vim.current.buffer").startswith("<buffer burndown.daily") assert self.command(":py print vim.current.buffer").startswith("<buffer burndown.daily")
assert "Daily Burndown" in self.read_buffer()[0] assert "Daily Burndown" in self.read_buffer()[0]
class TestViewports(TestIntegration): class TestViewports(IntegrationTest):
def test_viewport_filling(self): def execute(self):
lines = ["=== Work tasks | +work ==="] lines = ["=== Work tasks | +work ==="]
self.write_buffer(lines) self.write_buffer(lines)
self.command("w") self.command("w")