From 4d7ea13d7ff3f988da64aeabd6e3610c6547c8ea Mon Sep 17 00:00:00 2001
From: Tomas Babej <tomasbabej@gmail.com>
Date: Thu, 26 Mar 2015 20:55:51 +0100
Subject: [PATCH] tests: Make existing tests use the new infrastructure

---
 tests/test_integration.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/test_integration.py b/tests/test_integration.py
index ebe4f7d..84b4e73 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -103,14 +103,16 @@ class IntegrationTest(object):
         if self.output:
             assert self.read_buffer() == self.output
 
-    def test_focus_burndown_daily(self):
+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(TestIntegration):
+class TestViewports(IntegrationTest):
 
-    def test_viewport_filling(self):
+    def execute(self):
         lines = ["=== Work tasks | +work ==="]
         self.write_buffer(lines)
         self.command("w")