|
11 | 11 | import org.elasticsearch.common.collect.List; |
12 | 12 | import org.elasticsearch.common.collect.Set; |
13 | 13 | import org.elasticsearch.index.mapper.TypeFieldMapper.TypeFieldType; |
| 14 | +import org.elasticsearch.index.mapper.flattened.FlattenedFieldMapper; |
14 | 15 | import org.elasticsearch.test.ESTestCase; |
| 16 | +import org.hamcrest.Matchers; |
15 | 17 |
|
16 | 18 | import java.util.Arrays; |
17 | 19 | import java.util.Collection; |
18 | 20 | import java.util.Collections; |
| 21 | +import java.util.HashMap; |
| 22 | +import java.util.HashSet; |
| 23 | +import java.util.Map; |
19 | 24 |
|
20 | 25 | import static java.util.Collections.emptyList; |
21 | 26 | import static java.util.Collections.singletonList; |
22 | 27 | import static org.hamcrest.CoreMatchers.instanceOf; |
23 | 28 | import static org.hamcrest.Matchers.equalTo; |
| 29 | +import static org.hamcrest.collection.IsIterableContainingInAnyOrder.containsInAnyOrder; |
24 | 30 |
|
25 | 31 | public class FieldTypeLookupTests extends ESTestCase { |
26 | 32 |
|
@@ -218,4 +224,106 @@ private static int size(Iterable<MappedFieldType> iterable) { |
218 | 224 | } |
219 | 225 | return count; |
220 | 226 | } |
| 227 | + |
| 228 | + public void testFlattenedLookup() { |
| 229 | + String fieldName = "object1.object2.field"; |
| 230 | + FlattenedFieldMapper mapper = createFlattenedMapper(fieldName); |
| 231 | + |
| 232 | + FieldTypeLookup lookup = new FieldTypeLookup("_doc", singletonList(mapper), emptyList(), emptyList()); |
| 233 | + assertEquals(mapper.fieldType(), lookup.get(fieldName)); |
| 234 | + |
| 235 | + String objectKey = "key1.key2"; |
| 236 | + String searchFieldName = fieldName + "." + objectKey; |
| 237 | + |
| 238 | + MappedFieldType searchFieldType = lookup.get(searchFieldName); |
| 239 | + assertEquals(mapper.keyedFieldName(), searchFieldType.name()); |
| 240 | + assertThat(searchFieldType, Matchers.instanceOf(FlattenedFieldMapper.KeyedFlattenedFieldType.class)); |
| 241 | + |
| 242 | + FlattenedFieldMapper.KeyedFlattenedFieldType keyedFieldType = (FlattenedFieldMapper.KeyedFlattenedFieldType) searchFieldType; |
| 243 | + assertEquals(objectKey, keyedFieldType.key()); |
| 244 | + } |
| 245 | + |
| 246 | + public void testFlattenedLookupWithAlias() { |
| 247 | + String fieldName = "object1.object2.field"; |
| 248 | + FlattenedFieldMapper mapper = createFlattenedMapper(fieldName); |
| 249 | + |
| 250 | + String aliasName = "alias"; |
| 251 | + FieldAliasMapper alias = new FieldAliasMapper(aliasName, aliasName, fieldName); |
| 252 | + |
| 253 | + FieldTypeLookup lookup = new FieldTypeLookup("_doc", singletonList(mapper), singletonList(alias), emptyList()); |
| 254 | + assertEquals(mapper.fieldType(), lookup.get(aliasName)); |
| 255 | + |
| 256 | + String objectKey = "key1.key2"; |
| 257 | + String searchFieldName = aliasName + "." + objectKey; |
| 258 | + |
| 259 | + MappedFieldType searchFieldType = lookup.get(searchFieldName); |
| 260 | + assertEquals(mapper.keyedFieldName(), searchFieldType.name()); |
| 261 | + assertThat(searchFieldType, Matchers.instanceOf(FlattenedFieldMapper.KeyedFlattenedFieldType.class)); |
| 262 | + |
| 263 | + FlattenedFieldMapper.KeyedFlattenedFieldType keyedFieldType = (FlattenedFieldMapper.KeyedFlattenedFieldType) searchFieldType; |
| 264 | + assertEquals(objectKey, keyedFieldType.key()); |
| 265 | + } |
| 266 | + |
| 267 | + public void testFlattenedLookupWithMultipleFields() { |
| 268 | + String field1 = "object1.object2.field"; |
| 269 | + String field2 = "object1.field"; |
| 270 | + String field3 = "object2.field"; |
| 271 | + |
| 272 | + FlattenedFieldMapper mapper1 = createFlattenedMapper(field1); |
| 273 | + FlattenedFieldMapper mapper2 = createFlattenedMapper(field2); |
| 274 | + FlattenedFieldMapper mapper3 = createFlattenedMapper(field3); |
| 275 | + |
| 276 | + FieldTypeLookup lookup = new FieldTypeLookup("_doc", Arrays.asList(mapper1, mapper2), emptyList(), emptyList()); |
| 277 | + assertNotNull(lookup.get(field1 + ".some.key")); |
| 278 | + assertNotNull(lookup.get(field2 + ".some.key")); |
| 279 | + |
| 280 | + lookup = new FieldTypeLookup("_doc", Arrays.asList(mapper1, mapper2, mapper3), emptyList(), emptyList()); |
| 281 | + assertNotNull(lookup.get(field1 + ".some.key")); |
| 282 | + assertNotNull(lookup.get(field2 + ".some.key")); |
| 283 | + assertNotNull(lookup.get(field3 + ".some.key")); |
| 284 | + } |
| 285 | + |
| 286 | + public void testMaxDynamicKeyDepth() { |
| 287 | + Map<String, DynamicKeyFieldMapper> mappers = new HashMap<>(); |
| 288 | + Map<String, String> aliases = new HashMap<>(); |
| 289 | + assertEquals(0, DynamicKeyFieldTypeLookup.getMaxKeyDepth(mappers, aliases)); |
| 290 | + |
| 291 | + // Add a flattened object field. |
| 292 | + String name = "object1.object2.field"; |
| 293 | + FlattenedFieldMapper flattenedMapper = createFlattenedMapper(name); |
| 294 | + mappers.put(name, flattenedMapper); |
| 295 | + assertEquals(3, DynamicKeyFieldTypeLookup.getMaxKeyDepth(mappers, aliases)); |
| 296 | + |
| 297 | + // Add a short alias to that field. |
| 298 | + String aliasName = "alias"; |
| 299 | + aliases.put(aliasName, name); |
| 300 | + assertEquals(3, DynamicKeyFieldTypeLookup.getMaxKeyDepth(mappers, aliases)); |
| 301 | + |
| 302 | + // Add a longer alias to that field. |
| 303 | + String longAliasName = "object1.object2.object3.alias"; |
| 304 | + aliases.put(longAliasName, name); |
| 305 | + assertEquals(4, DynamicKeyFieldTypeLookup.getMaxKeyDepth(mappers, aliases)); |
| 306 | + |
| 307 | + // Update the long alias to refer to a non-flattened object field. |
| 308 | + String fieldName = "field"; |
| 309 | + aliases.put(longAliasName, fieldName); |
| 310 | + assertEquals(3, DynamicKeyFieldTypeLookup.getMaxKeyDepth(mappers, aliases)); |
| 311 | + } |
| 312 | + |
| 313 | + public void testFieldLookupIterator() { |
| 314 | + MockFieldMapper mapper = new MockFieldMapper("foo"); |
| 315 | + FlattenedFieldMapper flattenedMapper = createFlattenedMapper("object1.object2.field"); |
| 316 | + |
| 317 | + FieldTypeLookup lookup = new FieldTypeLookup("_doc", Arrays.asList(mapper, flattenedMapper), emptyList(), emptyList()); |
| 318 | + |
| 319 | + Collection<String> fieldNames = new HashSet<String>(); |
| 320 | + lookup.filter(ft -> true).forEach(ft -> fieldNames.add(ft.name())); |
| 321 | + |
| 322 | + assertThat(fieldNames, containsInAnyOrder( |
| 323 | + mapper.name(), flattenedMapper.name(), flattenedMapper.keyedFieldName())); |
| 324 | + } |
| 325 | + |
| 326 | + private FlattenedFieldMapper createFlattenedMapper(String fieldName) { |
| 327 | + return new FlattenedFieldMapper.Builder(fieldName).build(new ContentPath()); |
| 328 | + } |
221 | 329 | } |
0 commit comments