def send_task(self, task_name, queue, deployment_id=None, args=None, kwargs=None, timeout=10, execution_env=None): cloudify_context = test_utils.op_context(task_name, task_target=queue, plugin_name=PLUGIN_NAME, execution_env=execution_env, deployment_id=deployment_id) kwargs = kwargs or {} kwargs['__cloudify_context'] = cloudify_context handler = amqp_client.BlockingRequestResponseHandler(exchange=queue) client = amqp_client.get_client() client.add_handler(handler) with client: task = {'cloudify_task': {'kwargs': kwargs}} result = handler.publish(task, routing_key='operation', timeout=timeout) error = result.get('error') if error: raise deserialize_known_exception(error) else: return result.get('result')
def send_task(self, task_name, queue, deployment_id=None, args=None, kwargs=None, timeout=10, execution_env=None): cloudify_context = test_utils.op_context(task_name, task_target=queue, plugin_name=PLUGIN_NAME, execution_env=execution_env, deployment_id=deployment_id) kwargs = kwargs or {} kwargs['__cloudify_context'] = cloudify_context handler = amqp_client.BlockingRequestResponseHandler(queue) client = amqp_client.get_client() client.add_handler(handler) with client: task = {'cloudify_task': {'kwargs': kwargs}} result = handler.publish(task, routing_key='operation', timeout=timeout) error = result.get('error') if error: raise deserialize_known_exception(error) else: return result.get('result')
def _assert_task_not_runnable(task_name, deployment_id=None, package_name=plugins.PACKAGE_NAME, package_version=None): pytest.raises( cloudify_exceptions.NonRecoverableError, dispatch.dispatch, test_utils.op_context(task_name, plugin_name=plugins.PLUGIN_NAME, deployment_id=deployment_id, package_name=package_name, package_version=package_version))
def _assert_task_runnable(task_name, expected_return=None, package_name=plugins.PACKAGE_NAME, package_version=None, deployment_id=None): assert (dispatch.dispatch( test_utils.op_context(task_name, plugin_name=plugins.PLUGIN_NAME, package_name=package_name, package_version=package_version, deployment_id=deployment_id))) == expected_return
def _assert_task_not_runnable(self, task_name, deployment_id=None, package_name=None, package_version=None): self.assertRaises( cloudify_exceptions.NonRecoverableError, dispatch.dispatch, test_utils.op_context(task_name, plugin_name=PLUGIN_NAME, deployment_id=deployment_id, package_name=package_name, package_version=package_version))
def _assert_task_runnable(self, task_name, expected_return=None, package_name=None, package_version=None, deployment_id=None): self.assertEqual( dispatch.dispatch(test_utils.op_context( task_name, plugin_name=PLUGIN_NAME, package_name=package_name, package_version=package_version, deployment_id=deployment_id)), expected_return)
def send_task(self, task_name, queue, deployment_id=None, args=None, kwargs=None, timeout=10, execution_env=None): cloudify_context = test_utils.op_context(task_name, task_target=queue, plugin_name=PLUGIN_NAME, execution_env=execution_env, deployment_id=deployment_id) kwargs = kwargs or {} kwargs['__cloudify_context'] = cloudify_context return self.celery.send_task(name='cloudify.dispatch.dispatch', queue=queue, args=args, kwargs=kwargs).get(timeout=timeout)
def send_task(self, task_name, queue, deployment_id=None, args=None, kwargs=None, timeout=10, execution_env=None): cloudify_context = test_utils.op_context(task_name, task_target=queue, plugin_name=PLUGIN_NAME, execution_env=execution_env, deployment_id=deployment_id) kwargs = kwargs or {} kwargs['__cloudify_context'] = cloudify_context return self.celery.send_task( name='cloudify.dispatch.dispatch', queue=queue, args=args, kwargs=kwargs).get(timeout=timeout)