Suggestions

PicnicSupermarket · Oct 6, 2022 · a8e8c58 · a8e8c58
1 parent c3f6d15
commit a8e8c58
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/...rone-contrib/src/main/java/tech/picnic/errorprone/refastertemplates/ReactorTemplates.java b/...rone-contrib/src/main/java/tech/picnic/errorprone/refastertemplates/ReactorTemplates.java
@@ -288,7 +288,7 @@ Mono<T> after(Mono<T> mono) {
   static final class FluxOnErrorComplete<T> {
     @BeforeTemplate
     Flux<T> before(Flux<T> flux) {
-      return flux.onErrorResume(e -> Flux.empty());
+      return flux.onErrorResume(e -> Refaster.anyOf(Mono.empty(), Flux.empty()));
     }
 
     @AfterTemplate
@@ -301,7 +301,7 @@ Flux<T> after(Flux<T> flux) {
   static final class PublisherProbeEmpty<T> {
     @BeforeTemplate
     PublisherProbe<T> before() {
-      return Refaster.anyOf(PublisherProbe.of(Mono.empty()), PublisherProbe.of(Flux.empty()));
+      return PublisherProbe.of(Refaster.anyOf(Mono.empty(), Flux.empty()));
     }
 
     @AfterTemplate

diff --git a/...rc/test/resources/tech/picnic/errorprone/refastertemplates/ReactorTemplatesTestInput.java b/...rc/test/resources/tech/picnic/errorprone/refastertemplates/ReactorTemplatesTestInput.java
@@ -99,8 +99,10 @@ Mono<Integer> testMonoOnErrorComplete() {
     return Mono.just(1).onErrorResume(e -> Mono.empty());
   }
 
-  Flux<Integer> testFluxOnErrorComplete() {
-    return Flux.just(1).onErrorResume(e -> Flux.empty());
+  ImmutableSet<Flux<Integer>> testFluxOnErrorComplete() {
+    return ImmutableSet.of(
+        Flux.just(1).onErrorResume(e -> Mono.empty()),
+        Flux.just(2).onErrorResume(e -> Flux.empty()));
   }
 
   ImmutableSet<PublisherProbe<Void>> testPublisherProbeEmpty() {

diff --git a/...c/test/resources/tech/picnic/errorprone/refastertemplates/ReactorTemplatesTestOutput.java b/...c/test/resources/tech/picnic/errorprone/refastertemplates/ReactorTemplatesTestOutput.java
@@ -98,8 +98,8 @@ Mono<Integer> testMonoOnErrorComplete() {
     return Mono.just(1).onErrorComplete();
   }
 
-  Flux<Integer> testFluxOnErrorComplete() {
-    return Flux.just(1).onErrorComplete();
+  ImmutableSet<Flux<Integer>> testFluxOnErrorComplete() {
+    return ImmutableSet.of(Flux.just(1).onErrorComplete(), Flux.just(2).onErrorComplete());
   }
 
   ImmutableSet<PublisherProbe<Void>> testPublisherProbeEmpty() {