#243 - Polishing.

Add author tags. Extend test. Reformat code.
This commit is contained in:
Mark Paluch
2019-12-02 08:56:45 +01:00
parent 38bbd4838e
commit 86ce98d08d

View File

@@ -20,8 +20,8 @@ import static org.springframework.data.r2dbc.query.Criteria.*;
import io.r2dbc.spi.Connection; import io.r2dbc.spi.Connection;
import io.r2dbc.spi.ConnectionFactory; import io.r2dbc.spi.ConnectionFactory;
import io.r2dbc.spi.Statement;
import io.r2dbc.spi.Result; import io.r2dbc.spi.Result;
import io.r2dbc.spi.Statement;
import reactor.core.CoreSubscriber; import reactor.core.CoreSubscriber;
import reactor.core.publisher.Flux; import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono; import reactor.core.publisher.Mono;
@@ -45,6 +45,7 @@ import org.springframework.data.r2dbc.support.R2dbcExceptionTranslator;
* Unit tests for {@link DefaultDatabaseClient}. * Unit tests for {@link DefaultDatabaseClient}.
* *
* @author Mark Paluch * @author Mark Paluch
* @author Ferdinand Jacobs
*/ */
@RunWith(MockitoJUnitRunner.class) @RunWith(MockitoJUnitRunner.class)
public class DefaultDatabaseClientUnitTests { public class DefaultDatabaseClientUnitTests {
@@ -330,7 +331,7 @@ public class DefaultDatabaseClientUnitTests {
.fetch() // .fetch() //
.rowsUpdated() // .rowsUpdated() //
.as(StepVerifier::create) // .as(StepVerifier::create) //
.expectNextCount(1) .expectNextCount(1) //
.verifyComplete(); .verifyComplete();
when(result.getRowsUpdated()).thenReturn(Mono.just(2)); when(result.getRowsUpdated()).thenReturn(Mono.just(2));
@@ -339,7 +340,16 @@ public class DefaultDatabaseClientUnitTests {
.fetch() // .fetch() //
.rowsUpdated() // .rowsUpdated() //
.as(StepVerifier::create) // .as(StepVerifier::create) //
.expectNextCount(1) .expectNextCount(1) //
.verifyComplete();
when(result.getRowsUpdated()).thenReturn(Flux.just(1, 2, 3));
databaseClient.execute("DROP TABLE tab;") //
.fetch() //
.rowsUpdated() //
.as(StepVerifier::create) //
.expectNextCount(1) //
.verifyComplete(); .verifyComplete();
} }
} }