From 7ca18a4137f3046f22c25fbadfe2bc16c9ef0afc Mon Sep 17 00:00:00 2001
From: Tomas Babej <tomasbabej@gmail.com>
Date: Mon, 12 Dec 2016 23:43:02 +0100
Subject: [PATCH] tests: Add test for viewport being terminated by a header

---
 tests/test_viewport.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/tests/test_viewport.py b/tests/test_viewport.py
index 9ccfe91..e874ac8 100644
--- a/tests/test_viewport.py
+++ b/tests/test_viewport.py
@@ -194,6 +194,39 @@ class TestViewportDefaultsExplicitEmpty(IntegrationTest):
         assert task['tags'] == set()
 
 
+class TestViewportDefaultsTerminatedByHeader(IntegrationTest):
+
+    viminput = """
+    === Work tasks | +work ===
+    * [ ] tag work task
+
+    === Unrelated work tasks ===
+    * [ ] not tagged work task
+    """
+
+    vimoutput = """
+    === Work tasks | +work ===
+    * [ ] tag work task  #{uuid}
+
+    === Unrelated work tasks ===
+    * [ ] not tagged work task  #{uuid}
+    """
+
+    def execute(self):
+        self.command("w", regex="written$", lines=1)
+        assert len(self.tw.tasks.pending()) == 2
+
+        task = self.tw.tasks.filter('+work')[0]
+        assert task['description'] == 'tag work task'
+        assert task['status'] == 'pending'
+        assert task['tags'] == set(['work'])
+
+        task = self.tw.tasks.filter('-work')[0]
+        assert task['description'] == 'not tagged work task'
+        assert task['status'] == 'pending'
+        assert task['tags'] == set()
+
+
 class TestViewportInspection(IntegrationTest):
 
     viminput = """