diff options
Diffstat (limited to 'docprocs/src/test/java/com/yahoo/docprocs/indexing/DocumentScriptTestCase.java')
| -rw-r--r-- | docprocs/src/test/java/com/yahoo/docprocs/indexing/DocumentScriptTestCase.java | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/docprocs/src/test/java/com/yahoo/docprocs/indexing/DocumentScriptTestCase.java b/docprocs/src/test/java/com/yahoo/docprocs/indexing/DocumentScriptTestCase.java index 6e4057c57a0..bbce023647f 100644 --- a/docprocs/src/test/java/com/yahoo/docprocs/indexing/DocumentScriptTestCase.java +++ b/docprocs/src/test/java/com/yahoo/docprocs/indexing/DocumentScriptTestCase.java @@ -19,6 +19,7 @@ import com.yahoo.document.datatypes.Struct; import com.yahoo.document.datatypes.WeightedSet; import com.yahoo.document.fieldpathupdate.AssignFieldPathUpdate; import com.yahoo.document.fieldpathupdate.FieldPathUpdate; +import com.yahoo.document.update.AssignValueUpdate; import com.yahoo.document.update.FieldUpdate; import com.yahoo.document.update.MapValueUpdate; import com.yahoo.document.update.ValueUpdate; @@ -154,10 +155,8 @@ public class DocumentScriptTestCase { Struct out = (Struct)processDocument(in); assertSpanTrees(out.getFieldValue("myString"), "mySpanTree"); - StringFieldValue str = (StringFieldValue)((MapValueUpdate)processFieldUpdate(in)).getUpdate().getValue(); - assertSpanTrees(str, "mySpanTree"); - - str = (StringFieldValue)((MapValueUpdate)processFieldUpdate(in)).getUpdate().getValue(); + var updStruct = (Struct)((AssignValueUpdate)processFieldUpdate(in)).getValue(); + var str = (StringFieldValue)updStruct.getFieldValue(updStruct.getField("myString")); assertSpanTrees(str, "mySpanTree"); } |