BAEL-53 - minor formatting changes

This commit is contained in:
Slavisa Baeldung 2016-09-16 15:47:46 +02:00
parent 5f159cab6c
commit 7a0deaceb4

View File

@ -26,7 +26,6 @@ import com.alibaba.fastjson.JSON;
public class ElasticSearchUnitTests { public class ElasticSearchUnitTests {
private List<Person> listOfPersons = new ArrayList<Person>(); private List<Person> listOfPersons = new ArrayList<Person>();
String jsonString = null; String jsonString = null;
Client client = null; Client client = null;
@Before @Before
@ -44,71 +43,68 @@ public class ElasticSearchUnitTests {
public void givenJsonString_whenJavaObject_thenIndexDocument() { public void givenJsonString_whenJavaObject_thenIndexDocument() {
String jsonObject = "{\"age\":20,\"dateOfBirth\":1471466076564,\"fullName\":\"John Doe\"}"; String jsonObject = "{\"age\":20,\"dateOfBirth\":1471466076564,\"fullName\":\"John Doe\"}";
IndexResponse response = client.prepareIndex("people", "Doe") IndexResponse response = client.prepareIndex("people", "Doe")
.setSource(jsonObject).get(); .setSource(jsonObject).get();
String id = response.getId();
String index = response.getIndex(); String index = response.getIndex();
String type = response.getType(); String type = response.getType();
assertTrue(response.isCreated()); assertTrue(response.isCreated());
assertEquals(index, "people"); assertEquals(index, "people");
assertEquals(type, "Doe"); assertEquals(type, "Doe");
} }
@Test @Test
public void givenDocumentId_whenJavaObject_thenDeleteDocument() { public void givenDocumentId_whenJavaObject_thenDeleteDocument() {
String jsonObject = "{\"age\":10,\"dateOfBirth\":1471455886564,\"fullName\":\"Johan Doe\"}"; String jsonObject = "{\"age\":10,\"dateOfBirth\":1471455886564,\"fullName\":\"Johan Doe\"}";
IndexResponse response = client.prepareIndex("people", "Doe") IndexResponse response = client.prepareIndex("people", "Doe")
.setSource(jsonObject).get(); .setSource(jsonObject).get();
String id = response.getId(); String id = response.getId();
DeleteResponse deleteResponse = client.prepareDelete("people", "Doe", id).get(); DeleteResponse deleteResponse = client.prepareDelete("people", "Doe", id).get();
assertTrue(deleteResponse.isFound()); assertTrue(deleteResponse.isFound());
} }
@Test @Test
public void givenSearchRequest_whenMatchAll_thenReturnAllResults() { public void givenSearchRequest_whenMatchAll_thenReturnAllResults() {
SearchResponse response = client.prepareSearch().execute().actionGet(); SearchResponse response = client.prepareSearch().execute().actionGet();
SearchHit[] searchHits = response.getHits().getHits(); SearchHit[] searchHits = response.getHits().getHits();
List<Person> results = new ArrayList<Person>(); List<Person> results = new ArrayList<Person>();
for(SearchHit hit : searchHits){ for (SearchHit hit : searchHits) {
String sourceAsString = hit.getSourceAsString(); String sourceAsString = hit.getSourceAsString();
Person person = JSON.parseObject(sourceAsString, Person.class); Person person = JSON.parseObject(sourceAsString, Person.class);
results.add(person); results.add(person);
} }
} }
@Test @Test
public void givenSearchParamters_thenReturnResults() { public void givenSearchParamters_thenReturnResults() {
boolean isExecutedSuccessfully = true; boolean isExecutedSuccessfully = true;
SearchResponse response = client.prepareSearch() SearchResponse response = client.prepareSearch()
.setTypes() .setTypes()
.setSearchType(SearchType.DFS_QUERY_THEN_FETCH) .setSearchType(SearchType.DFS_QUERY_THEN_FETCH)
.setPostFilter(QueryBuilders.rangeQuery("age").from(5).to(15)) .setPostFilter(QueryBuilders.rangeQuery("age").from(5).to(15))
.setFrom(0).setSize(60).setExplain(true) .setFrom(0).setSize(60).setExplain(true)
.execute() .execute()
.actionGet(); .actionGet();
SearchResponse response2 = client.prepareSearch() SearchResponse response2 = client.prepareSearch()
.setTypes() .setTypes()
.setSearchType(SearchType.DFS_QUERY_THEN_FETCH) .setSearchType(SearchType.DFS_QUERY_THEN_FETCH)
.setPostFilter(QueryBuilders .setPostFilter(QueryBuilders.simpleQueryStringQuery("+John -Doe OR Janette"))
.simpleQueryStringQuery("+John -Doe OR Janette")) .setFrom(0).setSize(60).setExplain(true)
.setFrom(0).setSize(60).setExplain(true) .execute()
.execute() .actionGet();
.actionGet();
SearchResponse response3 = client.prepareSearch() SearchResponse response3 = client.prepareSearch()
.setTypes() .setTypes()
.setSearchType(SearchType.DFS_QUERY_THEN_FETCH) .setSearchType(SearchType.DFS_QUERY_THEN_FETCH)
.setPostFilter(QueryBuilders.matchQuery( .setPostFilter(QueryBuilders.matchQuery("John", "Name*"))
"John", "Name*")) .setFrom(0).setSize(60).setExplain(true)
.setFrom(0).setSize(60).setExplain(true) .execute()
.execute() .actionGet();
.actionGet();
try { try {
response2.getHits(); response2.getHits();
response3.getHits(); response3.getHits();
SearchHit[] searchHits = response.getHits().getHits(); SearchHit[] searchHits = response.getHits().getHits();
List<Person> results = new ArrayList<Person>(); List<Person> results = new ArrayList<Person>();
for(SearchHit hit : searchHits){ for (SearchHit hit : searchHits) {
String sourceAsString = hit.getSourceAsString(); String sourceAsString = hit.getSourceAsString();
Person person = JSON.parseObject(sourceAsString, Person.class); Person person = JSON.parseObject(sourceAsString, Person.class);
results.add(person); results.add(person);
@ -118,17 +114,17 @@ public class ElasticSearchUnitTests {
} }
assertTrue(isExecutedSuccessfully); assertTrue(isExecutedSuccessfully);
} }
@Test @Test
public void givenContentBuilder_whenHelpers_thanIndexJson() throws IOException { public void givenContentBuilder_whenHelpers_thanIndexJson() throws IOException {
XContentBuilder builder = XContentFactory.jsonBuilder() XContentBuilder builder = XContentFactory.jsonBuilder()
.startObject() .startObject()
.field("fullName", "Test") .field("fullName", "Test")
.field("salary", "11500") .field("salary", "11500")
.field("age", "10") .field("age", "10")
.endObject(); .endObject();
IndexResponse response = client.prepareIndex("people", "Doe") IndexResponse response = client.prepareIndex("people", "Doe")
.setSource(builder).get(); .setSource(builder).get();
assertTrue(response.isCreated()); assertTrue(response.isCreated());
} }
} }