1 files changed, 182 insertions, 0 deletions
diff --git a/config-model/src/test/java/com/yahoo/schema/derived/SummaryTestCase.java b/config-model/src/test/java/com/yahoo/schema/derived/SummaryTestCase.java
new file mode 100644
index 00000000000..341d3ef7d43
--- /dev/null
+++ b/config-model/src/test/java/com/yahoo/schema/derived/SummaryTestCase.java
@@ -0,0 +1,182 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.schema.derived;
+
+import com.yahoo.config.model.application.provider.BaseDeployLogger;
+import com.yahoo.schema.Schema;
+import com.yahoo.schema.ApplicationBuilder;
+import com.yahoo.schema.AbstractSchemaTestCase;
+import com.yahoo.schema.parser.ParseException;
+import com.yahoo.vespa.config.search.SummaryConfig;
+import org.junit.Test;
+
+import java.io.IOException;
+import java.util.Iterator;
+
+import static com.yahoo.config.model.test.TestUtil.joinLines;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * Tests summary extraction
+ *
+ * @author bratseth
+ */
+public class SummaryTestCase extends AbstractSchemaTestCase {
+
+    @Test
+    public void deriveRawAsBase64() throws ParseException {
+        String sd = joinLines(
+                "schema s {",
+                "  raw-as-base64-in-summary",
+                "  document s {",
+                "      field raw_field type raw {",
+                "          indexing: summary",
+                "      }",
+                "  }",
+                "}");
+        Schema schema = ApplicationBuilder.createFromString(sd).getSchema();
+        SummaryClass summary = new SummaryClass(schema, schema.getSummary("default"), new BaseDeployLogger());
+        assertEquals(SummaryClassField.Type.RAW, summary.fields().get("raw_field").getType());
+    }
+
+    @Test
+    public void deriveRawAsLegacy() throws ParseException {
+        String sd = joinLines(
+                "schema s {",
+                "  document s {",
+                "      field raw_field type raw {",
+                "          indexing: summary",
+                "      }",
+                "  }",
+                "}");
+        Schema schema = ApplicationBuilder.createFromString(sd).getSchema();
+        SummaryClass summary = new SummaryClass(schema, schema.getSummary("default"), new BaseDeployLogger());
+        assertEquals(SummaryClassField.Type.DATA, summary.fields().get("raw_field").getType());
+    }
+
+    @Test
+    public void testDeriving() throws IOException, ParseException {
+        Schema schema = ApplicationBuilder.buildFromFile("src/test/examples/simple.sd");
+        SummaryClass summary = new SummaryClass(schema, schema.getSummary("default"), new BaseDeployLogger());
+        assertEquals("default", summary.getName());
+
+        Iterator<SummaryClassField> fields = summary.fields().values().iterator();
+
+        SummaryClassField field;
+
+        assertEquals(13, summary.fields().size());
+
+        field = fields.next();
+        assertEquals("exactemento", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("exact", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("title", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("description", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("dyndesc", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("longdesc", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("longstat", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("dynlong", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("dyndesc2", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+
+        field = fields.next();
+        assertEquals("measurement", field.getName());
+        assertEquals(SummaryClassField.Type.INTEGER, field.getType());
+
+        field = fields.next();
+        assertEquals("rankfeatures", field.getName());
+        assertEquals(SummaryClassField.Type.FEATUREDATA, field.getType());
+
+        field = fields.next();
+        assertEquals("summaryfeatures", field.getName());
+        assertEquals(SummaryClassField.Type.FEATUREDATA, field.getType());
+
+        field = fields.next();
+        assertEquals("documentid", field.getName());
+        assertEquals(SummaryClassField.Type.LONGSTRING, field.getType());
+    }
+
+    @Test
+    public void reference_fields_can_be_part_of_summary_classes() throws ParseException {
+        Schema adSchema = buildCampaignAdModel();
+
+        SummaryClass defaultClass = new SummaryClass(adSchema, adSchema.getSummary("default"), new BaseDeployLogger());
+        assertEquals(SummaryClassField.Type.LONGSTRING, defaultClass.fields().get("campaign_ref").getType());
+        assertEquals(SummaryClassField.Type.LONGSTRING, defaultClass.fields().get("other_campaign_ref").getType());
+
+        SummaryClass myClass = new SummaryClass(adSchema, adSchema.getSummary("my_summary"), new BaseDeployLogger());
+        assertNull(myClass.fields().get("campaign_ref"));
+        assertEquals(SummaryClassField.Type.LONGSTRING, myClass.fields().get("other_campaign_ref").getType());
+    }
+
+    private static Schema buildCampaignAdModel() throws ParseException {
+        ApplicationBuilder builder = new ApplicationBuilder();
+        builder.addSchema("search campaign { document campaign {} }");
+        builder.addSchema(joinLines("search ad {",
+                                    "  document ad {",
+                                    "    field campaign_ref type reference<campaign> {",
+                                    "      indexing: summary | attribute",
+                                    "    }",
+                                    "    field other_campaign_ref type reference<campaign> {",
+                                    "      indexing: summary | attribute",
+                                    "    }",
+                                    "  }",
+                                    "  document-summary my_summary {",
+                                    "    summary other_campaign_ref type reference<campaign> {}",
+                                    "  }",
+                                    "}"));
+        builder.build(true);
+        return builder.getSchema("ad");
+    }
+
+    @Test
+    public void omit_summary_features_specified_for_document_summary() throws ParseException {
+        String sd = joinLines(
+                "schema test {",
+                "  document test {",
+                "    field foo type string { indexing: summary }",
+                "  }",
+                "  document-summary bar {",
+                "    summary foo type string {}",
+                "    omit-summary-features",
+                "  }",
+                "  document-summary baz {",
+                "    summary foo type string {}",
+                "  }",
+                "}");
+        var search = ApplicationBuilder.createFromString(sd).getSchema();
+        assertOmitSummaryFeatures(true, search, "bar");
+        assertOmitSummaryFeatures(false, search, "baz");
+    }
+
+    private void assertOmitSummaryFeatures(boolean expected, Schema schema, String summaryName) {
+        var summary = new SummaryClass(schema, schema.getSummary(summaryName), new BaseDeployLogger());
+        var config = new SummaryConfig.Classes(summary.getSummaryClassConfig());
+        assertEquals(expected, config.omitsummaryfeatures());
+    }
+
+}