Fixed #24603 -- Allowed Context.update() to be used as a context manager.

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()