p-unit 是一款开放源码的性能测试框架，和 JUnit 不同，JUnit 关注的是测试案例的正确性，而 p-unit 不仅关注测试案例的正确性，还收集测试案例的性能参数，默认情况下，p-unit 收集测试案例的时间和内存消耗情况，可以产生文件，图片，和 PDF 格式的报表。此外，p-unit 还支持参数化测试，多线程测试以及不同 Java 虚拟机性能之间的比较。
p-unit 简介
或许我们已经习惯了使用 JUnit 来写单元测试来保证代码质量（我也一直这么做），但可能经常碰到这样的问题：
程序多线程下正确性如何？
如何测试程序的性能？
当有多个方案可以选择时，技术上如何比较不同方案的性能？
对于问题 1，我们或许听天由命？或是凭借人工分析，或是根据用户反馈？很多软件单线程下的单元测试覆盖率相当高，从而保证了代码的健壮性。然而多线程测试时常被忽略，这并不代表多线程测试不重要，相反，修正一个用户报告的多线程 BUG 往往比单线程的要高出很多，因为测试案例经常不是 100% 可重现的。这更要求程序员在开发阶段充分的重视。目前多线程单元测试力度不够的一个重要原因是没有一个像 JUnit 那样易用的测试工具，另外重复写测试案例往往不被程序员接受。
对于问题 2，一个成熟的关心性能的产品往往有一个性能测试平台。这个测试平台应该关注的是测试业务逻辑本身，而无需关心如何运行测试案例。你是否为写这样的测试平台痛苦过？以及花费时间在产生一些直观的报表上面？
对于问题 3，我们往往写一个原型来比较不同产品之间的性能，如何比较执行速度和内存消耗？或是选择最适合你的虚拟机？
p-unit 就是这么一款开源的性能测试软件，它能帮助你很好的解决上述问题。p-unit 可以：
多线程支持：同一个测试案例可以单线程执行，也可以多线程执行，测试案例开发者只需写一套测试案例。
参数化测试案例：很多测试案例，需要测试同一功能在不同数量级上的性能表现。
不同虚拟机性能测试：只需指定虚拟机路径，即可测试同一个测试案例在不同虚拟机上的表现，报表上可以非常直观显示性能差别。
事件机制构架：punit 是基于事件机制构架的，如果用户想定制报表，只需实现事件响应器，并注册该响应器到 punit 核心即可。
多线程执行测试案例
在了解如何多线程执行测试案例之前，我们先了解一下如何利用 p-unit 单线程执行测试案例。不同于 JUnit， p-unit 测试用例无需继承任何测试类或是实现接口，即可执行 test 开始的方法。尽管 JUnit 4 中加入了注释（Annotation） 的特性，但测试方法前缀为 "test" 仍然是测试者们的首选。因此如果你的 JUnit 测试案例遵循的是 test 命名规则，那么 p-uni t可以兼容运行 JUnit 测试案例。
下面的代码清单 1 是一个最为普通的测试案例:
清单 1. 测试案例 1
public class SimpleTestClass {
public void setUp() {
SampleUtil.doSomething();
}
public void tearDown() {
SampleUtil.doSomething();
}
public void testA() {
System.out.println("testA");
SampleUtil.doSomething();
}
public void testB() {
SampleUtil.doSomething();
}
public void testC() {
SampleUtil.doSomething();
}
}
public class SampleUtil {
private static Random _random = new Random();
public static void consumeMemory(int length) {
byte[] data = new byte[length];
for(int i = 0, j = 0; i < data.length; ++i) {
++j;
}
}
public static void consumeTime(int time) {
ThreadUtil.sleepIgnoreInterruption(time);
}
public static void doSomething() {
consumeTime(Math.abs(_random.nextInt()) % 500);
consumeMemory(Math.abs(_random.nextInt()) % 100000);
}
}
这是做为普通的测试案例，但是注意到这仅仅是一个测试案例，不包含其他任何逻辑，这也是 p-unit 追求的业务逻辑和测试运行环境分离的一个理念。同一个测试案例，用户可以选择不同的测试环境去运行，而不是绑定在某一个特定的测试软件工具上。现在我们来看 p-unit 是如何运行这个测试案例的。你只需要在 main 函数中写一行代码来运行它:
清单 2. 单线程运行测试案例
public static void main(String[] args) {
new PUnitSoloRunner().run(SimpleTestClass.class);
}
清单 3. 单线程运行测试案例结果
[solo] Started running samples.SimpleTestClass
samples.SimpleTestClass
testA
testA() - [287.0ms]
testB() - [27.0ms]
testC() - [213.0ms]
total: 3, failures:0 (GREEN) - 2025.0ms
是否和想象中的一样？下面我们来看如何多线程执行这个测试案例。或许从上面的例子你已经猜到了，在 main 函数还是只需一句代码，只用把 PUnitSoloRunner 换成 PUnitConcurrentRunner 即可！
清单 4. 多线程运行测试案例
public static void main(String[] args) {
new PUnitConcurrentRunner().run(SimpleTestClass.class);
}
清单 5. 多线程运行测试案例结果
[concurrent] Started running samples.SimpleTestClass
samples.SimpleTestClass
testA
testA
testA
testA
testA
testA
testA
testA
testA
testA
testA() - [405.0ms]
testB() - [469.0ms]
testC() - [503.0ms]
total: 3, failures:0 (GREEN) - 1447.0ms
是否和想象中的一样？默认情况 p-unit 启动 10 个线程来执行，要指定不同的线程数，只需将线程数做为参数传入 PUnitConcurrentRunner 即可。p-unit 甚至支持不同的测试案例有不同的线程数，这要求测试案例实现 p-unit 中定义的 Concurrent 接口，该接口的定义为：
清单 6. p-unit Concurrent 接口
public interface Concurrent {
public int concurrentCount();
}
该接口的意思，相信无需再多做解释了，返回该测试案例需要的线程数。
