@@ -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();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user