Skip to content

Commit

Permalink
adapt ShapeQuery test to XyShapeQuery
Browse files Browse the repository at this point in the history
Signed-off-by: MESSAOUDI Khadidja <kmessaoudi@thefork.com>
  • Loading branch information
kmessaoudi committed Mar 18, 2024
1 parent e1cda72 commit d175588
Showing 1 changed file with 18 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -37,12 +37,14 @@
import org.opensearch.client.opensearch._types.ErrorCause;
import org.opensearch.client.opensearch._types.FieldValue;
import org.opensearch.client.opensearch._types.GeoLocation;
import org.opensearch.client.opensearch._types.GeoShapeRelation;
import org.opensearch.client.opensearch._types.SortOptions;
import org.opensearch.client.opensearch._types.SortOptionsBuilders;
import org.opensearch.client.opensearch._types.SortOrder;
import org.opensearch.client.opensearch._types.query_dsl.Query;
import org.opensearch.client.opensearch._types.query_dsl.TermQuery;
import org.opensearch.client.opensearch._types.query_dsl.WrapperQuery;
import org.opensearch.client.opensearch._types.query_dsl.XyShapeQuery;
import org.opensearch.client.util.MapBuilder;

public class BehaviorsTest extends ModelTestCase {
Expand All @@ -58,6 +60,22 @@ public void testSingleKeyDictionary() {
assertEquals("some-value", q.value().stringValue());
}

@Test
public void testAdditionalPropertyOnClass() {
XyShapeQuery q = new XyShapeQuery.Builder().queryName("query-name")
.field("field-name")
.xyShape(_0 -> _0.relation(GeoShapeRelation.Disjoint))
.ignoreUnmapped(true)
.build();

q = checkJsonRoundtrip(q, "{\"field-name\":{\"relation\":\"disjoint\"},\"_name\":\"query-name\",\"ignore_unmapped\":true}");

assertEquals("query-name", q.queryName());
assertTrue(q.ignoreUnmapped());
assertEquals(GeoShapeRelation.Disjoint, q.xyShape().relation());
System.out.println(toJson(q));
}

@Test
public void testWrapperQuery() {
WrapperQuery q = new WrapperQuery.Builder().query("encoded_query").build();
Expand Down

0 comments on commit d175588

Please sign in to comment.