diff options
Diffstat (limited to 'tests/template_tests/test_context.py')
| -rw-r--r-- | tests/template_tests/test_context.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/template_tests/test_context.py b/tests/template_tests/test_context.py index 9c23a38af2..f7959e6415 100644 --- a/tests/template_tests/test_context.py +++ b/tests/template_tests/test_context.py @@ -21,6 +21,8 @@ class ContextTests(SimpleTestCase): self.assertEqual(c["a"], 1) self.assertEqual(c.get("foo", 42), 42) + def test_push_context_manager(self): + c = Context({"a": 1}) with c.push(): c['a'] = 2 self.assertEqual(c['a'], 2) @@ -30,6 +32,17 @@ class ContextTests(SimpleTestCase): self.assertEqual(c['a'], 3) self.assertEqual(c['a'], 1) + def test_update_context_manager(self): + c = Context({"a": 1}) + with c.update({}): + c['a'] = 2 + self.assertEqual(c['a'], 2) + self.assertEqual(c['a'], 1) + + with c.update({'a': 3}): + self.assertEqual(c['a'], 3) + self.assertEqual(c['a'], 1) + def test_setdefault(self): c = Context() |