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