def test_send(self, mock_gofer_agent):
context = Mock()
context.agent_id = "123"
context.url = "http://broker.com"
context.transport = "qpid"
context.authenticator = Mock()
task_id = "5678"
criteria = {"match": {"task_id": task_id}}
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_admin = Mock()
mock_agent.Admin = Mock(return_value=mock_admin)
agent = PulpAgent()
agent.cancel(context, task_id)
mock_gofer_agent.assert_called_with(
context.agent_id,
url=context.url,
transport=context.transport,
authenticator=context.authenticator,
async=True,
)
mock_admin.cancel.assert_called_with(criteria=criteria)
def test_send(self, mock_gofer_agent):
context = Mock()
context.agent_id = '123'
context.url = 'http://broker.com'
context.transport = 'qpid'
context.authenticator = Mock()
task_id = '5678'
criteria = {'match': {'task_id': task_id}}
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_admin = Mock()
mock_agent.Admin = Mock(return_value=mock_admin)
agent = PulpAgent()
agent.cancel(context, task_id)
mock_gofer_agent.assert_called_with(
context.agent_id,
url=context.url,
transport=context.transport,
authenticator=context.authenticator,
async=True)
mock_admin.cancel.assert_called_with(criteria=criteria)
def test_cancel(self):
# Test
task_id = '123'
agent = DirectAgent(CONSUMER)
agent.cancel(task_id)
# Verify
criteria = {'eq': task_id}
mock_agent.Admin.cancel.assert_called_once_with(criteria=criteria)
def test_cancel(self):
# Test
task_id = '123'
agent = DirectAgent(CONSUMER)
agent.cancel(task_id)
# Verify
criteria = {'eq': task_id}
mock_agent.Admin.cancel.assert_called_once_with(criteria=criteria)
def cancel_request(self, consumer_id, task_id):
"""
Cancel an agent request associated with the specified task ID.
:param consumer_id: The consumer ID.
:type consumer_id: str
:param: task_id: The task ID associated with the request.
:type: str
"""
manager = managers.consumer_manager()
consumer = manager.get_consumer(consumer_id)
context = Context(consumer)
agent = PulpAgent()
agent.cancel(context, task_id)
def cancel_request(self, consumer_id, task_id):
"""
Cancel an agent request associated with the specified task ID.
:param consumer_id: The consumer ID.
:type consumer_id: str
:param: task_id: The task ID associated with the request.
:type: str
"""
manager = managers.consumer_manager()
consumer = manager.get_consumer(consumer_id)
context = Context(consumer)
agent = PulpAgent()
agent.cancel(context, task_id)
def test_cancel(self, _agent):
context = Context()
task_id = '5678'
# test
agent = PulpAgent()
agent.cancel(context, task_id)
# validation
context.__enter__.assert_called_once_with()
self.assertTrue(context.__exit__.called)
_agent.assert_called_with(
context.url,
context.address,
authenticator=context.authenticator,
wait=0)
_agent.return_value.Admin.return_value.cancel.assert_called_with(
criteria={'match': {'task_id': task_id}})
def test_cancel(self, _agent):
context = Context()
task_id = '5678'
# test
agent = PulpAgent()
agent.cancel(context, task_id)
# validation
context.__enter__.assert_called_once_with()
self.assertTrue(context.__exit__.called)
_agent.assert_called_with(context.url,
context.address,
authenticator=context.authenticator,
wait=0)
_agent.return_value.Admin.return_value.cancel.assert_called_with(
criteria={'match': {
'task_id': task_id
}})
def test_send(self, mock_gofer_agent):
context = Mock()
context.route = '123'
context.url = 'http://broker.com'
context.authenticator = Mock()
task_id = '5678'
criteria = {'match': {'task_id': task_id}}
mock_agent = Mock()
mock_gofer_agent.return_value = mock_agent
mock_admin = Mock()
mock_agent.Admin = Mock(return_value=mock_admin)
agent = PulpAgent()
agent.cancel(context, task_id)
mock_gofer_agent.assert_called_with(
context.url,
context.route,
authenticator=context.authenticator,
wait=0)
mock_admin.cancel.assert_called_with(criteria=criteria)
