Add failing test

Signed-off-by: Thomas Farr <[email protected]>
Xtansia · Aug 1, 2024 · 4d8ec04 · 4d8ec04
1 parent 3d8061e
commit 4d8ec04
Showing 1 changed file with 75 additions and 0 deletions.
diff --git a/java-client/src/test/java/org/opensearch/client/opensearch/core/SearchRequestTest.java b/java-client/src/test/java/org/opensearch/client/opensearch/core/SearchRequestTest.java
@@ -9,9 +9,12 @@
 package org.opensearch.client.opensearch.core;
 
 import java.util.Collections;
+import java.util.List;
 import org.junit.Test;
 import org.opensearch.client.json.JsonData;
 import org.opensearch.client.opensearch._types.FieldValue;
+import org.opensearch.client.opensearch.core.search.SourceConfig;
+import org.opensearch.client.opensearch.core.search.SourceFilter;
 import org.opensearch.client.opensearch.model.ModelTestCase;
 
 public class SearchRequestTest extends ModelTestCase {
@@ -57,4 +60,76 @@ public void toBuilder() {
 
         assertEquals(copied.index(), origin.index());
     }
+
+    @Test
+    public void canDeserializeSourceAsBoolean() {
+        String json = "{\"query\":{\"match_all\":{}},\"_source\":true,\"size\":1}";
+
+        SearchRequest searchRequest = fromJson(json, SearchRequest._DESERIALIZER);
+
+        SourceConfig _source = searchRequest.source();
+        assertNotNull(_source);
+        assertTrue(_source.isFetch());
+        assertTrue(_source.fetch());
+    }
+
+    @Test
+    public void canDeserializeSourceAsString() {
+        String json = "{\"query\":{\"match_all\":{}},\"_source\":\"_id\",\"size\":1}";
+
+        SearchRequest searchRequest = fromJson(json, SearchRequest._DESERIALIZER);
+
+        SourceConfig _source = searchRequest.source();
+        assertNotNull(_source);
+        assertTrue(_source.isFilter());
+
+        SourceFilter filter = _source.filter();
+        assertNotNull(filter);
+
+        List<String> includes = filter.includes();
+        assertEquals(1, includes.size());
+        assertEquals("_id", includes.get(0));
+
+        assertTrue(filter.excludes().isEmpty());
+    }
+
+    @Test
+    public void canDeserializeSourceAsArray() {
+        String json = "{\"query\":{\"match_all\":{}},\"_source\":[\"_id\"],\"size\":1}";
+
+        SearchRequest searchRequest = fromJson(json, SearchRequest._DESERIALIZER);
+
+        SourceConfig _source = searchRequest.source();
+        assertNotNull(_source);
+        assertTrue(_source.isFilter());
+
+        SourceFilter filter = _source.filter();
+        assertNotNull(filter);
+
+        List<String> includes = filter.includes();
+        assertEquals(1, includes.size());
+        assertEquals("_id", includes.get(0));
+
+        assertTrue(filter.excludes().isEmpty());
+    }
+
+    @Test
+    public void canDeserializeSourceAsObject() {
+        String json = "{\"query\":{\"match_all\":{}},\"_source\":{\"includes\":[\"_id\"]},\"size\":1}";
+
+        SearchRequest searchRequest = fromJson(json, SearchRequest._DESERIALIZER);
+
+        SourceConfig _source = searchRequest.source();
+        assertNotNull(_source);
+        assertTrue(_source.isFilter());
+
+        SourceFilter filter = _source.filter();
+        assertNotNull(filter);
+
+        List<String> includes = filter.includes();
+        assertEquals(1, includes.size());
+        assertEquals("_id", includes.get(0));
+
+        assertTrue(filter.excludes().isEmpty());
+    }
 }