Question

我有一套使用Spring的JUnit运行器的系统测试，数据库配置如下所示：

<bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource">
    <property name="driverClassName" value="org.postgresql.Driver" />
    <property name="url" value="${clustercatalog.jdbc.url}" />
    <property name="username" value="${clustercatalog.jdbc.username}" />
    <property name="password" value="${clustercatalog.jdbc.password}" />
    <property name="initialSize" value="5" />
    <property name="maxActive" value="50" />
    <property name="poolPreparedStatements" value="true" />
    <property name="maxOpenPreparedStatements" value="100" />
</bean>

<bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">
    <property name="dataSource" ref="dataSource" />
    <property name="packagesToScan" value="com.xxx" />
    <property name="hibernateProperties">
        <props>
            <prop key="hibernate.dialect">org.hibernate.dialect.PostgreSQLDialect</prop>
            <prop key="hibernate.show_sql">${jdbc.show.sql}</prop>
            <prop key="hibernate.id.new_generator_mappings">true</prop>
        </props>
    </property>
    <property name="namingStrategy" ref="namingStrategy" />
</bean>

<bean id="transactionManager" class="org.springframework.orm.hibernate3.HibernateTransactionManager">
    <property name="sessionFactory" ref="sessionFactory" />
</bean>
<tx:annotation-driven />

测试用例有一个设置，我在其中运行一些bash脚本，这些脚本在先前完成的备份的底层PostgreSQL数据库上运行pg_restore。这是因为我需要在每个测试方法之前使数据库的相同状态相同。此恢复是在与@BeforeTransaction一起使用的方法中完成的。

测试类用

注释

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration({ "/systemTests-applicationContext.xml", "/applicationContext.xml" })
@TransactionConfiguration(defaultRollback = false)
@Transactional()

在我执行使用当前hibernate会话的代码的测试中，它没有看到已恢复的表。当我重新启动整个测试然后看到它们，但显然这是我想要的但是它证明了db很好但是当我做pg_restore时Spring / Hibernate丢失了。我得到SQLGrammarException，表格不存在。

我正在寻找一种手动重启与DB连接的方法。我怎样才能做到这一点？我应该在sessionFactory或一些Spring组件上以某种方式做到吗？

Answer 1

好的，我解决了这个问题。似乎Spring和Hibernate都没有意识到DB中发生的变化，如果DBCP池中有任何连接，它们会被重用，然后发生错误，因为模式在连接下面发生了变化。所以我认为在重置数据库时在测试期间戳的地方是提供给Hibernate SessionFactory的javax.sql.DataSource对象。 DBCP BasicDataSource没有此功能，因此我：

为dataSource bean创建了自己的包装器。
将bean标记为原型范围，所以每次我需要一个新的时候我都不需要自己构建它
向包装类添加了一个刷新方法，该方法为新的bean实例调用applicationContext（自动连接到包装类）
将包装bean注入Hibernate的sessionFactory
将包装器自动装配到测试类，并在测试时需要刷新方法

代码如下：

@Component
public class RefreshableDataSource implements DataSource {

  @Autowired
  DataSource dataSource;
  @Autowired
  ApplicationContext applicationContext;

  public void refresh() {
    dataSource = (DataSource) applicationContext.getBean("dataSource");
  }

  @Override
  public Connection getConnection() throws SQLException {
    return dataSource.getConnection();
  }

  ...other DataSource methods ...
}

的applicationContext.xml：

<bean id="dataSource" scope="prototype" class="org.apache.commons.dbcp.BasicDataSource">
    <property name="driverClassName" value="org.postgresql.Driver" />
            ......
</bean>

<bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">
    <property name="dataSource" ref="refreshableDataSource" />
    <property name="packagesToScan" value="com.xxx" />
    <property name="hibernateProperties">
        <props>
            <prop key="hibernate.dialect">org.hibernate.dialect.PostgreSQLDialect</prop>
            <prop key="hibernate.show_sql">${jdbc.show.sql}</prop>
            <prop key="hibernate.id.new_generator_mappings">true</prop>
        </props>
    </property>
    <property name="namingStrategy" ref="namingStrategy" />
</bean>

在Spring + Hibernate测试用例中重新加载数据库连接

1 个答案: