1 files changed, 92 insertions, 0 deletions
diff --git a/config-model/src/test/java/com/yahoo/schema/processing/ReferenceFieldTestCase.java b/config-model/src/test/java/com/yahoo/schema/processing/ReferenceFieldTestCase.java
new file mode 100644
index 00000000000..57b4d928a52
--- /dev/null
+++ b/config-model/src/test/java/com/yahoo/schema/processing/ReferenceFieldTestCase.java
@@ -0,0 +1,92 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.schema.processing;
+
+import com.yahoo.config.model.deploy.TestProperties;
+import com.yahoo.document.DataType;
+import com.yahoo.document.Field;
+import com.yahoo.documentmodel.NewDocumentReferenceDataType;
+import com.yahoo.schema.Schema;
+import com.yahoo.schema.ApplicationBuilder;
+import com.yahoo.schema.document.SDDocumentType;
+import com.yahoo.schema.parser.ParseException;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author bjorncs
+ */
+public class ReferenceFieldTestCase {
+
+    @SuppressWarnings("deprecation")
+    @Rule
+    public final ExpectedException exceptionRule = ExpectedException.none();
+
+    @Test
+    public void reference_fields_are_parsed_from_search_definition() throws ParseException {
+        ApplicationBuilder builder = new ApplicationBuilder();
+        String campaignSdContent =
+                "schema campaign {\n" +
+                "  document campaign {\n" +
+                "  }\n" +
+                "}";
+        String salespersonSdContent =
+                "schema salesperson {\n" +
+                "  document salesperson {\n" +
+                "  }\n" +
+                "}";
+        String adSdContent =
+                "schema ad {\n" +
+                "  document ad {\n" +
+                "    field campaign_ref type reference<campaign> { indexing: attribute }\n" +
+                "    field salesperson_ref type reference<salesperson> { indexing: attribute }\n" +
+                "  }\n" +
+                "}";
+        builder.addSchema(campaignSdContent);
+        builder.addSchema(salespersonSdContent);
+        builder.addSchema(adSdContent);
+        builder.build(true);
+        Schema schema = builder.getSchema("ad");
+        assertSearchContainsReferenceField("campaign_ref", "campaign", schema.getDocument());
+        assertSearchContainsReferenceField("salesperson_ref", "salesperson", schema.getDocument());
+    }
+
+    @Test
+    public void cyclic_document_dependencies_are_detected() throws ParseException {
+        var builder = new ApplicationBuilder(new TestProperties());
+        String campaignSdContent =
+                "schema campaign {\n" +
+                        "  document campaign {\n" +
+                        "    field ad_ref type reference<ad> { indexing: attribute }\n" +
+                        "  }\n" +
+                        "}";
+        String adSdContent =
+                "schema ad {\n" +
+                        "  document ad {\n" +
+                        "    field campaign_ref type reference<campaign> { indexing: attribute }\n" +
+                        "  }\n" +
+                        "}";
+        builder.addSchema(campaignSdContent);
+        builder.addSchema(adSdContent);
+        exceptionRule.expect(IllegalArgumentException.class);
+        exceptionRule.expectMessage("reference cycle for documents");
+        builder.build(true);
+    }
+
+    private static void assertSearchContainsReferenceField(String expectedFieldname,
+                                                           String referencedDocType,
+                                                           SDDocumentType documentType) {
+        Field field = documentType.getDocumentType().getField(expectedFieldname);
+        assertNotNull("Field does not exist in document type: " + expectedFieldname, field);
+        DataType dataType = field.getDataType();
+        assertTrue(dataType instanceof NewDocumentReferenceDataType);
+        NewDocumentReferenceDataType refField = (NewDocumentReferenceDataType) dataType;
+        assertEquals(referencedDocType, refField.getTargetTypeName());
+        assertTrue(! refField.isTemporary());
+    }
+
+}