[ML][Data Frame] have DataFrameTransformConfigUpdate#apply set Version (

elastic#45391)
BurnieLiang · Aug 9, 2019 · 8d4ffc6 · 8d4ffc6
1 parent 13a8835
commit 8d4ffc6
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/...main/java/org/elasticsearch/xpack/core/dataframe/transforms/DataFrameTransformConfig.java b/...main/java/org/elasticsearch/xpack/core/dataframe/transforms/DataFrameTransformConfig.java
@@ -447,6 +447,11 @@ public Builder setPivotConfig(PivotConfig pivotConfig) {
             return this;
         }
 
+        Builder setVersion(Version version) {
+            this.transformVersion = version;
+            return this;
+        }
+
         public DataFrameTransformConfig build() {
             return new DataFrameTransformConfig(id,
                 source,

diff --git a/...ava/org/elasticsearch/xpack/core/dataframe/transforms/DataFrameTransformConfigUpdate.java b/...ava/org/elasticsearch/xpack/core/dataframe/transforms/DataFrameTransformConfigUpdate.java
@@ -7,6 +7,7 @@
 package org.elasticsearch.xpack.core.dataframe.transforms;
 
 import org.elasticsearch.ElasticsearchStatusException;
+import org.elasticsearch.Version;
 import org.elasticsearch.common.Nullable;
 import org.elasticsearch.common.Strings;
 import org.elasticsearch.common.io.stream.StreamInput;
@@ -248,6 +249,7 @@ public DataFrameTransformConfig apply(DataFrameTransformConfig config) {
         if (headers != null) {
             builder.setHeaders(headers);
         }
+        builder.setVersion(Version.CURRENT);
         return builder.build();
     }
 }
diff --git a/...rg/elasticsearch/xpack/core/dataframe/transforms/DataFrameTransformConfigUpdateTests.java b/...rg/elasticsearch/xpack/core/dataframe/transforms/DataFrameTransformConfigUpdateTests.java
@@ -87,7 +87,7 @@ public void testApply() {
             PivotConfigTests.randomPivotConfig(),
             randomBoolean() ? null : randomAlphaOfLengthBetween(1, 1000),
             randomBoolean() ? null : Instant.now(),
-            randomBoolean() ? null : Version.CURRENT.toString());
+            randomBoolean() ? null : Version.V_7_2_0.toString());
         DataFrameTransformConfigUpdate update = new DataFrameTransformConfigUpdate(null, null, null, null, null);
 
         assertThat(config, equalTo(update.apply(config)));
@@ -108,6 +108,7 @@ public void testApply() {
         assertThat(updatedConfig.getSyncConfig(), equalTo(syncConfig));
         assertThat(updatedConfig.getDescription(), equalTo(newDescription));
         assertThat(updatedConfig.getHeaders(), equalTo(headers));
+        assertThat(updatedConfig.getVersion(), equalTo(Version.CURRENT));
     }
 
     public void testApplyWithSyncChange() {