使用GroboUtils多线程并发请求测试springmvc controller

1. 需求

有一个下载pdf的springmvc controller，希望模拟一下并发访问的情况，怎么办呢？

2. 解决方案

由于是pdf，其http响应为

那么我们访问时返回的是二进制，而不是网页那种text/html

于是写一个保存二进制文件的方法

public void saveBinaryFile(URL url) throws IOException {        URLConnection uc = url.openConnection();        String contentType = uc.getContentType();        int contentLength = uc.getContentLength();        if (contentType.startsWith("text/") || contentLength == -1 ) {            throw new IOException("This is not a binary file.");        }        InputStream raw = uc.getInputStream();        InputStream in = new BufferedInputStream(raw);        byte[] data = new byte[contentLength];        int offset = 0;        while (offset < contentLength) {            int bytesRead = in.read(data, offset, data.length - offset);            if (bytesRead == -1) break;            offset += bytesRead;        }        if (offset != contentLength) {            throw new IOException("Only read " + offset                    + " bytes; Expected " + contentLength + " bytes");        }        String filename = UUID.randomUUID()+".pdf";        try (FileOutputStream fout = new FileOutputStream(filename)) {            fout.write(data);            fout.flush();        }    }

测试方法

Junit本身是不支持普通的多线程测试的，这是因为Junit的底层实现上，是用System.exit退出用例执行的。JVM都终止了，在测试线程启动的其他线程自然也无法执行。

使用Junit多线程测试的开源的第三方的工具包GroboUtils

pom.xml

<!--springmvc中进行多线程测试--><dependency><groupId>net.sourceforge.groboutils</groupId><artifactId>groboutils-core</artifactId><version>5</version><scope>system</scope><systemPath>F:/jar_package/groboutils-core-5.jar</systemPath></dependency>

测试方法

    @Test    public void testSaveBinaryFile3() throws IOException {        TestRunnable runner = new TestRunnable() {            @Override            public void runTest() throws Throwable {                //测试内容                String urlStr = "http://localhost:8019/xxx”;                URL url = null;                try {                    url = new URL(urlStr);                    new RenderPdfControllerTest().saveBinaryFile(url);                } catch (IOException e) {                    e.printStackTrace();                }            }        };        //Rnner数组，相当于并发多少个        int runnerCount = 5;        TestRunnable[] trs = new TestRunnable[runnerCount];        for (int i = 0; i < runnerCount; i++) {            trs[i] = runner;        }        // 用于执行多线程测试用例的Runner，将前面定义的单个Runner组成的数组传入        MultiThreadedTestRunner mttr = new MultiThreadedTestRunner(trs);        try {            // 开发并发执行数组里定义的内容            mttr.runTestRunnables();        } catch (Throwable e) {            e.printStackTrace();        }    }

3. 原理

GroboUtils实现原理是让测试主线程等待所有测试子线程执行完成后再退出。想到的办法是Thread中的join方法。看一下其关键的源码实现

参考：Junit spring 多线程测试