59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
// Copyright 2012-present Oliver Eilhard. All rights reserved.
|
|
// Use of this source code is governed by a MIT-license.
|
|
// See http://olivere.mit-license.org/license.txt for details.
|
|
|
|
package elastic
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
func TestUpdateWithScript(t *testing.T) {
|
|
client := setupTestClientAndCreateIndexAndAddDocs(t) //, SetTraceLog(log.New(os.Stdout, "", 0)))
|
|
|
|
// Get original
|
|
getRes, err := client.Get().Index(testIndexName).Type("tweet").Id("1").Do(context.TODO())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var original tweet
|
|
if err := json.Unmarshal(*getRes.Source, &original); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Update with script
|
|
updRes, err := client.Update().Index(testIndexName).Type("tweet").Id("1").
|
|
Script(
|
|
NewScript(`ctx._source.message = "Updated message text."`).Lang("painless"),
|
|
).
|
|
Do(context.TODO())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if updRes == nil {
|
|
t.Fatal("response is nil")
|
|
}
|
|
if want, have := "updated", updRes.Result; want != have {
|
|
t.Fatalf("want Result = %q, have %v", want, have)
|
|
}
|
|
|
|
// Get new version
|
|
getRes, err = client.Get().Index(testIndexName).Type("tweet").Id("1").Do(context.TODO())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var updated tweet
|
|
if err := json.Unmarshal(*getRes.Source, &updated); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if want, have := original.User, updated.User; want != have {
|
|
t.Fatalf("want User = %q, have %v", want, have)
|
|
}
|
|
if want, have := "Updated message text.", updated.Message; want != have {
|
|
t.Fatalf("want Message = %q, have %v", want, have)
|
|
}
|
|
}
|