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)