Examples with Wrapped io.r2dbc.spi.Wrapped used on opensource projects

Example 1 with Wrapped

use of io.r2dbc.spi.Wrapped in project spring-framework by spring-projects.

the class TransactionAwareConnectionFactoryProxyUnitTests method shouldEmitBoundConnection.

@Test
void shouldEmitBoundConnection() {
    when(connectionMock1.beginTransaction()).thenReturn(Mono.empty());
    when(connectionMock1.commitTransaction()).thenReturn(Mono.empty());
    when(connectionMock1.close()).thenReturn(Mono.empty());
    TransactionalOperator rxtx = TransactionalOperator.create(tm);
    AtomicReference<Connection> transactionalConnection = new AtomicReference<>();
    TransactionAwareConnectionFactoryProxy proxyCf = new TransactionAwareConnectionFactoryProxy(connectionFactoryMock);
    ConnectionFactoryUtils.getConnection(connectionFactoryMock).doOnNext(transactionalConnection::set).flatMap(connection -> proxyCf.create().doOnNext(wrappedConnection -> assertThat(((Wrapped<?>) wrappedConnection).unwrap()).isSameAs(connection))).as(rxtx::transactional).flatMapMany(Connection::close).as(StepVerifier::create).verifyComplete();
    verifyNoInteractions(connectionMock2);
    verifyNoInteractions(connectionMock3);
    verify(connectionFactoryMock, times(1)).create();
}

Example 2 with Wrapped

use of io.r2dbc.spi.Wrapped in project spring-framework by spring-projects.

the class SingleConnectionFactoryUnitTests method shouldSuppressClose.

@Test
@SuppressWarnings("rawtypes")
void shouldSuppressClose() {
    SingleConnectionFactory factory = new SingleConnectionFactory("r2dbc:h2:mem:///foo", true);
    Connection connection = factory.create().block();
    StepVerifier.create(connection.close()).verifyComplete();
    assertThat(connection).isInstanceOf(Wrapped.class);
    assertThat(((Wrapped) connection).unwrap()).isInstanceOf(H2Connection.class);
    StepVerifier.create(connection.setTransactionIsolationLevel(IsolationLevel.READ_COMMITTED)).verifyComplete();
    factory.destroy();
}