1 files changed, 146 insertions, 0 deletions
diff --git a/vendor/github.com/olivere/elastic/field_caps_test.go b/vendor/github.com/olivere/elastic/field_caps_test.go
new file mode 100644
index 000000000..e299fd516
--- /dev/null
+++ b/vendor/github.com/olivere/elastic/field_caps_test.go
@@ -0,0 +1,146 @@
+// 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"
+	"net/url"
+	"reflect"
+	"sort"
+	"testing"
+)
+
+func TestFieldCapsURLs(t *testing.T) {
+	tests := []struct {
+		Service        *FieldCapsService
+		ExpectedPath   string
+		ExpectedParams url.Values
+	}{
+		{
+			Service:        &FieldCapsService{},
+			ExpectedPath:   "/_field_caps",
+			ExpectedParams: url.Values{},
+		},
+		{
+			Service: &FieldCapsService{
+				index: []string{"index1", "index2"},
+			},
+			ExpectedPath:   "/index1%2Cindex2/_field_caps",
+			ExpectedParams: url.Values{},
+		},
+		{
+			Service: &FieldCapsService{
+				index:  []string{"index_*"},
+				pretty: true,
+			},
+			ExpectedPath:   "/index_%2A/_field_caps",
+			ExpectedParams: url.Values{"pretty": []string{"true"}},
+		},
+	}
+
+	for _, test := range tests {
+		gotPath, gotParams, err := test.Service.buildURL()
+		if err != nil {
+			t.Fatalf("expected no error; got: %v", err)
+		}
+		if gotPath != test.ExpectedPath {
+			t.Errorf("expected URL path = %q; got: %q", test.ExpectedPath, gotPath)
+		}
+		if gotParams.Encode() != test.ExpectedParams.Encode() {
+			t.Errorf("expected URL params = %v; got: %v", test.ExpectedParams, gotParams)
+		}
+	}
+}
+
+func TestFieldCapsRequestSerialize(t *testing.T) {
+	req := &FieldCapsRequest{
+		Fields: []string{"creation_date", "answer_count"},
+	}
+	data, err := json.Marshal(req)
+	if err != nil {
+		t.Fatalf("marshaling to JSON failed: %v", err)
+	}
+	got := string(data)
+	expected := `{"fields":["creation_date","answer_count"]}`
+	if got != expected {
+		t.Fatalf("expected\n%s\n,got:\n%s", expected, got)
+	}
+}
+
+func TestFieldCapsRequestDeserialize(t *testing.T) {
+	body := `{
+		"fields" : ["creation_date", "answer_count"]
+	}`
+
+	var request FieldCapsRequest
+	if err := json.Unmarshal([]byte(body), &request); err != nil {
+		t.Fatalf("unexpected error during unmarshalling: %v", err)
+	}
+
+	sort.Sort(lexicographically{request.Fields})
+
+	expectedFields := []string{"answer_count", "creation_date"}
+	if !reflect.DeepEqual(request.Fields, expectedFields) {
+		t.Fatalf("expected fields to be %v, got %v", expectedFields, request.Fields)
+	}
+}
+
+func TestFieldCapsResponseUnmarshalling(t *testing.T) {
+	clusterStats := `{
+		"_shards": {
+				"total": 1,
+				"successful": 1,
+				"failed": 0
+		},
+		"fields": {
+			"creation_date": {
+				"type": "date",
+				"searchable": true,
+				"aggregatable": true,
+				"indices": ["index1", "index2"],
+				"non_searchable_indices": null,
+				"non_aggregatable_indices": null
+			},
+			"answer": {
+				"type": "keyword",
+				"searchable": true,
+				"aggregatable": true
+		}
+		}
+	}`
+
+	var resp FieldCapsResponse
+	if err := json.Unmarshal([]byte(clusterStats), &resp); err != nil {
+		t.Errorf("unexpected error during unmarshalling: %v", err)
+	}
+
+	caps, ok := resp.Fields["creation_date"]
+	if !ok {
+		t.Errorf("expected creation_date to be in the fields map, didn't find it")
+	}
+	if want, have := true, caps.Searchable; want != have {
+		t.Errorf("expected creation_date searchable to be %v, got %v", want, have)
+	}
+	if want, have := true, caps.Aggregatable; want != have {
+		t.Errorf("expected creation_date aggregatable to be %v, got %v", want, have)
+	}
+	if want, have := []string{"index1", "index2"}, caps.Indices; !reflect.DeepEqual(want, have) {
+		t.Errorf("expected creation_date indices to be %v, got %v", want, have)
+	}
+}
+
+func TestFieldCaps123(t *testing.T) {
+	client := setupTestClientAndCreateIndexAndAddDocs(t)
+	// client := setupTestClientAndCreateIndexAndAddDocs(t, SetTraceLog(log.New(os.Stdout, "", 0)))
+
+	res, err := client.FieldCaps("_all").Fields("user", "message", "retweets", "created").Pretty(true).Do(context.TODO())
+	if err != nil {
+		t.Fatalf("expected no error; got: %v", err)
+	}
+	if res == nil {
+		t.Fatalf("expected response; got: %v", res)
+	}
+}