# -*- test-case-name: vumi.transports.intergrat.tests.test_failures -*-

from vumi.transports.failures import FailureWorker


class IntegratFailureWorker(FailureWorker):

    def do_retry(self, message, reason):
        message = self.update_retry_metadata(message)
        self.store_failure(message, reason, message['retry_metadata']['delay'])

    def handle_failure(self, message, reason):
        if reason == "connection refused":
            self.do_retry(message, reason)
        else:
            self.store_failure(message, reason)

# -*- test-case-name: vumi.transports.integrat.tests.test_failures -*-

from vumi.transports.failures import FailureWorker


class IntegratFailureWorker(FailureWorker):

    def do_retry(self, message, reason):
        message = self.update_retry_metadata(message)
        self.store_failure(message, reason, message['retry_metadata']['delay'])

    def handle_failure(self, message, reason):
        if reason == "connection refused":
            self.do_retry(message, reason)
        else:
            self.store_failure(message, reason)

Fix link to failure tests.
Fix link to failure tests.
