junit 参数化测试中参数流的限制

blmhpbnm 于 2023-10-20 发布在其他

关注(0)|答案(1)|浏览(111)

我想写一个参数化测试，它可以帮助MethodSource生成一个随机值序列。现在我想知道，是否有任何声明性的方法来限制测试运行的数量。

@ParameterizedTest
@MethodSource("create64RandomBytes")
// @SomeLimitAnnotation(100) wanted here
void testStuff(byte[] randomBytes) {
    // ...
}

static Stream<Arguments> create64RandomBytes() {
    Random rnd = new Random(42l);
    return Stream.generate(() -> {
        byte[] bytes = new byte[64];
        rnd.nextBytes(bytes);
        return Arguments.of(bytes);
    }).limit(100); // I want to avoid this "limit" call
}

当然，我可以限制流中项目的数量，但由于@MethodSource是为消费流而构建的（JUnit开发人员肯定知道无限流），我想知道是否有更好的方法来做到这一点。
TLDR：我可以在我的测试方法上用注解替换limit(100)调用吗？

junit

来源：https://stackoverflow.com/questions/61517640/limit-infinite-stream-of-arguments-in-parameterized-test

1条答案

按热度按时间

np8igboo1#

以下内容对你有好处吗？

@RegisterExtension
static final IntegerResolver integerResolver = new IntegerResolver();

static class IntegerResolver implements ParameterResolver {
    @Override
    public boolean supportsParameter(
        ParameterContext parameterContext,
        ExtensionContext extensionContext
    ) {
        return parameterContext.getParameter().getType() == int.class;
    }

    @Override
    public Object resolveParameter(
        ParameterContext parameterContext,
        ExtensionContext extensionContext
    ) {
        return 100;
    }
}

@ParameterizedTest
@MethodSource("create64RandomBytes")
void myTest(byte[] randomBytes) {
    // ...
}

static Stream<Arguments> create64RandomBytes(int count) {
    Random rnd = new Random(42l);
    return Stream.generate(() -> {
        byte[] bytes = new byte[64];
        rnd.nextBytes(bytes);
        return Arguments.of(bytes);
    }).limit(count);
}

赞(0）回复(0）举报 2023-10-20

我来回答

junit 参数化测试中参数流的限制

1条答案

相关问题

热门标签

最新问答