SpaceVim/bundle/JavaUnit.vim/src/com/wsdjeg/util/TestMethod.java

package com.wsdjeg.util;

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;

public class TestMethod{
    public static void main(String[] args) {
        if(args.length == 2){
            testSpecifiedMethod(args[0],args[1]);
        }else if (args.length == 1) {
            testAllMethods(args[0]);
        }else if (args.length > 2){
            testMethods(args);
        }
    }
    public static void testAllMethods(String className){
        Class<?> clazz = null;
        try {
            clazz = Class.forName(className);
        } catch(Exception e){
            e.printStackTrace();
        }
        Method[] mds = clazz.getMethods();
        for (int i = 0; i < mds.length; i++) {
            if (mds[i].getName().startsWith("test")) {
                testSpecifiedMethod(className,mds[i].getName());
            }
        }
    }
    public static void testMethods(String[] args) {
        for (int i = 1; i < args.length; i++) {
            testSpecifiedMethod(args[0],args[i]);
        }
    }
    @SuppressWarnings("unchecked")
    public static void testSpecifiedMethod(String className,String methodName){
        try{
            System.out.println("========================= JavaUnite Test =============================");
            System.out.println("CLASS:" + className);
            System.out.println("METHOD:" + methodName + "()");
            System.out.println("============================= OUTPUT =================================");
            Class<?> clazz = Class.forName(className);
            Constructor<?> c = clazz.getDeclaredConstructor();
            c.setAccessible(true);
            long startTime = System.currentTimeMillis();
            clazz.getMethod(methodName).invoke(c.newInstance());
            long endTime = System.currentTimeMillis();
            System.out.println("================ SUCCESS: " + (endTime - startTime) + "ms ======================");
        }catch(Exception e){
            System.out.println("============================= FAILED ================================");
            e.printStackTrace();
        }
    }
}