def wait(self, msg):
while not self.transport:
waiting()
# wait for msg
deferred = WaitingDeferred(msg)
self.expected_response.insert(0, deferred)
return deferred
def wait_for(self, msg):
self.waiting = True
deferred = self.factory.wait(msg)
deferred.addCallback(self._finished)
deferred.addErrback(self._failed)
while self.waiting:
waiting()
def check(self, msg, deferred=None):
while not self.transport:
waiting()
# send msg
if not deferred:
deferred = defer.Deferred()
self.expected_response.insert(0, deferred)
self.transport.write(msg+"\r\n")
return deferred
def check_next(self, in_msg, out_msg):
self.waiting = True
def _assert_msg(msg):
self.waiting = False
self.assertEquals(msg, out_msg)
deferred = self.factory.check(in_msg)
deferred.addCallback(_assert_msg)
deferred.addErrback(self._failed)
while self.waiting:
waiting()
def tearDown(self):
"""overrides TestCase method"""
# close connection
self.factory.stopTrying()
self.connector.disconnect()
self.done = True
# remove timeout
try:
self.timeout.cancel()
reactor.iterate(0.1)
except (error.AlreadyCancelled, error.AlreadyCalled):
pass
# stop listening
defered = self.navigator.stopListening()
if defered:
self.done = False
defered.addCallback(self._finished)
defered.addErrback(self._failed)
while not self.done:
waiting()
def tearDown(self):
"""overrides TestCase method"""
# close connection
self.factory.stopTrying()
del self.factory
self.connector.disconnect()
# remove timeout
try:
self.timeout.cancel()
except (error.AlreadyCancelled, error.AlreadyCalled):
print "timeout canceled"
# stop listening
deferred = self.navigator.stopListening()
if deferred:
self.waiting = True
deferred.addCallback(self._finished)
deferred.addErrback(self._failed)
while self.waiting:
waiting()
pendings = reactor.getDelayedCalls()
for pending in pendings:
pending.cancel()
reactor.iterate()
def write(self, msg):
while not self.transport:
waiting()
# send msg
self.expected_response.insert(0, None)
self.transport.write(msg+"\r\n")
def _wait_for(self, msg):
self.waiting = True
deferred = self.factory.wait(msg)
deferred.addCallback(self._stop_waiting)
while self.waiting:
waiting()
