def test_set_default_context_with_context(self):
context = context_stack_test_utils.TestContext('test')
context_stack = context_stack_impl.ContextStackImpl()
self.assertIsNot(context_stack.current, context)
context_stack.set_default_context(context)
self.assertIs(context_stack.current, context)
def test_set_default_context_with_none(self):
context = context_stack_test_utils.TestContext('test')
context_stack = context_stack_impl.ContextStackImpl()
context_stack.set_default_context(context)
self.assertIs(context_stack.current, context)
context_stack.set_default_context(None)
self.assertIsNot(context_stack.current, context)
self.assertIsInstance(context_stack.current, context_base.Context)
def test_install_pushes_context_on_stack(self):
context_stack = context_stack_impl.ContextStackImpl()
self.assertIsInstance(context_stack.current, context_base.Context)
context_one = context_stack_test_utils.TestContext('test')
with context_stack.install(context_one):
self.assertIs(context_stack.current, context_one)
context_two = context_stack_test_utils.TestContext('test')
with context_stack.install(context_two):
self.assertIs(context_stack.current, context_two)
self.assertIs(context_stack.current, context_one)
self.assertIsInstance(context_stack.current, context_base.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 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)