diff --git a/src/test/java/org/elasticsearch/river/mongodb/RiverMongoDBTestAbstract.java b/src/test/java/org/elasticsearch/river/mongodb/RiverMongoDBTestAbstract.java index 50af0309..95dcfc8d 100644 --- a/src/test/java/org/elasticsearch/river/mongodb/RiverMongoDBTestAbstract.java +++ b/src/test/java/org/elasticsearch/river/mongodb/RiverMongoDBTestAbstract.java @@ -77,7 +77,7 @@ public abstract class RiverMongoDBTestAbstract { public static final String TEST_MONGODB_RIVER_WITH_SCRIPT_JSON = "/org/elasticsearch/river/mongodb/script/test-mongodb-river-with-script.json"; public static final String TEST_MONGODB_RIVER_EXCLUDE_FIELDS_JSON = "/org/elasticsearch/river/mongodb/simple/test-simple-mongodb-river-exclude-fields.json"; public static final String TEST_MONGODB_RIVER_INCLUDE_FIELDS_JSON = "/org/elasticsearch/river/mongodb/simple/test-simple-mongodb-river-include-fields.json"; - public static final String TEST_SIMPLE_MONGODB_DOCUMENT_JSON = "/org/elasticsearch/river/mongodb/script/test-simple-mongodb-document.json"; + public static final String TEST_SIMPLE_MONGODB_DOCUMENT_JSON = "/org/elasticsearch/river/mongodb/simple/test-simple-mongodb-document.json"; protected final ESLogger logger = Loggers.getLogger(getClass()); protected final static long wait = 2000; diff --git a/src/test/java/org/elasticsearch/river/mongodb/simple/RiverMongoDBTest.java b/src/test/java/org/elasticsearch/river/mongodb/simple/RiverMongoDBTest.java index 6efa62bf..6da3ba56 100644 --- a/src/test/java/org/elasticsearch/river/mongodb/simple/RiverMongoDBTest.java +++ b/src/test/java/org/elasticsearch/river/mongodb/simple/RiverMongoDBTest.java @@ -28,6 +28,9 @@ import org.elasticsearch.action.admin.indices.exists.indices.IndicesExistsRequest; import org.elasticsearch.action.admin.indices.exists.indices.IndicesExistsResponse; import org.elasticsearch.action.count.CountResponse; +import org.elasticsearch.action.search.SearchRequest; +import org.elasticsearch.action.search.SearchResponse; +import org.elasticsearch.index.query.QueryBuilders; import org.elasticsearch.river.mongodb.RiverMongoDBTestAbstract; import org.testng.Assert; import org.testng.annotations.AfterClass; @@ -104,12 +107,17 @@ public void simpleBSONObject() throws Throwable { .exists(new IndicesExistsRequest(getIndex())); assertThat(response.actionGet().isExists(), equalTo(true)); refreshIndex(); - CountResponse countResponse = getNode() + SearchRequest search = getNode().client() + .prepareSearch(getIndex()) + .setQuery(QueryBuilders.fieldQuery("name", "Richard")) + .request(); + SearchResponse searchResponse = getNode() .client() - .count(countRequest(getIndex()).query( - fieldQuery("name", "Richard"))).actionGet(); - logger.info("Document count: {}", countResponse.getCount()); - countResponse = getNode() + .search(search).actionGet(); + assertThat(searchResponse.getHits().getTotalHits(), equalTo(1l)); + String chinese = (String) searchResponse.getHits().getAt(0).getSource().get("chinese"); + assertThat(chinese, equalTo("中国菜很好吃。")); + CountResponse countResponse = getNode() .client() .count(countRequest(getIndex()) .query(fieldQuery("_id", id))).actionGet(); diff --git a/src/test/java/org/elasticsearch/river/mongodb/simple/test-simple-mongodb-document.json b/src/test/java/org/elasticsearch/river/mongodb/simple/test-simple-mongodb-document.json index 4df5b5d3..4e194e5f 100644 --- a/src/test/java/org/elasticsearch/river/mongodb/simple/test-simple-mongodb-document.json +++ b/src/test/java/org/elasticsearch/river/mongodb/simple/test-simple-mongodb-document.json @@ -1,4 +1,5 @@ { name: "Richard", + chinese: "中国菜很好吃。", score: 100 } \ No newline at end of file