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)