如何在 junit 运行时测试期间使用多个数据源和 sessionFactory

Question

在junit运行时测试中如何使用多个datasouce和sessionFactory？

我使用 hibernateDaoTemplate 进行持久化，我的结构就在这里：

我只想从不同的数据库中获取产品，然后在 ProductStatServiceTest 运行时将它们持久化到不同的数据库。

提前致谢。

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"/dataSource.xml", "/hibernate.xml", "/applicationContext.xml"})
@TestExecutionListeners({DependencyInjectionTestExecutionListener.class})
@TransactionConfiguration(transactionManager = "txManager", defaultRollback = true)
public abstract class GenericTransactionalUnitTest extends   AbstractTransactionalJUnit4SpringContextTests {
  private Session session;

  protected abstract SessionFactory getSessionFactory();
  ...
}

@Transactional(readOnly = true)
abstract public class ProductStatUnitTest extends GenericTransactionalUnitTest {
  @Autowired
  @Qualifier(value = "productStatSessionFactory")
  private SessionFactory sessionFactory;

  @Resource(name = "productStatDS")
  public void setDataSource(DataSource dataSource) {
    super.setDataSource(dataSource);
  }

  @Override
  protected SessionFactory getSessionFactory() {
         return sessionFactory;
  }
}

public class ProductStatServiceTest extends ProductStatUnitTest {
  @Autowired
  private ProductStatService productStatService;  

  @Test
  public List<ProductStat> testInjection() throws ParseException {
     productStatService.findById(123456); 
  }
..
 }

Answer 1

2 数据库 => 2 个会话工厂和数据源。这目前在您的代码中是不可能的。

我假设您有 1 个ProductStatService，但您的要求是此服务将与 sessionfactoryA 一起用于“选择”，sessionfactoryB 用于“插入”。

因此，您需要在单元测试中创建（并注入）该服务的 2 个实例，当然这些实例不能限定为单例（假设您使用 Spring 进行 DI）。
你将不得不重写你的一些代码，实际的 sessionfactory 需要被注入到服务中，而不是由GenericTransactionalUnitTest.getSessionFactory() 检索。

更新

如果您不想创建 2 个 DAO，那么这也是一个选项：

public class ProductStatDAO extends AbstractHibernateDao {

@Autowired(required = false)
@Qualifier("sessionFactoryA")
private SessionFactory sessionFactoryA;

@Autowired(required = false)
@Qualifier("sessionFactoryB")
private SessionFactory sessionFactoryB;

//called by Spring (init-method)
public void init() {
    if (getSessionFactory() == null) {
        if (sessionFactoryA != null) {
            setSessionFactory(sessionFactoryA);
        } else if (sessionFactoryB != null) {
            setSessionFactory(sessionFactoryB);
        } 
        // else throw exception
    }
}

}

但是，您仍需要在您的服务中注入此 DAO 的 2 个不同实例（因此 scope='prototype'）。