【发布时间】:2011-06-14 21:03:30
【问题描述】:
我正在尝试对我正在编写的 Web 服务中使用的部分代码运行高级测试。在此过程中,代码将调用数据库(我使用 Spring 和 Hibernate 设置)。自动装配在测试用例类本身中工作正常,但由于运行测试而被调用的其他类没有被自动装配。
单元测试
@RunWith( SpringJUnit4ClassRunner.class )
@ContextConfiguration
public class ProviderTest
{
@Autowired
protected ApplicationContext applicationContext;
protected ProviderA provider;
@Before
public void setUp()
{
Transaction transaction = DbLogHelper.getInstance().getLog();
transaction.setCompanyName("Unit Test");
transaction.setCreateTime(Calendar.getInstance());
transaction.setPhoneNumber("Provider Test - N/A");
transaction.setQueryId("Unit Test");
transaction.setRequest("<NONE - UNIT TEST>");
transaction.setResponse("<NONE - UNIT TEST>");
LogJob job = new LogJob(transaction, DbLogHelper.getInstance.getDao());
job.run();
provider = new ProviderA(transaction);
}
@After
public void tearDown()
{
Transaction transaction = DbLogHelper.getInstance().getLog();
transaction.setResponseTime(Calendar.getInstance());
DbLogHelper.getInstance().writeLog(transaction);
DbLogHelper.getInstance().resetLog();
}
@Test
public void testProvider()
{
try
{
List<TAddress> results = provider.searchItem("SOME_ITEM", "12345");
Assert.assertNotNull(results);
}
catch (Exception e)
{
Assert.fail();
}
}
}
单例日志助手
public class DbLogHelper
{
private static DbLogHelper instance;
@Autowired
private LogDao dao;
private Executor executor = new ThreadPoolExecutor(3, 10, 10, TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>());
private ThreadLocal<Transaction> transactionRef = new ThreadLocal<Transaction>() {
@Override
protected Transaction initialValue()
{
return new Transaction();
}
};
public static DbLogHelper getInstance()
{
if (instance == null)
{
synchronized (DbLogHelper.class)
{
if (instance == null)
{
instance = new DbLogHelper();
}
}
}
return instance;
}
/**
* Gets the Transaction object attached to this thread. Generates a new one if it does not
* yet exist.
* @return Transaction
*/
public Transaction getLog()
{
return transactionRef.get();
}
/**
* Removes the current Transaction object from the thread map. The next getLog() call will
* generate a new one.
*/
public void resetLog()
{
transactionRef.remove();
}
/**
* Gets the DAO used to persist Transaction objects
*/
public LogDao getDao()
{
return dao;
}
/**
* Queues the passed Transaction object to be written to the database. This happens in a
* separate thread so the service is not waiting on database persistence before responding.
* @param transaction
*/
public void writeLog(Transaction transaction)
{
LogJob job = new LogJob(transaction, dao);
executor.execute(job);
}
}
记录 DAO
@Repository
public class LogDao
{
private static Logger log = LoggerFactory.getLogger(LogDao.class);
private EntityManager entityManager;
@PersistenceContext
public void setEntityManager(EntityManager entityManager) {
log.debug("Entity Manager set on LogDao");
this.entityManager = entityManager;
}
@Transactional
public void updateTransaction(Transaction log)
{
log = entityManager.merge(log);
}
}
ProviderTest-context.xml
<!-- database -->
<bean id="datasource" class="org.springframework.jdbc.datasource.DriverManagerDataSource">
<property name="driverClassName" value="com.mysql.jdbc.Driver" />
<!-- uncomment to see sql
<property name="driverClassName" value="com.p6spy.engine.spy.P6SpyDriver" />
-->
<property name="url" value="jdbc:mysql://localhost" />
<property name="username" value="root" />
<property name="password" value="" />
</bean>
<!-- LOCAL entity manager factory -->
<bean id="entityManagerFactory" class="org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean">
<!-- <property name="persistenceUnitManager" ref="persistenceUnitManager"/> -->
<property name="persistenceUnitName" value="domainPersistenceUnit" />
<property name="dataSource" ref="datasource" />
<property name="jpaVendorAdapter">
<bean class="org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter">
<property name="showSql" value="false" />
<!-- <property name="generateDdl" value="true" /> -->
<property name="databasePlatform" value="org.hibernate.dialect.MySQLDialect" />
</bean>
</property>
</bean>
<bean class="org.springframework.dao.annotation.PersistenceExceptionTranslationPostProcessor"/>
<bean class="org.springframework.orm.jpa.support.PersistenceAnnotationBeanPostProcessor" />
<bean id="transactionManager" class="org.springframework.orm.jpa.JpaTransactionManager">
<property name="entityManagerFactory" ref="entityManagerFactory" />
<property name="dataSource" ref="datasource" />
</bean>
<context:annotation-config/>
<context:component-scan base-package="com.company.application.webservice"/>
DbLogHelper 中的自动装配 DAO 总是以 null 结束。但是,如果我直接在单元测试中自动装配它,它就会被注入到实体管理器中。当然,我担心的是,如果它在测试中不起作用,那么在正确部署时它也不会起作用。即使我不必担心,我仍然想测试它。
【问题讨论】:
标签: java hibernate spring dependency-injection junit4