1// Copyright 2012-present Oliver Eilhard. All rights reserved. 2// Use of this source code is governed by a MIT-license. 3// See http://olivere.mit-license.org/license.txt for details. 4 5package elastic 6 7import ( 8 "encoding/json" 9 "testing" 10) 11 12func TestScriptQuery(t *testing.T) { 13 q := NewScriptQuery(NewScript("doc['num1'.value > 1")) 14 src, err := q.Source() 15 if err != nil { 16 t.Fatal(err) 17 } 18 data, err := json.Marshal(src) 19 if err != nil { 20 t.Fatalf("marshaling to JSON failed: %v", err) 21 } 22 got := string(data) 23 expected := `{"script":{"script":{"source":"doc['num1'.value \u003e 1"}}}` 24 if got != expected { 25 t.Errorf("expected\n%s\n,got:\n%s", expected, got) 26 } 27} 28 29func TestScriptQueryWithParams(t *testing.T) { 30 q := NewScriptQuery(NewScript("doc['num1'.value > 1")) 31 q = q.QueryName("MyQueryName") 32 src, err := q.Source() 33 if err != nil { 34 t.Fatal(err) 35 } 36 data, err := json.Marshal(src) 37 if err != nil { 38 t.Fatalf("marshaling to JSON failed: %v", err) 39 } 40 got := string(data) 41 expected := `{"script":{"_name":"MyQueryName","script":{"source":"doc['num1'.value \u003e 1"}}}` 42 if got != expected { 43 t.Errorf("expected\n%s\n,got:\n%s", expected, got) 44 } 45} 46