Check that error is raises if no more matches

1 files changed, 12 insertions, 1 deletions

diff --git a/yarnhelper_tests.py b/yarnhelper_tests.py
index d13abb8..a61102f 100644
--- a/yarnhelper_tests.py
+++ b/yarnhelper_tests.py
@@ -31,7 +31,9 @@ class GetNextMatchTests(unittest.TestCase):
 
     def test_returns_first_match_if_there(self):
         h = yarnhelper.YarnHelper()
-        h.set_environment({'MATCH_1': 'first'})
+        h.set_environment({
+            'MATCH_1': 'first',
+        })
         self.assertEqual(h.get_next_match(), 'first')
 
     def test_returns_second_match_if_there(self):
@@ -42,3 +44,12 @@ class GetNextMatchTests(unittest.TestCase):
         })
         self.assertEqual(h.get_next_match(), 'first')
         self.assertEqual(h.get_next_match(), 'second')
+
+    def test_raises_error_if_no_more_matches(self):
+        h = yarnhelper.YarnHelper()
+        h.set_environment({
+            'MATCH_1': 'first',
+        })
+        self.assertEqual(h.get_next_match(), 'first')
+        with self.assertRaises(yarnhelper.Error):
+            h.get_next_match()