/**
1. 將本util類原始碼加到主專案的src/test/java/目錄下
2. 修改程式碼中的path引數，
String path = "D:/workspace/xxxx-util";//專案根目錄
3. 啟動main方法，自動生成專案中所有需要進行單元測試的java檔案。

下面為單元測試原始碼
*/

/**
 * maven 依賴
 * 
    <!-- testng 支援 -->
    <dependency>
        <groupId>org.testng</groupId>
        <artifactId>testng</artifactId>
    </dependency>
    <dependency>
        <groupId>org.mockito</groupId>
        <artifactId>mockito-all</artifactId>
        <version>1.10.19</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-api-mockito</artifactId>
        <version>1.7.4</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.springframework</groupId>
        <artifactId>spring-test</artifactId>
        <scope>test</scope>
    </dependency>
    <!-- testng 支援 end -->
*/
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
 * 生產單元測試的工具類
 * ps: 只是為了單元測試覆蓋率 ， 實際功能需要測試
 * 2018-04-12
 *
 */
public class CommonTestUtil {
    public static final String mainPath = "main\\java\\";
    public static final String testPath = "test\\java\\";
    public static final String endSuffix = ".java";

    public static void main(String[] args) {

        String path = "D:/workspace/xxxx-util";//專案根目錄

        CommonTestUtil.getFiles(new File(path + "/src/" + CommonTestUtil.mainPath));
    }
    /**
     * 獲取測試類主方法
     * @param clzss
     * @return
     */
    public static String getTestCode(Class<?> clzss) {
        String beanName = Character.toLowerCase(
                clzss.getSimpleName().charAt(0)) + clzss.getSimpleName().substring(1);

        StringBuilder sb = new StringBuilder();
        sb.append(clzss.getPackage().toString()).append(";").append("\r\n");
        sb.append("\r\n");
        sb.append("public class ").append(clzss.getSimpleName()).append("Test {").append("\r\n");
        sb.append("\r\n");
        //單元測試類
        sb.append("\t").append("@org.mockito.InjectMocks").append("\r\n");
        sb.append("\t").append(clzss.getName()).append(" ").append(beanName).append(";").append("\r\n");
        //@Autowired @Resource 的類
        sb.append(CommonTestUtil.getMockObj(clzss));

        //輸入引數 初始化
        Map<String, Set<String>> info = CommonTestUtil.getInfo(clzss);
        for(String str : info.get("param")) {
            sb.append("\t").append("@org.mockito.Mock").append("\r\n");
            sb.append("\t").append(str).append(";").append("\r\n");
        }
        //初始化Mockito虛擬化物件
        sb.append(CommonTestUtil.beforeClass());

        //單元測試
        for(String str : info.get("method")) {
            sb.append("\r\n");
            sb.append("\t").append("@org.testng.annotations.Test").append("\r\n");
            String testMethodName = str.substring(0, str.indexOf("(")) + "Test";
            while(sb.indexOf(testMethodName) > -1) {
                testMethodName += "1";
            }
            sb.append("\t").append("public void ").append(testMethodName).append("() {").append("\r\n");
            sb.append("\t\t").append("try{").append("\r\n");
            sb.append("\t\t\t").append(beanName).append(".").append(str).append(";").append("\r\n");
            sb.append("\t\t").append("} catch(java.lang.Throwable e) {").append("\r\n");
            sb.append("//\t\t\t").append("e.printStackTrace();").append("\r\n");
            sb.append("\t\t").append("}").append("\r\n");
            sb.append("\t}").append("\r\n");
        }
        sb.append("}");
        return sb.toString();
    }
    /**
     * 建立單元測試主方法
     * @param directory
     */
    public static void getFiles(File directory) {
        if(directory.getName().endsWith("svn")) {
            return;
        }
        File[] list = directory.listFiles();
        if(null == list || 0 == list.length) {
            return;
        }
        for(File f : list) {
            if(f.isDirectory()) {
                getFiles(f);
                continue;
            }
            if(!f.getName().endsWith(endSuffix)) {
                continue;
            }
            String absolutePath = f.getAbsolutePath();
            String qualifiedName = absolutePath.substring(
                    absolutePath.indexOf(mainPath)+10).replace("\\", ".").replace(endSuffix, "");
            Class<?> clzss = null;
            try {
                clzss = Class.forName(qualifiedName);
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            }
            if(null == clzss || !isNeedTest(clzss)) {
                continue;
            }
            StringBuilder sb = new StringBuilder().append(absolutePath.replace(mainPath, testPath));
            sb.insert(sb.lastIndexOf(endSuffix), "Test");
            File testFile = new File(sb.toString());
            //建立目錄
            if(testFile.exists()) {
                continue;
            }
            String path = sb.delete(sb.lastIndexOf("\\"), sb.length()).toString();
            createDirectory(path);
            String testCode = CommonTestUtil.getTestCode(clzss);
            BufferedWriter bw = null;
            try {
                bw = new BufferedWriter(new FileWriter(testFile));
                bw.write(testCode);
                bw.flush();
                System.out.println(testFile.getAbsolutePath());
            } catch (IOException e) {
                e.printStackTrace();
            } finally {
                try {
                    bw.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
    /**
     * 判斷類是否符合構造單元測試類的方法
     * @param clzss
     * @return
     */
    public static boolean isNeedTest(Class<?> clzss) {
        return "public".equals(Modifier.toString(clzss.getModifiers()));
    }
    /**
     * 建立目錄及上層目錄
     * @param path
     * @return
     */
    public static boolean createDirectory(String path) {
        File f = new File(path);
        if(!f.exists()) {
            return f.mkdirs();
        }
        return true;
    }
    /**
     * 
     * @param clzss
     * @return
     */
    @SuppressWarnings("restriction")
    public static String getMockObj(Class<?> clzss) {
        StringBuilder sb = new StringBuilder();
        for(Field f : clzss.getDeclaredFields()) {
            if(f.getAnnotation(org.springframework.beans.factory.annotation.Autowired.class) != null
                    || f.getAnnotation(javax.annotation.Resource.class) != null) {
                sb.append("\t").append("@org.mockito.Mock").append("\r\n");
                sb.append("\t").append(f.getType().getName()).append(" ")
                        .append(f.getName()).append(";").append("\r\n");
            }
        }
        return sb.toString();
    }
    /**
     * init
     * @return
     */
    public static String beforeClass() {
        StringBuilder sb = new StringBuilder();
        sb.append("\t").append("@org.testng.annotations.BeforeClass").append("\r\n");
        sb.append("\t").append("public void beforeClass() {").append("\r\n");
        sb.append("\t\t").append("org.mockito.MockitoAnnotations.initMocks(this);").append("\r\n");
        sb.append("\t").append("}").append("\r\n");
        return sb.toString();
    }
    /**
     * 獲取需要測試類的 所有方法名       sss( 1, 1, 1L, 1L, false, false, dateFor, mapFor)
     * 及相應需要Mock虛擬化的引數  java.util.Map mapFor
     * @param clzss
     * @return
     */
    public static Map<String, Set<String>> getInfo(Class<?> clzss){
        Map<String, Set<String>> methodAndParam = new HashMap<String,Set<String>>();

        Set<String> methodSet = new HashSet<String>();
        Set<String> paramSet = new HashSet<String>();
        for(Method method : clzss.getDeclaredMethods()) {
            if(Modifier.toString(method.getModifiers()).indexOf("private") > -1) {
                continue;
            }
            StringBuilder tmp = new StringBuilder();

            Class<?>[] pt = method.getParameterTypes();
            if (pt != null && pt.length > 0) {
                for (Class<?> an : pt) {
                    tmp.append(", ");
                    tmp.append(getParamter(an, paramSet));
                }
            } else {
                tmp.append(",");
            }
            tmp.delete(0, 1);
            tmp.insert(0, method.getName()+"(");
            tmp.append(")");
            methodSet.add(tmp.toString());
        }
        methodAndParam.put("param", paramSet);
        methodAndParam.put("method", methodSet);
        return methodAndParam;
    }
    /**
     * 對輸入引數進行遍歷
     * @param an
     * @param paramSet
     * @return
     */
    public static String getParamter(Class<?> an, Set<String> paramSet){
        StringBuilder tmp = new StringBuilder();
        if(an.getName().startsWith("[L")) {
            String str = an.getName().substring(2, an.getName().length()-1);
            try {
                tmp.append("new ").append(str).append("[]{")
                .append(getParamter(Class.forName(str), paramSet))
                .append("}");
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            }
        } else if (String.class.equals(an)) {
            tmp.append("\"00\"");
        } else if (Boolean.class.equals(an) || "boolean".equals(an.getName())) {
            tmp.append("false");
        } else if (Integer.class.equals(an) || "int".equals(an.getName())) {
            tmp.append("1");
        } else if (Long.class.equals(an) || "long".equals(an.getName())) {
            tmp.append("1L");
        } else if (Double.class.equals(an) || "double".equals(an.getName())) {
            tmp.append("12D");
        } else if (StringBuffer.class.equals(an)) {
            tmp.append("new StringBuffer()");
        } else if (StringBuilder.class.equals(an)) {
            tmp.append("new StringBuilder()");
        } else if (List.class.equals(an)) {
            tmp.append("new java.util.ArrayList()");
        } else if (Map.class.equals(an)) {
            tmp.append("new java.util.HashMap()");
        } else if (Object.class.equals(an)) {
            tmp.append("new java.lang.Object()");
        } else {
            String paramName = Character.toLowerCase(
                    an.getSimpleName().charAt(0)) + an.getSimpleName().substring(1) + "For";
            String modi = Modifier.toString(an.getModifiers());
            if(!modi.contains("final")) {
                tmp.append(paramName);

                String param = an.getName() + " " + paramName;
                paramSet.add(param);
            } else {
                tmp.append("null");
            }
        }
        return tmp.toString();
    }
}