自动化测试工具(Java)---测试框架代码--ClassUtils

public class ClassUtils {
    private static final Logger logger = RequestIdentityLogger.getLogger(ClassUtils.class);

    public static String getClassPath(Class cls) {

        return cls.getProtectionDomain().getCodeSource().getLocation().getFile();
    }

    public static List<String> listAllClassesInCurrentDir(Class cls) {

        String path = getClassPath(cls);

        System.out.println(path);

        if (path.endsWith(".jar")) {
            return listAllClassesInJar(path);
        } else {
            return listAllClassesInFiles(path, "", null);
        }
    }

    /*
     * 获取所有的.class文件
     */
    private static List<String> listAllClassesInFiles(String path, String classPath, List<String> classesList) {

        if (path == null || classPath == null) {
            logger.error("in listAllClasses, the argument is null");
            return null;
        }

        if (classPath.equals("") == false) {
            classPath = classPath + ".";
        }

        if (classesList == null) {
            classesList = new ArrayList<String>();
        }

        File classFiles = new File(path);

        for (File classSubFile : classFiles.listFiles()) {
            if (classSubFile.isDirectory()) {
                listAllClassesInFiles(classSubFile.getPath(), classPath + classSubFile.getName(), classesList);
            } else {
                if (classSubFile.getName().endsWith(".class")) {
                    classesList.add(classPath
                            + classSubFile.getName().substring(0, classSubFile.getName().length() - 6));
                }
            }
        }

        return classesList;
    }

    private static List<String> listAllClassesInJar(String path) {

        String clsName = null;
        List<String> clsNameList = new ArrayList<String>();

        try {
            JarFile jarFile = new JarFile(path);
            Enumeration<JarEntry> entrys = jarFile.entries();

            while (entrys.hasMoreElements()) {
                JarEntry jarEnty = entrys.nextElement();

                clsName = jarEnty.getName();
                clsName = clsName.replace("/", ".");

                if (clsName.endsWith(".class")) {
                    clsNameList.add(clsName.substring(0, clsName.length() - 6));
                }
            }
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            return null;
        }

        return clsNameList;
    }
}