基于Java+HttpClient+TestNG的接口自动化测试框架（八）------ 针对文件的处理

在实际的接口测试中，有时需要根据情况进行文件的上传和下载。在文件数量比较小的时候，我们当然可以直接处理（比如若干个接口都用一个文件）。但是，如果我们上传的文件需要使用不同文件夹里不同的文件，而且数量又比较大的情况下，直接写路径就比较麻烦了。

那么，怎样来处理这样的问题呢？还是需要先整理一下思路。

首先，我们需要根据某些特征，找到这些文件。例如，我们可以在指定的文件夹下面，按照文件后缀名来找寻文件。通常，我们在DOS下查询该文件夹下寻找同一后缀名文件的时候，会使用通配符“*”来代替文件名，例如：*.jpg,*.xls等等。那么，我们可以指定一个文件夹，然后使用将通配符变为正则的方式，来匹配查找文件。

另外，如果在我们指定的文件夹中，存在子文件夹的情况，我们需要进行递归处理。即，进入子文件夹后，再次对子文件夹下的对象进行遍历，以此类推。

先来看一段代码：

private static String generatePattern(String fileConf) {
        fileConf = fileConf.trim();
        // 根据配置生成正确的正则
        fileConf = fileConf.replace(‘*‘, ‘#‘);
        // 将*号之前加上.
        fileConf = fileConf.replaceAll("#", ".*");
        return fileConf;
    }

这里，我们定义了一个生成正则的方法。就是将"*.XXX"变成“.*.XXX”（仔细看，就是在之前加了一个"."）。我们可以利用这个正则表达式来匹配我们想要寻找的文件。

接下来，我们来看一下递归查找的问题。首先，我们需要定义一个要查找的文件夹。而在java中，我们可以直接定义一个主路径作为file的对象。

例如：

String c = "k:/abc/";
        File file = new File(c);

这样的话，可以使用getName()方法获取文件或文件夹的名称（不包含上级路径）。另外，也可以使用listFiles() 返回目录下的文件或者目录对象（File类实例），包含隐藏文件。当然，对于文件，这样操作会返回null。

下面，我们来看一下递归的操作。思路是这样的：

1.判断当前的file对象是文件还是文件夹

2.如果是文件，若匹配正则，则将文件加入list。

3.如果是文件夹，则获取该文件夹下面所有的文件或者目录对象，若匹配正则，则将文件加入list。

private static ArrayList<File> filePattern(File file, Pattern p) {
        if (file == null) {
            return null;
        }
        // 如果是文件,则加入fileList
        if (file.isFile()) {
            Matcher fMatcher = p.matcher(file.getName());
            if (fMatcher.matches()) {
                ArrayList<File> list = new ArrayList<File>();
                list.add(file);
                return list;
            }
            // 如果是目录，则进行递归查找
        } else if (file.isDirectory()) {
            File[] files = file.listFiles();
            if (files != null && files.length > 0) {
                ArrayList<File> list = new ArrayList<File>();
                for (File f : files) {
                    //此处进行递归
                    ArrayList<File> rlist = filePattern(f, p);
                    if (rlist != null) {
                        //将查找结果都加入fileList
                        list.addAll(rlist);
                    }
                }
                return list;
            }
        }
        return null;
    }

那么，根据上面的代码，我们可以将路径下所有的匹配正则的文件都找到，并放入fileList中。

接下来，我们来定义一个方法，使用主路径以及通配符作为参数，来进行文件的查找。在通配符参数的指定中，我们使用“;”来进行分隔，来找到我们所需要的文件。

例如：

String dirs = "k:/abc/";
String file_con = "*.doc;*.xls";

这里会出现一个新的问题，如果我按照上面所写，会把“k:/abc/”下的所有匹配的文件都找出来。

假如，我是需要寻找主路经下test1文件夹下的*.doc,和test2文件夹下的*.xls怎么办？

这里在指定参数的部分，我们需要对文件夹作出判断，即对最后一个“/”进行分隔处理。

请看下面一段代码：

public static List<File> getFilesByConf(String dir, String fileConf) {
        String[] fileConfArr = fileConf.split(";");// 多个配置‘;’分开
        List<File> list = new ArrayList<File>();
        if (fileConfArr != null && fileConfArr.length > 0) {
            for (String conftemp : fileConfArr) {
                int at = conftemp.lastIndexOf("/");
                File file = null;
                String fileContextPath = "";
                // 绝对目录路径
                String contextPath = dir;
                // 针对目录部分的处理
                if (at > 0) {
                    fileContextPath = fileConf.substring(0, at);
                }
                if (StringUtil.isNotEmpty(fileContextPath)) {
                    contextPath = contextPath + fileContextPath;
                }
                file = new File(contextPath);
                String fileNameConf = conftemp.substring(at + 1,conftemp.length());// 文件名配置
                String fileConfTemp = generatePattern(fileNameConf);
                Pattern p = Pattern.compile(fileConfTemp);
                ArrayList<File> listtemp = filePattern(file, p);
                list.addAll(listtemp);
            }
        }
        return removeDuplicate(list);// 去重
    }

这里，我们就完成了针对路径的处理。顺便说一下，这里的去重，使用的是利用HashSet的特性来进行去重。

public static List removeDuplicate(List list) {
        if (list == null) {
            return null;
        }
        Set set = new HashSet();
        List newList = new ArrayList();
        for (Iterator iter = list.iterator(); iter.hasNext();) {
            Object obj = iter.next();
            if (set.add(obj)) {
                newList.add(obj);
            }
        }
        return newList;
    }

至此，我们就完成了对于文件查找的处理。

总结一下，我们只需要提供两个参数，一个是主路经，一个匹配的通配符的字符串参数（可以带路径），即可批量查找到我们所需要的文件。

接下来，我们看一下，下载文件的处理。

下载文件一般比较好处理，基本思路就是使用输入流来写入，然后保存即可。

需要注意的是，针对比较大的文件时，我们需要设置Buffer（即缓冲区）。不应用缓冲区的时候，每次读取一个字节，写入一个字节，由于操作磁盘比内存慢的很多，所以不应用缓冲区效率很低；应用缓冲区，可以一次读取多个字节，先不写入磁盘，而是放入内存之中，到了缓冲区大小的时候，在写入磁盘，减少了对磁盘的操作，效率较高。

public static boolean writeFile(InputStream is, String filePath) {
        File file = new File(filePath);
        if (!file.getParentFile().exists()) {
            file.getParentFile().mkdirs();
        }
        FileOutputStream fileout;
        try {
            fileout = new FileOutputStream(file);
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            return false;
        }
        /**
         * 根据实际运行效果 设置缓冲区大小
         */
        byte[] buffer = new byte[10 * 1024];
        int ch = 0;
        try {
            while ((ch = is.read(buffer)) != -1) {
                fileout.write(buffer, 0, ch);
            }
            return true;
        } catch (IOException e) {
            e.printStackTrace();
            return false;
        } finally {
            try {
                is.close();
                fileout.flush();
                fileout.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

这里根据实际情况，来设置缓冲区的大小。

这样就可以根据接口返回的内容，写入文件到指定的路径中。