mindol1004/spring-data-mongodb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

DATAMONGO-586 - Some refactorings in the integration test.

Browse Source
This commit is contained in:
Oliver Gierke
2013-07-16 21:57:05 +02:00
parent aa23c579e8
commit 966f971bee
1 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/aggregation/AggregationTests.java
View File

@@ -31,20 +31,22 @@ import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.dao.DataAccessException;
import org.springframework.data.mongodb.core.CollectionCallback;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import com.mongodb.BasicDBObject;
import com.mongodb.CommandResult;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.MongoException;
import com.mongodb.WriteConcern;
import com.mongodb.util.JSON;
/**
@@ -55,17 +57,17 @@ import com.mongodb.util.JSON;
* @author Thomas Darimont
*/
@RunWith(SpringJUnit4ClassRunner.class)
@DirtiesContext
@ContextConfiguration("classpath:infrastructure.xml")
public class AggregationTests {
private static final String INPUT_COLLECTION = "aggregation_test_collection";
private static boolean initialized = false;
private static final Logger LOGGER = LoggerFactory.getLogger(AggregationTests.class);
@Autowired MongoTemplate mongoTemplate;
@Before
public void setUp() {
mongoTemplate.setWriteConcern(WriteConcern.SAFE); // safe due to DirtiesContext
cleanDb();
initSampleDataIfNecessary();
}
@@ -88,12 +90,17 @@ public class AggregationTests {
*/
private void initSampleDataIfNecessary() {
if (!mongoTemplate.collectionExists(ZipInfo.class)) {
mongoTemplate.dropCollection(ZipInfo.class);
if (!initialized) {
CommandResult result = mongoTemplate.executeCommand(new BasicDBObject("buildInfo", 1));
LOGGER.error(result.toString());
mongoTemplate.dropCollection(ZipInfo.class);
mongoTemplate.execute(ZipInfo.class, new CollectionCallback<Void>() {
@Override
public Void doInCollection(DBCollection collection) throws MongoException, DataAccessException {
Scanner scanner = null;
try {
scanner = new Scanner(new BufferedInputStream(new ClassPathResource("zips.json").getInputStream()));
@@ -111,6 +118,11 @@ public class AggregationTests {
return null;
}
});
long count = mongoTemplate.count(new Query(), ZipInfo.class);
assertThat(count, is(29467L));
initialized = true;
}
}