在Arquillian测试期间清理数据库

时间:2014-06-30 20:51:21

标签: java mongodb testing morphia jboss-arquillian

我目前正在为我编写的一些REST-full服务编写测试。我正在测试的服务是用Java编写的,并使用MongoDb / Morphia。测试调用服务,其中一些服务又编写一个测试集合。我需要在测试后清理并删除我注入的数据。最好的方法是什么?

以下是我的一个简单服务的示例:

package org.haib.myerslab.services;

@Path("/database")
public class DatabaseService {
    @Inject
    private Datastore ds;

    @Path("/genre/")
    @POST
    @Produces("application/json")
    public GenreDTO postFromGenreDTO(@Context UriInfo uri, GenreDTO form) throws ParseException {
        Genre myNewGenre = DtoToDomainMapper.gerneFromGenreDTO(form);
        myNewGenre.setId(form.getId());
        ds.save(myNewGenre);
        return new GenreDTO(myNewGenre);
    }
}       

这是我的Arquillian测试的一个例子:

@RunWith(Arquillian.class)
public class GeneTest {
    private static String myId = "myGenreId";
    private static String myGenre = "myGenre";
    private static String myGenreInfo = "myGenreInfo";

    @Deployment
    public static WebArchive getDeployment() {
        return TestHelper.getDeployment();
    }

    @Test
    @RunAsClient
    @InSequence(1)
    public void canPostGenre(@ArquillianResource URL baseURL) throws Exception {
        GenreDTO newGenre = new GenreDTO();
        newGenre.setGenre(myGenre);
        newGenre.setGenreInfo(myGenreInfo);
        newGenre.setId(myId);

        String url = baseURL.toURI().resolve("/database/genre/").toString();
        JsonNode rootNode = TestHelper.postUrl(url, newGene);

        assertEquals(myGenre,     rootNode.get("genre").asText());
        assertEquals(myGenreInfo, rootNode.get("genreInfo").asText());
        assertEquals(myId,        rootNode.get("id").asText());
    } 
}

getDeployment函数如下所示:

public static WebArchive getDeployment() {
    File[] depend = Maven.resolver().loadPomFromFile("pom.xml").importRuntimeDependencies().resolve().withTransitivity().asFile();
    WebArchive war = ShrinkWrap.create(WebArchive.class).addClass(TestHelper.class)
        .addClass(Genre.class).addClass(Application.class).addPackage("org/haib/myerslab")
        .addPackage("org/haib/myerslab/database").addPackage("org/haib/myerslab/genre")
        .addPackage("org/haib/myerslab/dto").addPackage("org/haib/myerslab/dto/genre")
        .addAsLibraries(depend).addAsWebInfResource("jboss-deployment-structure.xml")
        .addAsWebInfResource(EmptyAsset.INSTANCE, "beans.xml").setWebXML("test-web.xml");
    return war;
}

所以我丢失的地方是,在@After中注入数据库的最佳方法是什么,并清除我发布到它的类型类,以便我的下一个测试没有它。 我该怎么做?还有另一种方式吗?

1 个答案:

答案 0 :(得分:2)

看看nosql-unit。它提供了注释和规则,可以帮助您播种数据集,比较期望和清理MongoDB。

要在执行测试之前让MongoDB进入原始状态,您只需使用以下注释和`CLEAN_INSERT':

@UsingDataSet(locations="my_data_set.json", loadStrategy=LoadStrategyEnum.CLEAN_INSERT) 
public void canPostGenre() { ...}

如果你需要围绕集成测试生命周期的行为使MongoDB更强大,你也可以根据nosql-unit的想法推出自己的行为。另请务必查看Junit Rules