Skip to content

Commit 711cb7e

Browse files
authored
Fix script update tests in Spark (#1783)
Stored script request body has changed in 8.0. Updating test code to reflect this change.
1 parent 7177bad commit 711cb7e

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)