spring - 如何在 Spring Autowiring 之前让 DBUnit @DatabaseSetup 发生？

Question

考虑典型的 DBUnit Spring 测试(参见 https://github.com/springtestdbunit/spring-test-dbunit):

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {
"classpath:/META-INF/spring/applicationContext-database.xml",
"classpath:spring-*.xml"
})
@TestExecutionListeners({ DependencyInjectionTestExecutionListener.class,
    DirtiesContextTestExecutionListener.class,
    TransactionalTestExecutionListener.class,
    DbUnitTestExecutionListener.class })
@DatabaseSetup("/dbunit/data.xml")
public class UnitTest {

    @Autowired
    private UnitUnderTest uut;

    @Test
    public void shouldInitDB() {
         ...
    }
}

我已经证实，并且已经预料到， Autowiring 将在 DatabaseSetup 之前发生。
这必须发生，因为 DBUnit 依赖于应用程序上下文来提供配置的数据源。

问题是 UnitUnderTest bean 有一个 @PostConstruct，它从 DB 加载一些数据，但是，由于 Autowiring 发生在 DBunit 设置之前，数据在这个阶段将不可用。

关于如何以干净的方式解决这个问题的任何想法？

Answer 1

您可以使用 Spring 的 ResourceDatabasePopulator。

我想你可以使用这样的东西

@PostConstruct
public void myInMemryPopulator() {
final ResourceDatabasePopulator databasePopulator = new ResourceDatabasePopulator();

        try {

            Resource[] array = resourceResolver.getResources("classpath:/*.sql");       

            for (Resource resource : array) {

                databasePopulator.addScript(resource);
            }
            databasePopulator.populate(dataSource.getConnection());

        } catch (IOException | SQLException e) {
            LOGGER.error("Error in databasePopulator  {} ", e);
        }

    }