def test_signal_weakref_context_manager_delete_after():
import gc
result = []
class Foo:
@contextmanager
def on_ctx_test(self, before, after):
result.append(before)
yield
result.append(after)
foo = Foo()
register_object(foo)
with on_ctx_test(before=1, after=2):
assert result == [1]
assert result == [1, 2]
del foo
gc.collect()
with on_ctx_test(before=3, after=4):
assert result == [1, 2]
assert result == [1, 2]
def test_signal_weakref_context_manager_delete_during():
import gc
result = []
class Foo:
@contextmanager
def on_ctx_test(self, before, after):
result.append(before)
yield
result.append(after)
foo = Foo()
register_object(foo)
with on_ctx_test(before=1, after=2):
assert result == [1]
assert result == [1, 2]
with on_ctx_test(before=3, after=4):
assert result == [1, 2, 3]
del foo
gc.collect()
# The context manager should keep the signal handler alive
assert result == [1, 2, 3, 4]
def test_ctx():
result = []
class Foo:
@contextmanager
def on_ctx_test(self, before, after):
result.append(before)
yield
result.append(after)
foo = Foo()
register_object(foo)
assert result == []
with on_ctx_test(before=1, after=2):
assert result == [1]
assert result == [1, 2]
unregister_object(foo)
with on_ctx_test(before=3, after=4):
assert result == [1, 2]
assert result == [1, 2]