GH-208 polishing ec68f64 commit

Ensured that the composed Supplier is of type Mono<Void>
Fix tests
This commit is contained in:
Oleg Zhurakousky
2018-10-14 20:29:13 -04:00
parent 8c2fdfd193
commit 31e689b108
3 changed files with 7 additions and 6 deletions

View File

@@ -162,8 +162,8 @@ public class BeanFactoryFunctionCatalogTests {
processor.register(new FunctionRegistration<>(s, "supplier"));
Consumer<String> c = x -> ref.set(x.toUpperCase());
processor.register(new FunctionRegistration<>(c, "consumer"));
Supplier<Flux<Void>> f = processor.lookup("supplier|consumer");
f.get().blockFirst();
Supplier<Mono<Void>> f = processor.lookup("supplier|consumer");
((Mono<Void>)f.get()).block();
assertThat(ref.get()).isEqualTo("HELLO");
}

View File

@@ -121,8 +121,8 @@ public class ContextFunctionPostProcessorTests {
public void supplierAndConsumer() {
processor.register(new FunctionRegistration<Supplier<String>>(() -> "foo", "supplier"));
processor.register(new FunctionRegistration<Consumer<String>>(System.out::println, "consumer"));
Supplier<Flux<Void>> supplier = (Supplier<Flux<Void>>) processor.lookupSupplier("supplier|consumer");
assertNull(supplier.get().blockFirst());
Supplier<Mono<Void>> supplier = (Supplier<Mono<Void>>) processor.lookupSupplier("supplier|consumer");
assertNull(supplier.get().block());
}
@Test