def __init__(self, socket): self.socket = socket self.user = None self.name = None self.quest = None # Create callbacks that have weakref to self self.on_enter = smoke.weak(self._on_enter) self.on_leave = smoke.weak(self._on_leave) self.on_say = smoke.weak(self._on_say)
def test_weak_raises_disconnect(): d = Dummy() w = weak(d.spam) del d gc.collect() with assert_raises(instance_of(Disconnect)): w()
def test_weak_forwards_call(): d = Dummy() d.value = 'spam' w = weak(d.spam) r = w() assert_that(r, equal_to(d.value))