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
|
// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.searchdefinition.processing;
import com.google.common.collect.ImmutableMap;
import com.yahoo.config.application.api.ApplicationPackage;
import com.yahoo.config.model.test.MockApplicationPackage;
import com.yahoo.document.DataType;
import com.yahoo.document.ReferenceDataType;
import com.yahoo.document.TemporaryStructuredDataType;
import com.yahoo.searchdefinition.DocumentReference;
import com.yahoo.searchdefinition.DocumentReferences;
import com.yahoo.searchdefinition.Search;
import com.yahoo.searchdefinition.document.SDDocumentType;
import com.yahoo.searchdefinition.document.SDField;
import com.yahoo.searchdefinition.document.TemporaryImportedField;
import com.yahoo.searchdefinition.document.TemporarySDField;
/*
* Fixture class used for ImportedFieldsResolverTestCase and AdjustPositionSummaryFieldsTestCase.
*/
public class ParentChildSearchModel {
private final ApplicationPackage app = MockApplicationPackage.createEmpty();
public Search parentSearch;
public Search childSearch;
ParentChildSearchModel() {
parentSearch = createSearch("parent");
childSearch = createSearch("child");
}
protected Search createSearch(String name) {
Search result = new Search(name, app);
result.addDocument(new SDDocumentType(name));
return result;
}
protected static TemporarySDField createField(String name, DataType dataType, String indexingScript) {
TemporarySDField result = new TemporarySDField(name, dataType);
result.parseIndexingScript(indexingScript);
return result;
}
protected static SDField createRefField(String parentType, String fieldName) {
return new TemporarySDField(fieldName, ReferenceDataType.createWithInferredId(TemporaryStructuredDataType.create(parentType)));
}
protected static void addRefField(Search child, Search parent, String fieldName) {
SDField refField = createRefField(parent.getName(), fieldName);
child.getDocument().addField(refField);
child.getDocument().setDocumentReferences(new DocumentReferences(ImmutableMap.of(refField.getName(),
new DocumentReference(refField, parent))));
}
protected ParentChildSearchModel addImportedField(String fieldName, String referenceFieldName, String targetFieldName) {
return addImportedField(childSearch, fieldName, referenceFieldName, targetFieldName);
}
protected ParentChildSearchModel addImportedField(Search search, String fieldName, String referenceFieldName, String targetFieldName) {
search.temporaryImportedFields().get().add(new TemporaryImportedField(fieldName, referenceFieldName, targetFieldName));
return this;
}
}
|
