Examples with JdbcConnectionAccess org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess used on opensource projects

Example 1 with JdbcConnectionAccess

use of org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess in project hibernate-orm by hibernate.

the class AbstractRegionAccessStrategyTest method mockedSession.

protected SharedSessionContractImplementor mockedSession() {
    SessionMock session = mock(SessionMock.class);
    when(session.isClosed()).thenReturn(false);
    when(session.getTimestamp()).thenReturn(TIME_SERVICE.wallClockTime());
    if (jtaPlatform == BatchModeJtaPlatform.class) {
        BatchModeTransactionCoordinator txCoord = new BatchModeTransactionCoordinator();
        when(session.getTransactionCoordinator()).thenReturn(txCoord);
        when(session.beginTransaction()).then(invocation -> {
            Transaction tx = txCoord.newTransaction();
            tx.begin();
            return tx;
        });
    } else if (jtaPlatform == null) {
        Connection connection = mock(Connection.class);
        JdbcConnectionAccess jdbcConnectionAccess = mock(JdbcConnectionAccess.class);
        try {
            when(jdbcConnectionAccess.obtainConnection()).thenReturn(connection);
        } catch (SQLException e) {
        // never thrown from mock
        }
        JdbcSessionOwner jdbcSessionOwner = mock(JdbcSessionOwner.class);
        when(jdbcSessionOwner.getJdbcConnectionAccess()).thenReturn(jdbcConnectionAccess);
        SqlExceptionHelper sqlExceptionHelper = mock(SqlExceptionHelper.class);
        JdbcServices jdbcServices = mock(JdbcServices.class);
        when(jdbcServices.getSqlExceptionHelper()).thenReturn(sqlExceptionHelper);
        ServiceRegistry serviceRegistry = mock(ServiceRegistry.class);
        when(serviceRegistry.getService(JdbcServices.class)).thenReturn(jdbcServices);
        JdbcSessionContext jdbcSessionContext = mock(JdbcSessionContext.class);
        when(jdbcSessionContext.getServiceRegistry()).thenReturn(serviceRegistry);
        when(jdbcSessionOwner.getJdbcSessionContext()).thenReturn(jdbcSessionContext);
        NonJtaTransactionCoordinator txOwner = mock(NonJtaTransactionCoordinator.class);
        when(txOwner.getResourceLocalTransaction()).thenReturn(new JdbcResourceTransactionMock());
        when(txOwner.getJdbcSessionOwner()).thenReturn(jdbcSessionOwner);
        when(txOwner.isActive()).thenReturn(true);
        TransactionCoordinator txCoord = JdbcResourceLocalTransactionCoordinatorBuilderImpl.INSTANCE.buildTransactionCoordinator(txOwner, null);
        when(session.getTransactionCoordinator()).thenReturn(txCoord);
        when(session.beginTransaction()).then(invocation -> {
            Transaction tx = new TransactionImpl(txCoord, session.getExceptionConverter());
            tx.begin();
            return tx;
        });
    } else {
        throw new IllegalStateException("Unknown JtaPlatform: " + jtaPlatform);
    }
    return session;
}

Example 2 with JdbcConnectionAccess

use of org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess in project hibernate-orm by hibernate.

the class JdbcCoordinatorTest method testConnectionClose.

@Test
public void testConnectionClose() throws NoSuchFieldException, IllegalAccessException, SQLException {
    Connection connection = Mockito.mock(Connection.class);
    JdbcSessionOwner sessionOwner = Mockito.mock(JdbcSessionOwner.class);
    JdbcConnectionAccess jdbcConnectionAccess = Mockito.mock(JdbcConnectionAccess.class);
    when(jdbcConnectionAccess.obtainConnection()).thenReturn(connection);
    when(jdbcConnectionAccess.supportsAggressiveRelease()).thenReturn(false);
    JdbcSessionContext sessionContext = Mockito.mock(JdbcSessionContext.class);
    when(sessionOwner.getJdbcSessionContext()).thenReturn(sessionContext);
    when(sessionOwner.getJdbcConnectionAccess()).thenReturn(jdbcConnectionAccess);
    ServiceRegistry serviceRegistry = Mockito.mock(ServiceRegistry.class);
    when(sessionContext.getServiceRegistry()).thenReturn(serviceRegistry);
    when(sessionContext.getPhysicalConnectionHandlingMode()).thenReturn(PhysicalConnectionHandlingMode.IMMEDIATE_ACQUISITION_AND_HOLD);
    JdbcObserver jdbcObserver = Mockito.mock(JdbcObserver.class);
    when(sessionContext.getObserver()).thenReturn(jdbcObserver);
    JdbcServices jdbcServices = Mockito.mock(JdbcServices.class);
    when(serviceRegistry.getService(eq(JdbcServices.class))).thenReturn(jdbcServices);
    SqlExceptionHelper sqlExceptionHelper = Mockito.mock(SqlExceptionHelper.class);
    when(jdbcServices.getSqlExceptionHelper()).thenReturn(sqlExceptionHelper);
    JdbcCoordinatorImpl jdbcCoordinator = new JdbcCoordinatorImpl(null, sessionOwner);
    Batch currentBatch = Mockito.mock(Batch.class);
    Field currentBatchField = JdbcCoordinatorImpl.class.getDeclaredField("currentBatch");
    currentBatchField.setAccessible(true);
    currentBatchField.set(jdbcCoordinator, currentBatch);
    doThrow(IllegalStateException.class).when(currentBatch).release();
    try {
        jdbcCoordinator.close();
        fail("Should throw IllegalStateException");
    } catch (Exception expected) {
        assertEquals(IllegalStateException.class, expected.getClass());
    }
    verify(jdbcConnectionAccess, times(1)).releaseConnection(same(connection));
}

