diff --git a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/ReactorRules.java b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/ReactorRules.java index 21ca2d4a..f09a9ab5 100644 --- a/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/ReactorRules.java +++ b/error-prone-contrib/src/main/java/tech/picnic/errorprone/refasterrules/ReactorRules.java @@ -557,6 +557,7 @@ final class ReactorRules { @Matches(IsIdentityOperation.class) Function> identityOperation) { return Refaster.anyOf( + flux.concatMap(function, 0), flux.flatMap(function, 1), flux.flatMapSequential(function, 1), flux.map(function).concatMap(identityOperation)); diff --git a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestInput.java b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestInput.java index 300c70e8..881428ef 100644 --- a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestInput.java +++ b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestInput.java @@ -207,11 +207,12 @@ final class ReactorRulesTest implements RefasterRuleCollectionTestCase { ImmutableSet> testFluxConcatMap() { return ImmutableSet.of( - Flux.just(1).flatMap(Mono::just, 1), - Flux.just(2).flatMapSequential(Mono::just, 1), - Flux.just(3).map(Mono::just).concatMap(identity()), - Flux.just(4).map(Mono::just).concatMap(v -> v), - Flux.just(5).map(Mono::just).concatMap(v -> Mono.empty())); + Flux.just(1).concatMap(Mono::just, 0), + Flux.just(2).flatMap(Mono::just, 1), + Flux.just(3).flatMapSequential(Mono::just, 1), + Flux.just(4).map(Mono::just).concatMap(identity()), + Flux.just(5).map(Mono::just).concatMap(v -> v), + Flux.just(6).map(Mono::just).concatMap(v -> Mono.empty())); } ImmutableSet> testFluxConcatMapWithPrefetch() { diff --git a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestOutput.java b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestOutput.java index a8751ff1..a9dcd26c 100644 --- a/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestOutput.java +++ b/error-prone-contrib/src/test/resources/tech/picnic/errorprone/refasterrules/ReactorRulesTestOutput.java @@ -214,7 +214,8 @@ final class ReactorRulesTest implements RefasterRuleCollectionTestCase { Flux.just(2).concatMap(Mono::just), Flux.just(3).concatMap(Mono::just), Flux.just(4).concatMap(Mono::just), - Flux.just(5).map(Mono::just).concatMap(v -> Mono.empty())); + Flux.just(5).concatMap(Mono::just), + Flux.just(6).map(Mono::just).concatMap(v -> Mono.empty())); } ImmutableSet> testFluxConcatMapWithPrefetch() {