启动Bean中的Arquillian Persistence-EntityManager找不到测试数据

Question

我试图测试一个使用arquillian-persistence-impl1.0.0.alpha6从数据库加载国家/地区的单例启动Bean。

如果我运行测试类并尝试访问通过@UsingDataSet提供的一个测试数据元组，那么一切工作都很好。

以下是Arquillian测试类的外观

@RunWith(Arquillian.class)
public class CountryDAOTest {

    @Inject
    CountryDAO dao;

    @Produces
    @PersistenceContext(unitName="test-pu")
    EntityManager em;

    @Inject
    UserTransaction ut;

    @Deployment
    public static JavaArchive createDeployment() {

        JavaArchive jar = ShrinkWrap.create(JavaArchive.class)
                    .addClass(CountryDAO.class)
                    .addClass(UserTransaction.class)
                    .addAsResource("test-persistence.xml", "META-INF/persistence.xml")
                    .addAsManifestResource("jbossas-ds.xml")
                    .addAsManifestResource(EmptyAsset.INSTANCE, "beans.xml");
                 System.out.println(jar.toString(true));
                 return jar;
    }

    @Test
    @UsingDataSet("db-data/countryDAOTest-country.yml")
    @ShouldMatchDataSet("db-data/countryDAOTest-country-verify.yml")
    public void getCountryFromDAO(){    

        TypedQuery<Country> query = em.createQuery("FROM Country", Country.class);
        List<Country> newCountriesList = query.getResultList();
        System.out.println("LOADED COUNTRIES IN TEST CLASS: "+newCountriesList.size());/*this works fine and returns 2 elements */

        List<Country> daoCountry = dao.getCountries();
        System.out.println("LOADED COUNTRIES FROM DAO: "+daoCountry.size()); /*this returns 0 elements */

    }
}

如果我试图通过我的单例启动程序CountryDAO调用数据库，它在测试数据库中找不到任何数据。

@Startup
@Singleton
public class CountryDAO {

    @Inject
    private EntityManager entityManager;
    private List<Country> countries;

    @PostConstruct
    public void init(){
        TypedQuery<Country> query = entityManager.createQuery("FROM Country", Country.class);
        List<Country> newCountriesList = query.getResultList();

        System.out.println("LOADED COUNTRIES ON POSTCONSTRUCT:"+newCountriesList.size()); /* returns 0 elements */

        countries = Collections.unmodifiableList(newCountriesList);
    }

    public synchronized List<Country> getCountries(){
        return countries;
    }
}

我还尝试在我的dao entityManager上使用@PersistenceContext注释，得到了相同的结果。

我使用jboss AS7.1.1作为测试容器。

知道为什么我的DAO entityManager找不到我的测试数据吗？

Answer 1

感谢John Ament的评论,这就是我解决问题的方法：

在执行测试之前，我将重新调用DAO bean的@PostConstruct注释的init（）方法。所以我又加了一句

@Before
public void before(){
    dao.init();
}

到我的CountryDAOTest Junit测试类。

具体来说，问题是在CountryDaoTest中注入CountryDAO（）时调用了@PostConstruct注释的init（）方法。

我的测试数据的插入发生在调用getCountryFromDAO（）之前，此时CountryDAO已经被注入到CountryDAOTest中并且已经调用了init（）。