Examples with PreparedStatementSpyConnectionProvider org.hibernate.testing.orm.jdbc.PreparedStatementSpyConnectionProvider used on opensource projects

Example 1 with PreparedStatementSpyConnectionProvider

use of org.hibernate.testing.orm.jdbc.PreparedStatementSpyConnectionProvider in project hibernate-orm by hibernate.

the class MySQLDropConstraintThrowsExceptionTest method testEnumTypeInterpretation.

@Test
public void testEnumTypeInterpretation() {
    final PreparedStatementSpyConnectionProvider connectionProvider = new PreparedStatementSpyConnectionProvider(false, false);
    final StandardServiceRegistry serviceRegistry = new StandardServiceRegistryBuilder().enableAutoClose().applySetting(AvailableSettings.HBM2DDL_AUTO, "update").applySetting(AvailableSettings.CONNECTION_PROVIDER, connectionProvider).build();
    SessionFactory sessionFactory = null;
    try {
        final Metadata metadata = new MetadataSources(serviceRegistry).addAnnotatedClass(Customer.class).buildMetadata();
        sessionFactory = metadata.buildSessionFactory();
        List<String> alterStatements = connectionProvider.getExecuteStatements().stream().filter(sql -> sql.toLowerCase().contains("alter ")).map(String::trim).collect(Collectors.toList());
        if (metadata.getDatabase().getDialect() instanceof MariaDBDialect) {
            assertTrue(alterStatements.get(0).matches("alter table if exists CUSTOMER\\s+drop index .*?"));
            assertTrue(alterStatements.get(1).matches("alter table if exists CUSTOMER\\s+add constraint .*? unique \\(CUSTOMER_ID\\)"));
        } else {
            assertTrue(alterStatements.get(0).matches("alter table CUSTOMER\\s+drop index .*?"));
            assertTrue(alterStatements.get(1).matches("alter table CUSTOMER\\s+add constraint .*? unique \\(CUSTOMER_ID\\)"));
        }
    } finally {
        if (sessionFactory != null) {
            sessionFactory.close();
        }
        StandardServiceRegistryBuilder.destroy(serviceRegistry);
    }
}