java - 如何在 JUnit 测试用例中设置运行时超时

Question

我使用 JUnit v4 作为测试框架。我想知道如何在测试用例中设置运行时超时？

我正在使用参数化测试，其中我有一个场景列表，其中包含超时值和一些其他字段。每个场景可能有不同的超时。

timeout 参数并不能帮助我实现这一目标。

@Test(timeout = getTimeOut())
public void secureLoginWithLongUsername() {
    // Test case goes here

}

private final long getTimeOut() {
    // I am doing some processing here to calculate timeOut dynamically
    long timeOut = scenario.getTimeOut();
    return timeOut;
}

@Parameters
public static Collection<Scenario[]> getParameters() {

    List<Scenario[]> scenarioList = new ArrayList<Scenario[]>();
    Configuration config = new Configuration();
    List<Scenario> scenarios = config.getScenarios();
    for (Scenario scenario : scenarios) {
        scenarioList.add(new Scenario[] { scenario });
    }

    return scenarioList;
}

public class Configuration {
    private List<Scenario>   scenarios;
    //Some processing here
    public List<Scenario> getScenarios() {
        return scenarios;
    }
}

public class Scenario {
    private long   timeOut;
    private String   name;
    //Some more fields here
}

请帮助我找到动态设置超时的替代方案。

Answer 1

我认为，你需要自己构建它，例如:

private Timer timer;

@After
public void terminateTimeout() {
    if (timer != null) {
        timer.cancel();
        timer = null;
    }
}

@Test
public void testTimeout() throws Exception {
    setTimeout(1000);
    // run test...
}

private void setTimeout(int duration) {
    final Thread currentThread = Thread.currentThread();
    timer = new Timer();
    timer.schedule(new TimerTask() {
        @Override
        public void run() {
            currentThread.interrupt();
        }
    }, duration);
}