Add stump for get_variable

2 files changed, 12 insertions, 0 deletions

diff --git a/yarnhelper.py b/yarnhelper.py
index 4501eab..b5c45d0 100644
--- a/yarnhelper.py
+++ b/yarnhelper.py
@@ -35,6 +35,9 @@ class YarnHelper(object):
         self._next_match += 1
         return self._env[name]
 
+    def get_variable(self, name):
+        raise Error('no variable {}'.format(name))
+
 
 class Error(Exception):
 
diff --git a/yarnhelper_tests.py b/yarnhelper_tests.py
index a61102f..bdc0cd9 100644
--- a/yarnhelper_tests.py
+++ b/yarnhelper_tests.py
@@ -53,3 +53,12 @@ class GetNextMatchTests(unittest.TestCase):
         self.assertEqual(h.get_next_match(), 'first')
         with self.assertRaises(yarnhelper.Error):
             h.get_next_match()
+
+
+class PersistentVariableTests(unittest.TestCase):
+
+    def test_raises_error_if_no_such_variable(self):
+        h = yarnhelper.YarnHelper()
+        with self.assertRaises(yarnhelper.Error):
+            h.get_variable('FOO')
+