在@Rule中并行化测试执行

时间:2013-12-09 14:35:35

标签: java junit junit-rule

我想重用一些集成测试来进行负载测试。我实现了一个由注释参数化的规则:

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Parallel {

    int invocations() default 1;

    int rampUpTime() default 0;
}

在我的规则实现中,评估注释并设置一个语句,该语句具有类似的评估方法:

@Override
public void evaluate() throws Throwable {
    ScheduledExecutorService exe = Executors.newScheduledThreadPool(invocations);

    for (int i = 0; i < invocations; i++) {
        ScheduledFuture<?> scheduledFuture = exe.schedule(new Runnable() {

            @Override
            public void run() {
                try {
                    invocated++;

                    // Request test = Request.method(description.getTestClass(), description.getMethodName());
                    // new JUnitCore().run(test);

                    statement.evaluate();
                } catch (Throwable e) {
                    e.printStackTrace();
                }
            }
        }, i * rampUpTime, this.timeUnit);
        futures.add(scheduledFuture);
    }
}

因此,evaluate调用将包含在Runnable()中,并按照注释中的说明进行调度。事情是:在我的规则中,只有调度发生,跑步者不知道(或关心)所有可运行的只有在设置整个测试套件时才运行的运行。所以我试图将对evalute()的调用添加到测试运行器中。首先尝试使用JUnitCore.run(...)当然以递归结束。

接下来的尝试是收集所有的期货并等待它们完成。这在每个测试的基础上工作正常,但我想并行执行整个测试套件。而且,在我的测试报告中,我只看到测试被执行一次。

那么我认为我使用带有上下文的参数化套件(一个收集所有测试的所有未来的对象)作为参数,但我没有找到一种方法将这个上下文对象提升为测试,每个测试都有有自己的参数。

我现在要求一种方法将我的规则中的多个执行添加到正在执行它的测试运行器。

1 个答案:

答案 0 :(得分:2)

如果我正确理解您,您正试图控制测试运行器的执行行为。 TestRule可能仅限于实现该目标。但是如何编写自己的Runner来控制单元测试的执行流程呢?看看吧 ParentRunner of JUnit。从那里,您应该很好地了解基本调度的工作原理,并且可能实现您自己的org.junit.experimental.ParallelComputer版本。