|
@@ -33,7 +33,8 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
getMapOrEmpty(getTestPipelineSubstitutions()),
|
|
|
getMapOrEmpty(getTestComponentTemplateSubstitutions()),
|
|
|
getMapOrEmpty(getTestIndexTemplateSubstitutions()),
|
|
|
- getMapOrEmpty(getTestMappingAddition())
|
|
|
+ getMapOrEmpty(getTestMappingAddition()),
|
|
|
+ getTestMappingMergeType()
|
|
|
);
|
|
|
}
|
|
|
|
|
@@ -52,7 +53,8 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
null,
|
|
|
getTestPipelineSubstitutions(),
|
|
|
getTestComponentTemplateSubstitutions(),
|
|
|
- getTestMappingAddition()
|
|
|
+ getTestMappingAddition(),
|
|
|
+ getTestMappingMergeType()
|
|
|
)
|
|
|
);
|
|
|
assertThrows(
|
|
@@ -61,12 +63,19 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
getTestPipelineSubstitutions(),
|
|
|
null,
|
|
|
getTestComponentTemplateSubstitutions(),
|
|
|
- getTestMappingAddition()
|
|
|
+ getTestMappingAddition(),
|
|
|
+ getTestMappingMergeType()
|
|
|
)
|
|
|
);
|
|
|
assertThrows(
|
|
|
NullPointerException.class,
|
|
|
- () -> new SimulateBulkRequest(getTestPipelineSubstitutions(), getTestPipelineSubstitutions(), null, getTestMappingAddition())
|
|
|
+ () -> new SimulateBulkRequest(
|
|
|
+ getTestPipelineSubstitutions(),
|
|
|
+ getTestPipelineSubstitutions(),
|
|
|
+ null,
|
|
|
+ getTestMappingAddition(),
|
|
|
+ getTestMappingMergeType()
|
|
|
+ )
|
|
|
);
|
|
|
assertThrows(
|
|
|
NullPointerException.class,
|
|
@@ -74,7 +83,8 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
getTestPipelineSubstitutions(),
|
|
|
getTestPipelineSubstitutions(),
|
|
|
getTestComponentTemplateSubstitutions(),
|
|
|
- null
|
|
|
+ null,
|
|
|
+ getTestMappingMergeType()
|
|
|
)
|
|
|
);
|
|
|
}
|
|
@@ -83,13 +93,15 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
Map<String, Map<String, Object>> pipelineSubstitutions,
|
|
|
Map<String, Map<String, Object>> componentTemplateSubstitutions,
|
|
|
Map<String, Map<String, Object>> indexTemplateSubstitutions,
|
|
|
- Map<String, Object> mappingAddition
|
|
|
+ Map<String, Object> mappingAddition,
|
|
|
+ String mappingMergeType
|
|
|
) throws IOException {
|
|
|
SimulateBulkRequest simulateBulkRequest = new SimulateBulkRequest(
|
|
|
pipelineSubstitutions,
|
|
|
componentTemplateSubstitutions,
|
|
|
indexTemplateSubstitutions,
|
|
|
- mappingAddition
|
|
|
+ mappingAddition,
|
|
|
+ mappingMergeType
|
|
|
);
|
|
|
/*
|
|
|
* Note: SimulateBulkRequest does not implement equals or hashCode, so we can't test serialization in the usual way for a
|
|
@@ -97,11 +109,18 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
*/
|
|
|
SimulateBulkRequest copy = copyWriteable(simulateBulkRequest, null, SimulateBulkRequest::new);
|
|
|
assertThat(copy.getPipelineSubstitutions(), equalTo(simulateBulkRequest.getPipelineSubstitutions()));
|
|
|
+ assertThat(copy.getMappingMergeType(), equalTo(simulateBulkRequest.getMappingMergeType()));
|
|
|
}
|
|
|
|
|
|
@SuppressWarnings({ "unchecked", "rawtypes" })
|
|
|
public void testGetComponentTemplateSubstitutions() throws IOException {
|
|
|
- SimulateBulkRequest simulateBulkRequest = new SimulateBulkRequest(Map.of(), Map.of(), Map.of(), Map.of());
|
|
|
+ SimulateBulkRequest simulateBulkRequest = new SimulateBulkRequest(
|
|
|
+ Map.of(),
|
|
|
+ Map.of(),
|
|
|
+ Map.of(),
|
|
|
+ Map.of(),
|
|
|
+ getTestMappingMergeType()
|
|
|
+ );
|
|
|
assertThat(simulateBulkRequest.getComponentTemplateSubstitutions(), equalTo(Map.of()));
|
|
|
String substituteComponentTemplatesString = """
|
|
|
{
|
|
@@ -135,7 +154,13 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
XContentType.JSON
|
|
|
).v2();
|
|
|
Map<String, Map<String, Object>> substituteComponentTemplates = (Map<String, Map<String, Object>>) tempMap;
|
|
|
- simulateBulkRequest = new SimulateBulkRequest(Map.of(), substituteComponentTemplates, Map.of(), Map.of());
|
|
|
+ simulateBulkRequest = new SimulateBulkRequest(
|
|
|
+ Map.of(),
|
|
|
+ substituteComponentTemplates,
|
|
|
+ Map.of(),
|
|
|
+ Map.of(),
|
|
|
+ getTestMappingMergeType()
|
|
|
+ );
|
|
|
Map<String, ComponentTemplate> componentTemplateSubstitutions = simulateBulkRequest.getComponentTemplateSubstitutions();
|
|
|
assertThat(componentTemplateSubstitutions.size(), equalTo(2));
|
|
|
assertThat(
|
|
@@ -160,7 +185,13 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
}
|
|
|
|
|
|
public void testGetIndexTemplateSubstitutions() throws IOException {
|
|
|
- SimulateBulkRequest simulateBulkRequest = new SimulateBulkRequest(Map.of(), Map.of(), Map.of(), Map.of());
|
|
|
+ SimulateBulkRequest simulateBulkRequest = new SimulateBulkRequest(
|
|
|
+ Map.of(),
|
|
|
+ Map.of(),
|
|
|
+ Map.of(),
|
|
|
+ Map.of(),
|
|
|
+ getTestMappingMergeType()
|
|
|
+ );
|
|
|
assertThat(simulateBulkRequest.getIndexTemplateSubstitutions(), equalTo(Map.of()));
|
|
|
String substituteIndexTemplatesString = """
|
|
|
{
|
|
@@ -196,7 +227,7 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
randomBoolean(),
|
|
|
XContentType.JSON
|
|
|
).v2();
|
|
|
- simulateBulkRequest = new SimulateBulkRequest(Map.of(), Map.of(), substituteIndexTemplates, Map.of());
|
|
|
+ simulateBulkRequest = new SimulateBulkRequest(Map.of(), Map.of(), substituteIndexTemplates, Map.of(), getTestMappingMergeType());
|
|
|
Map<String, ComposableIndexTemplate> indexTemplateSubstitutions = simulateBulkRequest.getIndexTemplateSubstitutions();
|
|
|
assertThat(indexTemplateSubstitutions.size(), equalTo(2));
|
|
|
assertThat(
|
|
@@ -222,7 +253,8 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
getTestPipelineSubstitutions(),
|
|
|
getTestComponentTemplateSubstitutions(),
|
|
|
getTestIndexTemplateSubstitutions(),
|
|
|
- getTestMappingAddition()
|
|
|
+ getTestMappingAddition(),
|
|
|
+ getTestMappingMergeType()
|
|
|
);
|
|
|
simulateBulkRequest.setRefreshPolicy(randomFrom(WriteRequest.RefreshPolicy.values()));
|
|
|
simulateBulkRequest.waitForActiveShards(randomIntBetween(1, 10));
|
|
@@ -308,4 +340,12 @@ public class SimulateBulkRequestTests extends ESTestCase {
|
|
|
)
|
|
|
);
|
|
|
}
|
|
|
+
|
|
|
+ private static String getTestMappingMergeType() {
|
|
|
+ if (randomBoolean()) {
|
|
|
+ return null;
|
|
|
+ } else {
|
|
|
+ return randomFrom("index", "template");
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|