通过将运行时参数传递给java反射方法进行单元测试

时间:2014-11-19 20:31:05

标签: java unit-testing methods

我正在使用代码中的java bean创建一个类对象。然后我调用该类obj的特定方法

public static void runUnitTest(String className, String methodName, List<Object> inputParams, Object expectedReturnValue){
                try {
        // Make the class object to be tested on
        Object classObj = Class.forName(className).newInstance();
        Method calledMethod = classObj.getClass().getMethod(methodName,inputParams.get(0).getClass());
        Object returnVal = calledMethod.invoke(classObj,inputParams.get(0));
        }catch (InstantiationException | IllegalAccessException
                        | ClassNotFoundException | NoSuchMethodException | SecurityException | IllegalArgumentException | InvocationTargetException e) {
                    e.printStackTrace();
                }
    }

我这样称呼它:

public static void main(String[] args) throws IOException { 
        List<Object> inputParams = new ArrayList<Object>();
        inputParams.add(new BigDecimal(1234));
        runUnitTest("NumberTest","getOutputNumber",inputParams,new BigDecimal(5678));
    }

NumberTest的代码:

public class NumberTest{

    public BigDecimal getOutputNumber(BigDecimal numberId) {
        if(numberId.intValue() == 1234)
        {
            return new BigDecimal(5678);
        }else 
            return new BigDecimal(0);
    }

    public BigDecimal getAdditionalOutputNumber(BigDecimal numberId, String additionalInfo) {
        if(numberId.intValue() == 1234 && "Pass".equals(additionalInfo))
        {
            return new BigDecimal(5678);
        }else 
            return new BigDecimal(0);
    }
}

这很好用,因为我知道方法getOutputNumber只有一个参数。但是当我必须为参数数量不同的多种方法调用相同的代码时(例如getAdditionalOutputNumber),我不能使用相同的代码。我不想在inputParams的大小的基础上使用多个if else或case块。 是否有一种通用的方式来调用以下内容:

Method calledMethod = classObj.getClass().getMethod(methodName,**?? What to pass here ??**);
Object returnVal = calledMethod.invoke(classObj,**?? What to pass here ??**);

1 个答案:

答案 0 :(得分:2)

您只需从参数列表中构建合适的数组即可调用反射API。

Class[] types = new Class[inputParams.size()];
int i = 0;
for(Object param:inputParams) {
    types[i++] = param.getClass();
}
Method calledMethod = classObj.getClass().getMethod(methodName,types);
Object returnVal = calledMethod.invoke(classObj,inputParams.toArray());

原始类型和空值可能存在一些问题。