1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
package com.yahoo.searchdefinition;
import com.yahoo.document.DataType;
import com.yahoo.document.ReferenceDataType;
import com.yahoo.searchdefinition.document.SDDocumentType;
import com.yahoo.searchdefinition.document.SDField;
import com.yahoo.searchdefinition.document.TemporaryImportedField;
import org.junit.Test;
import java.util.HashSet;
import java.util.List;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
public class ImportedFieldsEnumeratorTest {
@Test
public void imported_fields_are_enumerated_and_copied_from_correct_search_instance() {
Search parentSearch = new Search();
SDDocumentType parentDocument = new SDDocumentType("parent", parentSearch);
var parentField = new SDField("their_field", DataType.INT);
AttributeUtils.addAttributeAspect(parentField);
parentDocument.addField(parentField);
parentSearch.addDocument(parentDocument);
Search fooSearch = new Search();
SDField fooRefToParent = new SDField(
"foo_ref", ReferenceDataType.createWithInferredId(parentDocument.getDocumentType()));
AttributeUtils.addAttributeAspect(fooRefToParent);
var fooImports = fooSearch.temporaryImportedFields().get();
fooImports.add(new TemporaryImportedField("my_first_import", "foo_ref", "their_field"));
fooImports.add(new TemporaryImportedField("my_second_import", "foo_ref", "their_field"));
SDDocumentType fooDocument = new SDDocumentType("foo", fooSearch);
fooSearch.addDocument(fooDocument);
Search barSearch = new Search();
SDField barRefToParent = new SDField(
"bar_ref", ReferenceDataType.createWithInferredId(parentDocument.getDocumentType()));
AttributeUtils.addAttributeAspect(barRefToParent);
var barImports = barSearch.temporaryImportedFields().get();
barImports.add(new TemporaryImportedField("my_cool_import", "my_ref", "their_field"));
SDDocumentType barDocument = new SDDocumentType("bar", barSearch);
barSearch.addDocument(barDocument);
var enumerator = new ImportedFieldsEnumerator(List.of(parentSearch, fooSearch, barSearch));
enumerator.enumerateImportedFields(parentDocument);
assertImportedFieldsAre(parentDocument, List.of()); // No imported fields in parent
enumerator.enumerateImportedFields(fooDocument);
assertImportedFieldsAre(fooDocument, List.of("my_first_import", "my_second_import"));
enumerator.enumerateImportedFields(barDocument);
assertImportedFieldsAre(barDocument, List.of("my_cool_import"));
}
private void assertImportedFieldsAre(SDDocumentType documentType, List<String> expectedNames) {
assertNotNull(documentType.getTemporaryImportedFields());
var actualNames = documentType.getTemporaryImportedFields().fields().keySet();
var expectedNameSet = new HashSet<>(expectedNames);
assertEquals(expectedNameSet, actualNames);
}
}
|
