Check that second match can be returned

1 files changed, 10 insertions, 1 deletions

diff --git a/yarnhelper_tests.py b/yarnhelper_tests.py
index 5c49901..d13abb8 100644
--- a/yarnhelper_tests.py
+++ b/yarnhelper_tests.py
@@ -29,7 +29,16 @@ class GetNextMatchTests(unittest.TestCase):
         with self.assertRaises(yarnhelper.Error):
             h.get_next_match()
 
-    def test_returns_match_if_there(self):
+    def test_returns_first_match_if_there(self):
         h = yarnhelper.YarnHelper()
         h.set_environment({'MATCH_1': 'first'})
         self.assertEqual(h.get_next_match(), 'first')
+
+    def test_returns_second_match_if_there(self):
+        h = yarnhelper.YarnHelper()
+        h.set_environment({
+            'MATCH_1': 'first',
+            'MATCH_2': 'second',
+        })
+        self.assertEqual(h.get_next_match(), 'first')
+        self.assertEqual(h.get_next_match(), 'second')