org.jboss.test.faces.mock.FacesMocksClassControl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jsf-mock Show documentation
There is a newer version: 10.0.0
package org.jboss.test.faces.mock;

import static org.easymock.internal.ClassExtensionHelper.getInterceptor;
import static org.easymock.internal.ClassExtensionHelper.setCurrentConstructorArgs;

import java.lang.reflect.Method;

import org.easymock.ConstructorArgs;
import org.easymock.IMocksControl;
import org.easymock.internal.IProxyFactory;
import org.easymock.internal.MocksControl;

public class FacesMocksClassControl extends MocksControl implements IMocksControl {

    private static final long serialVersionUID = -6032144451192179422L;

    public FacesMocksClassControl(MockType type) {
        super(type);
    }

    public  T createMock(String name, Class toMock,
            Method... mockedMethods) {

        if(toMock.isInterface()) {
            throw new IllegalArgumentException("Partial mocking doesn't make sense for interface");
        }

        T mock = createMock(name, toMock);

        // Set the mocked methods on the interceptor
        getInterceptor(mock).setMockedMethods(mockedMethods);

        return mock;
    }

    public  T createMock(Class toMock, Method... mockedMethods) {

        if(toMock.isInterface()) {
            throw new IllegalArgumentException("Partial mocking doesn't make sense for interface");
        }

        T mock = createMock(toMock);

        // Set the mocked methods on the interceptor
        getInterceptor(mock).setMockedMethods(mockedMethods);

        return mock;
    }

    public  T createMock(Class toMock, ConstructorArgs constructorArgs,
            Method... mockedMethods) {
        // Trick to allow the ClassProxyFactory to access constructor args
        setCurrentConstructorArgs(constructorArgs);
        try {
            return createMock(toMock, mockedMethods);
        } finally {
            setCurrentConstructorArgs(null);
        }
    }

    public  T createMock(String name, Class toMock,
            ConstructorArgs constructorArgs, Method... mockedMethods) {
        // Trick to allow the ClassProxyFactory to access constructor args
        setCurrentConstructorArgs(constructorArgs);
        try {
            return createMock(name, toMock, mockedMethods);
        } finally {
            setCurrentConstructorArgs(null);
        }
    }

    @Override
    protected  IProxyFactory createProxyFactory(Class toMock) {
        if (toMock.isInterface()) {
            return super.createProxyFactory(toMock);
        }
        return new FacesClassProxyFactory();
    }
}