def test_open_already(self, blocking):
url = 'test-url'
connection = Connection(url)
connection.is_open = Mock(return_value=True)
# test
connection.open()
# validation
self.assertFalse(blocking.called)
def test_open_already(self, blocking):
url = 'test-url'
connection = Connection(url)
connection.is_open = Mock(return_value=True)
# test
connection.open()
# validation
self.assertFalse(blocking.called)
class Sender(BaseSender):
"""
An AMQP message sender.
:ivar connection: A proton connection.
:type connection: Connection
"""
def __init__(self, url):
"""
:param url: The broker url.
:type url: str
"""
BaseSender.__init__(self, url)
self.connection = Connection(url)
def is_open(self):
"""
Get whether the messenger has been opened.
:return: True if open.
:rtype bool
"""
return self.connection.is_open()
@reliable
def open(self):
"""
Open the sender.
"""
if self.is_open():
# already opened
return
self.connection.open()
def repair(self):
"""
Repair the sender.
"""
self.close()
self.connection.close()
self.connection.open()
def close(self):
"""
Close the sender.
"""
pass
@resend
def send(self, address, content, ttl=None):
"""
Send a message.
:param address: An AMQP address.
:type address: str
:param content: The message content
:type content: buf
:param ttl: Time to Live (seconds)
:type ttl: float
"""
sender = self.connection.sender(address)
try:
message = build_message(content, ttl, self.durable)
sender.send(message)
log.debug('sent (%s)', address)
finally:
sender.close()
def test_is_open(self):
connection = Connection('')
self.assertFalse(connection.is_open())
connection._impl = Mock()
self.assertTrue(connection.is_open())
def test_is_open(self):
connection = Connection('')
self.assertFalse(connection.is_open())
connection._impl = Mock()
self.assertTrue(connection.is_open())
class Sender(BaseSender):
"""
An AMQP message sender.
:ivar connection: A proton connection.
:type connection: Connection
"""
def __init__(self, url):
"""
:param url: The broker url.
:type url: str
"""
BaseSender.__init__(self, url)
self.connection = Connection(url)
def is_open(self):
"""
Get whether the messenger has been opened.
:return: True if open.
:rtype bool
"""
return self.connection.is_open()
@reliable
def open(self):
"""
Open the sender.
"""
if self.is_open():
# already opened
return
self.connection.open()
def repair(self):
"""
Repair the sender.
"""
self.close()
self.connection.close()
self.connection.open()
def close(self):
"""
Close the sender.
"""
pass
@resend
def send(self, address, content, ttl=None):
"""
Send a message.
:param address: An AMQP address.
:type address: str
:param content: The message content
:type content: buf
:param ttl: Time to Live (seconds)
:type ttl: float
"""
sender = self.connection.sender(address)
try:
message = build_message(content, ttl, self.durable)
sender.send(message)
log.debug('sent (%s)', address)
finally:
sender.close()
