Spring/spring-data-relational
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#243 - Add unit test for rowsUpdated.

Browse Source 
rowsUpdated should always emit a single value.
This commit is contained in:
Ferdinand Jacobs
2019-12-02 02:17:56 +01:00
committed by Mark Paluch
parent 887f0ae289
commit 38bbd4838e
1 changed files with 32 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
src/test/java/org/springframework/data/r2dbc/core/DefaultDatabaseClientUnitTests.java
View File

@@ -21,6 +21,7 @@ import static org.springframework.data.r2dbc.query.Criteria.*;
import io.r2dbc.spi.Connection;
import io.r2dbc.spi.ConnectionFactory;
import io.r2dbc.spi.Statement;
import io.r2dbc.spi.Result;
import reactor.core.CoreSubscriber;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
@@ -310,4 +311,35 @@ public class DefaultDatabaseClientUnitTests {
verify(statement).bind(1, (Object) 1);
verify(statement).bind(2, (Object) 2);
}
@Test // gh-243
public void rowsUpdatedShouldEmitSingleValue() {
Statement statement = mock(Statement.class);
when(connection.createStatement("DROP TABLE tab;")).thenReturn(statement);
Result result = mock(Result.class);
doReturn(Flux.just(result)).when(statement).execute();
DefaultDatabaseClient databaseClient = (DefaultDatabaseClient) DatabaseClient.builder()
.connectionFactory(connectionFactory)
.dataAccessStrategy(new DefaultReactiveDataAccessStrategy(PostgresDialect.INSTANCE)).build();
when(result.getRowsUpdated()).thenReturn(Mono.empty());
databaseClient.execute("DROP TABLE tab;") //
.fetch() //
.rowsUpdated() //
.as(StepVerifier::create) //
.expectNextCount(1)
.verifyComplete();
when(result.getRowsUpdated()).thenReturn(Mono.just(2));
databaseClient.execute("DROP TABLE tab;") //
.fetch() //
.rowsUpdated() //
.as(StepVerifier::create) //
.expectNextCount(1)
.verifyComplete();
}
}