Skip to content

Commit 7501ddc

Browse files
authored
Fix script update tests in Spark (#1783) (#1796)
Stored script request body has changed in 8.0. Updating test code to reflect this change.
1 parent 4bea0e3 commit 7501ddc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

spark/core/src/itest/scala/org/elasticsearch/spark/integration/AbstractScalaEsSpark.scala

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -566,7 +566,9 @@ class AbstractScalaEsScalaSpark(prefix: String, readMetadata: jl.Boolean) extend
566566
"ctx._source.counter += 1"
567567
}
568568

569-
if (version.onOrAfter(EsMajorVersion.V_5_X)) {
569+
if (version.onOrAfter(EsMajorVersion.V_8_X)) {
570+
RestUtils.put(s"_scripts/$scriptName", s"""{"script":{"lang":"$lang", "source": "$script"}}""".getBytes(StringUtils.UTF_8))
571+
} else if (version.onOrAfter(EsMajorVersion.V_5_X)) {
570572
RestUtils.put(s"_scripts/$scriptName", s"""{"script":{"lang":"$lang", "code": "$script"}}""".getBytes(StringUtils.UTF_8))
571573
} else {
572574
RestUtils.put(s"_scripts/$lang/$scriptName", s"""{"script":"$script"}""".getBytes(StringUtils.UTF_8))

0 commit comments

Comments
 (0)