1 files changed, 21 insertions, 15 deletions
diff --git a/config-model/src/test/java/com/yahoo/searchdefinition/processing/AddAttributeTransformToSummaryOfImportedFieldsTest.java b/config-model/src/test/java/com/yahoo/searchdefinition/processing/AddAttributeTransformToSummaryOfImportedFieldsTest.java
index 78daa7b5fef..19634c4cfcc 100644
--- a/config-model/src/test/java/com/yahoo/searchdefinition/processing/AddAttributeTransformToSummaryOfImportedFieldsTest.java
+++ b/config-model/src/test/java/com/yahoo/searchdefinition/processing/AddAttributeTransformToSummaryOfImportedFieldsTest.java
@@ -29,7 +29,10 @@ public class AddAttributeTransformToSummaryOfImportedFieldsTest {
 
     @Test
     public void attribute_summary_transform_applied_to_summary_field_of_imported_field() {
-        Search search = createDocumentWithImportedFieldInDocumentSummary();
+        Search search = createSearchWithDocument(DOCUMENT_NAME);
+        search.setImportedFields(createSingleImportedField(IMPORTED_FIELD_NAME));
+        search.addSummary(createDocumentSummary(IMPORTED_FIELD_NAME));
+
         AddAttributeTransformToSummaryOfImportedFields processor = new AddAttributeTransformToSummaryOfImportedFields(
                 search,null,null,null);
         processor.process();
@@ -38,22 +41,25 @@ public class AddAttributeTransformToSummaryOfImportedFieldsTest {
         assertEquals(SummaryTransform.ATTRIBUTE, actualTransform);
     }
 
-    private static Search createDocumentWithImportedFieldInDocumentSummary() {
-        // Create search with a single imported field
-        Search search = new Search(DOCUMENT_NAME, MockApplicationPackage.createEmpty());
-        SDDocumentType document = new SDDocumentType("mydoc", search);
+    private static Search createSearchWithDocument(String documentName) {
+        Search search = new Search(documentName, MockApplicationPackage.createEmpty());
+        SDDocumentType document = new SDDocumentType(documentName, search);
         search.addDocument(document);
-        Search targetDocument = new Search("otherdoc", MockApplicationPackage.createEmpty());
-        SDField targetField = new SDField("myfield", DataType.INT);
-        DocumentReference documentReference = new DocumentReference(new Field("reference_field"), targetDocument);
-        ImportedField importedField = new ImportedField(IMPORTED_FIELD_NAME, documentReference, targetField);
-        search.setImportedFields(new ImportedFields(Collections.singletonMap(IMPORTED_FIELD_NAME, importedField)));
-
-        // Create document summary
-        DocumentSummary summary = new DocumentSummary(SUMMARY_NAME);
-        summary.add(new SummaryField(IMPORTED_FIELD_NAME, DataType.INT));
-        search.addSummary(summary);
         return search;
     }
 
+    private static ImportedFields createSingleImportedField(String fieldName) {
+        Search targetSearch = new Search("target_doc", MockApplicationPackage.createEmpty());
+        SDField targetField = new SDField("target_field", DataType.INT);
+        DocumentReference documentReference = new DocumentReference(new Field("reference_field"), targetSearch);
+        ImportedField importedField = new ImportedField(fieldName, documentReference, targetField);
+        return new ImportedFields(Collections.singletonMap(fieldName, importedField));
+    }
+
+    private static DocumentSummary createDocumentSummary(String fieldName) {
+        DocumentSummary summary = new DocumentSummary("mysummary");
+        summary.add(new SummaryField(fieldName, DataType.INT));
+        return summary;
+    }
+
 }