def test_add_context_manager_inner(action_type, function_to_call, result): ac = Action(lambda: 10, lambda: -10) csm = SharedContextManager() ac.add_context_manager(action_type, csm.gen_context_manager("b", "a")) ac.add_context_manager(action_type, csm.gen_context_manager("b2", "a2")) ac.add_context_manager(action_type, csm.gen_context_manager("b3", "a3"), inner=True) getattr(ac, function_to_call)() assert csm.result == result
def test_copy(): a = Action(lambda: 10, lambda: -10) a.add_context_manager("action", MagicMock()) a.add_context_manager("rollback", MagicMock()) assert copy(a)