From 5397152ae9485ee50e94688a676667ab29d6b87b Mon Sep 17 00:00:00 2001 From: Mark Paluch Date: Tue, 6 Apr 2021 12:59:54 +0200 Subject: [PATCH] Polishing. Reformat integration test. See #573 Original pull request: #574. --- .../data/r2dbc/core/PostgresIntegrationTests.java | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/test/java/org/springframework/data/r2dbc/core/PostgresIntegrationTests.java b/src/test/java/org/springframework/data/r2dbc/core/PostgresIntegrationTests.java index 3acb08aa..8a8b79cf 100644 --- a/src/test/java/org/springframework/data/r2dbc/core/PostgresIntegrationTests.java +++ b/src/test/java/org/springframework/data/r2dbc/core/PostgresIntegrationTests.java @@ -249,6 +249,7 @@ public class PostgresIntegrationTests extends R2dbcIntegrationTestSupport { @Test // gh-573 void shouldReadAndWriteInterval() { + EntityWithInterval entityWithInterval = new EntityWithInterval(); entityWithInterval.interval = Interval.of(Duration.ofHours(3)); @@ -261,11 +262,12 @@ public class PostgresIntegrationTests extends R2dbcIntegrationTestSupport { R2dbcEntityTemplate template = new R2dbcEntityTemplate(client, new DefaultReactiveDataAccessStrategy(PostgresDialect.INSTANCE)); - EntityWithInterval saved = template.insert(entityWithInterval).block(); - EntityWithInterval loaded = template.select(Query.empty(), EntityWithInterval.class) // - .blockLast(); + template.insert(entityWithInterval).thenMany(template.select(Query.empty(), EntityWithInterval.class)) // + .as(StepVerifier::create) // + .consumeNextWith(actual -> { - assertThat(saved.interval).isEqualTo(loaded.interval); + assertThat(actual.getInterval()).isEqualTo(entityWithInterval.interval); + }).verifyComplete(); } private void insert(EntityWithArrays object) {