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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.searchdefinition;
import com.yahoo.searchdefinition.document.Stemming;
import com.yahoo.searchdefinition.parser.ParseException;
import com.yahoo.searchdefinition.processing.OnnxModelTypeResolver;
import com.yahoo.vespa.model.test.utils.DeployLoggerStub;
import org.junit.Test;
import static com.yahoo.config.model.test.TestUtil.joinLines;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
/**
* Schema tests that don't depend on files.
*
* @author bratseth
*/
public class SchemaTestCase {
@Test
public void testValidationOfInheritedSchema() throws ParseException {
try {
String schema = joinLines(
"schema test inherits nonesuch {" +
" document test inherits nonesuch {" +
" }" +
"}");
DeployLoggerStub logger = new DeployLoggerStub();
SearchBuilder.createFromStrings(logger, schema);
assertEquals("schema 'test' inherits 'nonesuch', but this schema does not exist",
logger.entries.get(0).message);
fail("Expected failure");
}
catch (IllegalArgumentException e) {
assertEquals("schema 'test' inherits 'nonesuch', but this schema does not exist", e.getMessage());
}
}
@Test
public void testValidationOfSchemaAndDocumentInheritanceConsistency() throws ParseException {
try {
String parent = joinLines(
"schema parent {" +
" document parent {" +
" field pf1 type string {" +
" indexing: summary" +
" }" +
" }" +
"}");
String child = joinLines(
"schema child inherits parent {" +
" document child {" +
" field cf1 type string {" +
" indexing: summary" +
" }" +
" }" +
"}");
SearchBuilder.createFromStrings(new DeployLoggerStub(), parent, child);
}
catch (IllegalArgumentException e) {
assertEquals("schema 'child' inherits 'parent', " +
"but its document type does not inherit the parent's document type"
, e.getMessage());
}
}
@Test
public void testSchemaInheritance() throws ParseException {
String parentLines = joinLines(
"schema parent {" +
" document parent {" +
" field pf1 type string {" +
" indexing: summary" +
" }" +
" }" +
" fieldset parent_set {" +
" fields: pf1" +
" }" +
" stemming: none" +
" index parent_index {" +
" stemming: best" +
" }" +
" field parent_field type string {" +
" indexing: input pf1 | lowercase | index | attribute | summary" +
" }" +
" rank-profile parent_profile {" +
" }" +
" constant parent_constant {" +
" file: constants/my_constant_tensor_file.json" +
" type: tensor<float>(x{},y{})" +
" }" +
" onnx-model parent_model {" +
" file: models/my_model.onnx" +
" }" +
" document-summary parent_summary {" +
" summary pf1 type string {}" +
" }" +
"}");
String childLines = joinLines(
"schema child inherits parent {" +
" document child inherits parent {" +
" field cf1 type string {" +
" indexing: summary" +
" }" +
" }" +
"}");
SearchBuilder builder = new SearchBuilder(new DeployLoggerStub());
builder.processorsToSkip().add(OnnxModelTypeResolver.class); // Avoid discovering the Onnx model referenced does not exist
builder.importString(parentLines);
builder.importString(childLines);
builder.build(true);
var application = builder.application();
var child = application.schemas().get("child");
assertEquals("pf1", child.fieldSets().userFieldSets().get("parent_set").getFieldNames().stream().findFirst().get());
assertEquals(Stemming.NONE, child.getStemming());
assertEquals(Stemming.BEST, child.getIndex("parent_index").getStemming());
assertNotNull(child.getField("parent_field"));
assertNotNull(child.getExtraField("parent_field"));
assertNotNull(application.rankProfileRegistry().get(child, "parent_profile"));
assertNotNull(child.rankingConstants().get("parent_constant"));
assertTrue(child.rankingConstants().asMap().containsKey("parent_constant"));
assertNotNull(child.onnxModels().get("parent_model"));
assertTrue(child.onnxModels().asMap().containsKey("parent_model"));
assertNotNull(child.getSummary("parent_summary"));
assertTrue(child.getSummaries().containsKey("parent_summary"));
assertNotNull(child.getSummaryField("pf1"));
assertNotNull(child.getExplicitSummaryField("pf1"));
assertNotNull(child.getUniqueNamedSummaryFields().get("pf1"));
}
}
|
