def test_set_default_context_with_context(self):
default_context = context_stack_test_utils.TestContext()
context_stack = context_stack_impl.ContextStackImpl(default_context)
context = context_stack_test_utils.TestContext()
self.assertIsNot(context_stack.current, context)
context_stack.set_default_context(context)
self.assertIs(context_stack.current, context)
def setUp(self):
super().setUp()
# In these tests we are setting the default context of the
# `context_stack_impl.context_stack`, so here we reset that context back to
# some known state.
self.context = context_stack_test_utils.TestContext()
context_stack_impl.context_stack.set_default_context(self.context)
def test_install_pushes_context_on_stack(self):
default_context = context_stack_test_utils.TestContext()
context_stack = context_stack_impl.ContextStackImpl(default_context)
self.assertIs(context_stack.current, default_context)
context_two = context_stack_test_utils.TestContext()
with context_stack.install(context_two):
self.assertIs(context_stack.current, context_two)
context_three = context_stack_test_utils.TestContext()
with context_stack.install(context_three):
self.assertIs(context_stack.current, context_three)
self.assertIs(context_stack.current, context_two)
self.assertIs(context_stack.current, default_context)
def setUp(self): super().setUp() context = context_stack_test_utils.TestContext() context_stack_impl.context_stack.set_default_context(context)
def test_set_default_context_raises_type_error_with_none(self):
default_context = context_stack_test_utils.TestContext()
context_stack = context_stack_impl.ContextStackImpl(default_context)
with self.assertRaises(TypeError):
context_stack.set_default_context(None)