import com.codetroopers.play.elasticsearch.*;
import com.codetroopers.play.elasticsearch.jest.JestRichResult;
import indextype.Index1Type1;
import indextype.Index2Type1;
import org.apache.commons.lang3.time.DateUtils;
import org.elasticsearch.common.geo.GeoPoint;
import org.elasticsearch.common.unit.DistanceUnit;
import org.elasticsearch.index.query.FilterBuilders;
import org.elasticsearch.index.query.GeoDistanceFilterBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.junit.Test;
import play.libs.F;
import play.test.FakeApplication;
import java.util.*;
import java.util.concurrent.TimeUnit;
import static org.fest.assertions.Assertions.assertThat;
import static play.test.Helpers.fakeApplication;
import static play.test.Helpers.running;
public class ElasticsearchTestJava {
public FakeApplication esFakeApplication() {
Map<String, Object> additionalConfiguration = new HashMap<String, Object>();
additionalConfiguration.put("elasticsearch.local", true);
additionalConfiguration.put("elasticsearch.index.name", "index1,index2");
additionalConfiguration.put("elasticsearch.index.clazzs", "indextype.*");
additionalConfiguration.put("elasticsearch.index.show_request", true);
additionalConfiguration.put("elasticsearch.index.dropOnShutdown", true);
List<String> additionalPlugin = new ArrayList<String>();
additionalPlugin.add("com.codetroopers.play.elasticsearch.plugin.IndexPlugin");
return fakeApplication(additionalConfiguration, additionalPlugin);
}
@Test
public void checkIndexNames() {
running(esFakeApplication(), new Runnable() {
public void run() {
assertThat(IndexClient.config.indexNames.length).isEqualTo(2);
assertThat(IndexClient.config.indexNames[0]).isEqualTo("index1");
assertThat(IndexClient.config.indexNames[1]).isEqualTo("index2");
}
});
}
@Test
public void checkIndex1Create() {
running(esFakeApplication(), new Runnable() {
public void run() {
String name1 = "name1";
String category = "category";
Date date = createDate();
// init new Type1
Index1Type1 type1 = new Index1Type1();
type1.name = name1;
type1.category = category;
type1.dateCreate = date;
// indexing
JestRichResult index = type1.index();
// Read
final String id = index.getId();
Index1Type1 typeResult = Index1Type1.find.byId(id);
assertThat(typeResult).isNotNull();
assertThat(typeResult.name).isEqualTo(name1);
assertThat(typeResult.category).isEqualTo(category);
assertThat(typeResult.dateCreate).isEqualTo(date);
assertThat(typeResult.getIndexPath().index).isEqualTo("index1");
assertThat(typeResult.getIndexPath().type).isEqualTo("type1");
}
});
}
public Date createDate() {
return DateUtils.truncate(new Date(), Calendar.SECOND);
}
@Test
public void checkIndex2Create() {
running(esFakeApplication(), new Runnable() {
public void run() {
String name1 = "name1";
String category = "category";
Date date = createDate();
// init new Type1
Index2Type1 type1 = new Index2Type1();
type1.name = name1;
type1.category = category;
type1.dateCreate = date;
// indexing
JestRichResult index = type1.index();
// Read
final String id = index.getId();
Index2Type1 typeResult = Index2Type1.find.byId(id);
assertThat(typeResult).isNotNull();
assertThat(typeResult.name).isEqualTo(name1);
assertThat(typeResult.category).isEqualTo(category);
assertThat(typeResult.dateCreate).isEqualTo(date);
assertThat(typeResult.getIndexPath().index).isEqualTo("index2");
assertThat(typeResult.getIndexPath().type).isEqualTo("type1");
}
});
}
@Test
public void asynchronousIndex() {
running(esFakeApplication(), new Runnable() {
@Override
public void run() {
Index1Type1 index1Type1 = new Index1Type1("1", "name1", "category", createDate());
Index1Type1 index1Type1Bis = new Index1Type1("2", "name2", "category", createDate());
F.Promise<JestRichResult> promise1 = index1Type1.indexAsync();
F.Promise<JestRichResult> promise2 = index1Type1Bis.indexAsync();
F.Promise<List<JestRichResult>> promise = F.Promise.sequence(promise1, promise2);
List<JestRichResult> indexResponses = promise.get(10L, TimeUnit.SECONDS);
assertThat(indexResponses.size()).isEqualTo(2);
}
});
}
@Test
public void asynchronousIndexBulk() {
running(esFakeApplication(), new Runnable() {
@Override
public void run() {
Index1Type1 index1Type1 = new Index1Type1("1", "name1", "category", createDate());
Index1Type1 index1Type1Bis = new Index1Type1("2", "name2", "category", createDate());
Index1Type1 index1Type1Ter = new Index1Type1("3", "name3", "category", createDate());
F.Promise<JestRichResult> promise = IndexService.indexBulkAsync(
index1Type1.getIndexPath(),
Arrays.asList(index1Type1, index1Type1Bis, index1Type1Ter)
);
JestRichResult response = promise.get(10L, TimeUnit.SECONDS);
assertThat(response.getJsonObject().get("items").getAsJsonArray().size()).isEqualTo(3);
}
});
}
@Test
public void asynchronousDelete() {
running(esFakeApplication(), new Runnable() {
@Override
public void run() {
Index1Type1 index1Type1 = new Index1Type1("1", "name1", "category", createDate());
Index1Type1 index1Type1Bis = new Index1Type1("2", "name2", "category", createDate());
Index1Type1 index1Type1Ter = new Index1Type1("3", "name3", "category", createDate());
IndexService.indexBulk(index1Type1.getIndexPath(), Arrays.asList(index1Type1, index1Type1Bis, index1Type1Ter));
F.Promise<JestRichResult> promise1 = index1Type1.deleteAsync();
F.Promise<JestRichResult> promise2 = index1Type1Bis.deleteAsync();
F.Promise<JestRichResult> promise3 = index1Type1Ter.deleteAsync();
F.Promise<List<JestRichResult>> promise = F.Promise.sequence(promise1, promise2, promise3);
List<JestRichResult> deleteResponses = promise.get(10L, TimeUnit.SECONDS);
assertThat(deleteResponses.size()).isEqualTo(3);
}
});
}
@Test
public void asynchronousGet() {
running(esFakeApplication(), new Runnable() {
@Override
public void run() {
Index1Type1 index1Type1 = new Index1Type1("1", "name1", "category", createDate());
Index1Type1 index1Type1Bis = new Index1Type1("2", "name2", "category", createDate());
Index1Type1 index1Type1Ter = new Index1Type1("3", "name3", "category", createDate());
IndexService.indexBulk(index1Type1.getIndexPath(), Arrays.asList(index1Type1, index1Type1Bis, index1Type1Ter));
F.Promise<Index1Type1> promise1 = IndexService.getAsync(index1Type1.getIndexPath(), Index1Type1.class, "1");
F.Promise<Index1Type1> promise2 = IndexService.getAsync(index1Type1.getIndexPath(), Index1Type1.class, "2");
F.Promise<Index1Type1> promise3 = IndexService.getAsync(index1Type1.getIndexPath(), Index1Type1.class, "3");
F.Promise<List<Index1Type1>> promise = F.Promise.sequence(promise1, promise2, promise3);
List<Index1Type1> objects = promise.get(10L, TimeUnit.SECONDS);
assertThat(objects.size()).isEqualTo(3);
assertThat(objects.get(0)).isEqualTo(index1Type1);
assertThat(objects.get(1)).isEqualTo(index1Type1Bis);
assertThat(objects.get(2)).isEqualTo(index1Type1Ter);
}
});
}
@Test
public void asynchronousSearch() {
running(esFakeApplication(), new Runnable() {
@Override
public void run() {
Index1Type1 index1Type1 = new Index1Type1("1", "name1", "category", createDate());
index1Type1.index();
IndexService.refresh();
IndexQuery<Index1Type1> query = new IndexQuery<Index1Type1>(Index1Type1.class);
List<F.Promise<? extends IndexResults<Index1Type1>>> promises = new ArrayList<F.Promise<? extends IndexResults<Index1Type1>>>();
for (int i = 0; i < 10; i++) {
promises.add(query.fetchAsync(index1Type1.getIndexPath()));
}
F.Promise<List<IndexResults<Index1Type1>>> combinedPromise = F.Promise.sequence(promises);
List<IndexResults<Index1Type1>> indexResultsList = combinedPromise.get(10L, TimeUnit.SECONDS);
assertThat(indexResultsList.size()).isEqualTo(10);
for (IndexResults<Index1Type1> indexResults : indexResultsList) {
assertThat(indexResults.totalCount).isEqualTo(1);
}
}
});
}
@Test
public void update() {
running(esFakeApplication(), new Runnable() {
@Override
public void run() {
// Blocking
Index1Type1 index1Type1 = new Index1Type1("1", "name1", "category", createDate());
index1Type1.index();
Map<String, Object> fieldNewValues = new HashMap<>();
fieldNewValues.put("name", "new-name");
String updateScript = "ctx._source.name = name";
index1Type1.update(fieldNewValues, updateScript);
Index1Type1 index1Type11 = Index1Type1.find.byId("1");
assertThat(index1Type11.name).isEqualTo("new-name");
// Async
fieldNewValues.put("name","new-name-async");
F.Promise<JestRichResult> updateResponsePromise = index1Type1.updateAsync(fieldNewValues, updateScript);
updateResponsePromise.get(2L, TimeUnit.SECONDS);
index1Type11 = Index1Type1.find.byId("1");
assertThat(index1Type11.name).isEqualTo("new-name-async");
}
});
}
@Test
public void searchWithGeoFilter() {
running(esFakeApplication(), new Runnable() {
@Override
public void run() {
GeoPoint location = new GeoPoint(30.6943566,-88.0430541);
Index1Type1 index1Type1 = new Index1Type1("1", "name1", "category", createDate(), location);
index1Type1.index();
location = new GeoPoint(12.6943566,-10.0430541);
Index1Type1 index1Type2 = new Index1Type1("2", "name1", "category", createDate(), location);
index1Type2.index();
// refresh the index so the documents appear in the search results
IndexService.refresh();
assertThat(Index1Type1.find.byId("1").name).isEqualTo("name1");
assertThat(Index1Type1.find.byId("2").name).isEqualTo("name1");
IndexQuery<Index1Type1> query = Index1Type1.find.query();
query.setBuilder(QueryBuilders.queryString("name1"));
GeoDistanceFilterBuilder filter = FilterBuilders.geoDistanceFilter("type1.location")
.point(30, -88)
.distance(100, DistanceUnit.KILOMETERS);
F.Promise<IndexResults<Index1Type1>> indexResultsPromise = Index1Type1.find.searchAsync(query, filter);
IndexResults<Index1Type1> index1Type1IndexResults = indexResultsPromise.get(2L, TimeUnit.SECONDS);
assertThat(index1Type1IndexResults.totalCount).isEqualTo(1);
}
});
}
}