class QueueSubscriberTestCase(unittest.TestCase):
def setUp(self):
self.qs = QueueSubscriber("dummy_queue", 0)
client = ClientStub("testhostname", None, None)
self.qs.setClient(client)
@deferred(timeout=30)
def test_resume_after_disconnect(self):
"""La production doit reprendre après une déconnexion"""
self.qs.consumer = Mock()
self.qs.consumer.write.side_effect = lambda msg: self.qs.resumeProducing()
self.qs.client.stub_connect()
self.qs.resumeProducing()
print "deconnexion"
self.qs._queue.close()
self.qs.connectionLost(None)
print "reconnexion"
self.qs.connectionInitialized()
print "réception d'un message"
self.qs.client.stub_receive("dummy message")
def check(r):
print "vérification"
print self.qs.consumer.write.call_args_list
self.assertTrue(self.qs.consumer.write.called,
"la fonction write() n'a pas été appelée")
self.assertEqual("dummy message",
self.qs.consumer.write.call_args_list[0][0][0])
self.qs.ready.addCallback(check)
return self.qs.ready
def setUp(self):
self.qs = QueueSubscriber("dummy_queue", 0)
client = ClientStub("testhostname", None, None)
self.qs.setClient(client)