Example 3 with JdbcConnectionAccess

use of org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess in project hibernate-orm by hibernate.

the class StoreProcedureRefCursorOutParameterByNameTest method createProcedures.

private void createProcedures(EntityManagerFactory emf) {
    final SessionFactoryImplementor sf = emf.unwrap(SessionFactoryImplementor.class);
    final JdbcConnectionAccess connectionAccess = sf.getServiceRegistry().getService(JdbcServices.class).getBootstrapJdbcConnectionAccess();
    final Connection conn;
    try {
        conn = connectionAccess.obtainConnection();
        conn.setAutoCommit(false);
        try {
            Statement statement = conn.createStatement();
            statement.execute("CREATE OR REPLACE PROCEDURE PROC_EXAMPLE ( " + "  USER_NAME_PARAM IN VARCHAR2, CURSOR_PARAM OUT SYS_REFCURSOR ) " + "AS " + "BEGIN " + "  OPEN CURSOR_PARAM FOR " + "  SELECT * FROM USERS WHERE NAME = USER_NAME_PARAM; " + "END PROC_EXAMPLE; ");
            try {
                statement.close();
            } catch (SQLException ignore) {
                fail();
            }
        } finally {
            try {
                conn.commit();
            } catch (SQLException e) {
                System.out.println("Unable to commit transaction afterQuery creating creating procedures");
                fail();
            }
            try {
                connectionAccess.releaseConnection(conn);
            } catch (SQLException ignore) {
                fail();
            }
        }
    } catch (SQLException e) {
        throw new RuntimeException("Unable to create stored procedures", e);
    }
}

Example 4 with JdbcConnectionAccess

use of org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess in project hibernate-orm by hibernate.

the class StoreProcedureRefCursorOutParameterByPositionTest method createProcedures.

private void createProcedures(EntityManagerFactory emf) {
    final SessionFactoryImplementor sf = emf.unwrap(SessionFactoryImplementor.class);
    final JdbcConnectionAccess connectionAccess = sf.getServiceRegistry().getService(JdbcServices.class).getBootstrapJdbcConnectionAccess();
    final Connection conn;
    try {
        conn = connectionAccess.obtainConnection();
        conn.setAutoCommit(false);
        try {
            Statement statement = conn.createStatement();
            statement.execute("CREATE OR REPLACE PROCEDURE PROC_EXAMPLE ( " + "  USER_NAME_PARAM IN VARCHAR2, CURSOR_PARAM OUT SYS_REFCURSOR ) " + "AS " + "BEGIN " + "  OPEN CURSOR_PARAM FOR " + "  SELECT * FROM USERS WHERE NAME = USER_NAME_PARAM; " + "END PROC_EXAMPLE; ");
            try {
                statement.close();
            } catch (SQLException ignore) {
                fail();
            }
        } finally {
            try {
                conn.commit();
            } catch (SQLException e) {
                System.out.println("Unable to commit transaction afterQuery creating creating procedures");
                fail();
            }
            try {
                connectionAccess.releaseConnection(conn);
            } catch (SQLException ignore) {
                fail();
            }
        }
    } catch (SQLException e) {
        throw new RuntimeException("Unable to create stored procedures", e);
    }
}

Example 5 with JdbcConnectionAccess

use of org.hibernate.engine.jdbc.connections.spi.JdbcConnectionAccess in project hibernate-orm by hibernate.

the class StoredProcedureNullParameterByNameTest method createProcedure.

private void createProcedure(EntityManagerFactory emf, String storedProc) {
    final SessionFactoryImplementor sf = emf.unwrap(SessionFactoryImplementor.class);
    final JdbcConnectionAccess connectionAccess = sf.getServiceRegistry().getService(JdbcServices.class).getBootstrapJdbcConnectionAccess();
    final Connection conn;
    try {
        conn = connectionAccess.obtainConnection();
        conn.setAutoCommit(false);
        try {
            Statement statement = conn.createStatement();
            statement.execute(storedProc);
            try {
                statement.close();
            } catch (SQLException ignore) {
                fail();
            }
        } finally {
            try {
                conn.commit();
            } catch (SQLException e) {
                System.out.println("Unable to commit transaction after creating creating procedures");
                fail();
            }
            try {
                connectionAccess.releaseConnection(conn);
            } catch (SQLException ignore) {
                fail();
            }
        }
    } catch (SQLException e) {
        throw new RuntimeException("Unable to create stored procedures", e);
    }
}