def testDisconnect2(self):
rr, target = self.setupTarget(HelperTarget())
self.lost = 0
rr.notifyOnDisconnect(self.disconnected)
rr.tracker.broker.transport.loseConnection(CONNECTION_LOST)
d = eventually()
d.addCallback(lambda res: self.failUnless(self.lost))
return d
def testDisconnect2(self):
rr, target = self.setupTarget(HelperTarget())
self.lost = 0
rr.notifyOnDisconnect(self.disconnected)
rr.tracker.broker.transport.loseConnection(CONNECTION_LOST)
d = eventually()
d.addCallback(lambda res: self.failUnless(self.lost))
return d
def flush(self):
self.connected = False
return eventually()
def loseConnection(self, why=failure.Failure(CONNECTION_DONE)):
if self.connected:
self.connected = False
# this one is slightly weird because 'why' is a Failure
eventually().addCallback(lambda res: self._loseConnection(why))
def write(self, data):
eventually(data).addCallback(self._write)
def flush(self):
self.connected = False
return eventually()
def loseConnection(self, why=failure.Failure(CONNECTION_DONE)):
if self.connected:
self.connected = False
# this one is slightly weird because 'why' is a Failure
eventually().addCallback(lambda res: self._loseConnection(why))
def write(self, data):
eventually(data).addCallback(self._write)
