org.mule.module.ibeans.spi.MuleMockCallAnnotationHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mule-module-ibeans
Exposes iBeans over endpoints allowing developers to poll iBean methods and invoke iBean methods as part of an outbound flow
There is a newer version: 3.3.1
Show newest version
/*
 * $Id: MuleMockCallAnnotationHandler.java 19453 2010-09-08 15:07:22Z rossmason $
 * -------------------------------------------------------------------------------------
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.module.ibeans.spi;

import org.mule.api.MuleContext;
import org.mule.transport.http.HttpConstants;

import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.activation.DataSource;
import javax.activation.MimeTypeParseException;

import org.ibeans.api.DataType;
import org.ibeans.api.InvocationContext;
import org.ibeans.api.Response;
import org.ibeans.api.channel.MimeType;
import org.ibeans.impl.support.datatype.DataTypeFactory;
import org.ibeans.impl.test.MockIBean;
import org.ibeans.impl.test.MockMessageCallback;
import org.ibeans.spi.IBeansPlugin;

/**
 * A handler for supporting the {@link org.ibeans.annotation.MockIntegrationBean} annotation. This implementation uses
 * Mockito to mock out the call itself. Mockito can be used in tests to supply response data.
 */
public class MuleMockCallAnnotationHandler extends MuleCallAnnotationHandler implements MockIBean
{
    private Object mock;
    private InvocationContext ctx;
    private MockMessageCallback callback;
    private IBeansPlugin plugin;

    public MuleMockCallAnnotationHandler(MuleContext muleContext, Object mock, IBeansPlugin plugin)
    {
        super(muleContext);
        this.mock = mock;
        this.plugin = plugin;
    }

    @Override
    public Response invoke(InvocationContext ctx) throws Exception
    {
        this.ctx = ctx;
        Method method = ctx.getMethod();
        //Special handling of methods with an ibean prefix, these are called by the the IBeansTestSupport
        //To pass in additional information from the testcase
        Object result;
        if (method.getName().startsWith("ibean"))
        {
            result = method.invoke(this, ctx.getArgs());
        }
        else
        {
            result = ctx.getMethod().invoke(mock, ctx.getArgs());
        }

        Map props = new HashMap();
        props.put(HttpConstants.HEADER_CONTENT_TYPE, (ctx.getInvocationReturnType()==null ?
                    ctx.getReturnType().getMimeType() : ctx.getInvocationReturnType().getMimeType()));
        Response response = (result instanceof Response ? (Response)result : plugin.createResponse(result, props, null));

        //Now handled in the test case
        if (callback != null)
        {
            try
            {
                callback.onMessage(response);
            }
            finally
            {
                //Only run it once
                callback = null;
                this.ctx = null;
            }

        }
        return response;
    }

    public String getScheme(Method method)
    {
        //Default to http, should not make a difference for a Mock iBean
        return "http";
    }

    public void ibeanSetMimeType(MimeType mime) throws MimeTypeParseException
    {
        if (ctx.getIBeanConfig().getReturnType() != null)
        {
            ctx.setInvocationReturnType(DataTypeFactory.create(ctx.getIBeanConfig().getReturnType().getType(), mime));
        }
    }

    public DataType ibeanReturnType()
    {
//        if (invocationContext == null || invocationContext.getReturnType().getType().getName().equals("void"))
//        {
//            return helper.getReturnType();
//        }
//        else
        {
            DataType type = ctx.getIBeanConfig().getReturnType();
            if(type==null)
            {
                type = ctx.getIBeanDefaultConfig().getReturnType();
            }
            return type;
        }
    }

    public Object ibeanUriParam(String name)
    {
//        if (invocationContext == null)
//        {
//            return helper.getDefaultUriParams().get(name);
//        }
//        else
        {
            return ctx.getIBeanConfig().getUriParams().get(name);
        }
    }

    public Object ibeanHeaderParam(String name)
    {
//        if (invocationContext == null)
//        {
//            return helper.getDefaultHeaderParams().get(name);
//        }
//        else
        {
            return ctx.getIBeanConfig().getHeaderParams().get(name);
        }
    }

    public Object ibeanPropertyParam(String name)
    {
//        if (invocationContext == null)
//        {
//            return helper.getDefaultPropertyParams().get(name);
//        }
//        else
        {
            return ctx.getIBeanConfig().getPropertyParams().get(name);
        }
    }

    public Object ibeanPayloadParam(String name)
    {
//        if (invocationContext == null)
//        {
//            return helper.getDefaultPayloadParams().get(name);
//        }
//        else
        {
            return ctx.getIBeanConfig().getPayloadParams().get(name);
        }
    }

    public List