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))
