elasticsearch_integration_test.go

// Licensed to Elasticsearch B.V. under one or more contributor
// license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright
// ownership. Elasticsearch B.V. licenses this file to you under
// the Apache License, Version 2.0 (the "License"); you may
// not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

//go:build integration && !multinode
// +build integration,!multinode

package elasticsearch_test

import (
	"context"
	"crypto/tls"
	"encoding/json"
	"fmt"
	"github.com/elastic/go-elasticsearch/v8/typedapi/core/search"
	"github.com/elastic/go-elasticsearch/v8/typedapi/indices/create"
	"github.com/elastic/go-elasticsearch/v8/typedapi/types"
	"github.com/elastic/go-elasticsearch/v8/typedapi/types/enums/refresh"
	"log"
	"net"
	"net/http"
	"net/url"
	"os"
	"strconv"
	"sync"
	"sync/atomic"
	"testing"
	"time"

	"github.com/elastic/elastic-transport-go/v8/elastictransport"
	"github.com/elastic/go-elasticsearch/v8"
	"github.com/elastic/go-elasticsearch/v8/esapi"
)

func TestClientTransport(t *testing.T) {
	t.Run("Persistent", func(t *testing.T) {
		es, err := elasticsearch.NewDefaultClient()
		if err != nil {
			t.Fatalf("Error creating the client: %s", err)
		}

		var total int

		for i := 0; i < 101; i++ {
			var curTotal int

			res, err := es.Nodes.Stats(es.Nodes.Stats.WithMetric("http"))
			if err != nil {
				t.Fatalf("Unexpected error: %s", err)
			}
			defer res.Body.Close()

			r := struct {
				Nodes map[string]struct {
					HTTP struct {
						TotalOpened int `json:"total_opened"`
					}
				}
			}{}

			if err := json.NewDecoder(res.Body).Decode(&r); err != nil {
				t.Fatalf("Error parsing the response body: %s", err)
			}

			for _, v := range r.Nodes {
				curTotal = v.HTTP.TotalOpened
				break
			}

			if curTotal < 1 {
				t.Errorf("Unexpected total_opened: %d", curTotal)
			}

			if total == 0 {
				total = curTotal
			}

			if total != curTotal {
				t.Errorf("Expected total_opened=%d, got: %d", total, curTotal)
			}
		}

		log.Printf("total_opened: %d", total)
	})

	t.Run("Concurrent", func(t *testing.T) {
		var wg sync.WaitGroup

		es, err := elasticsearch.NewDefaultClient()
		if err != nil {
			t.Fatalf("Error creating the client: %s", err)
		}

		for i := 0; i < 101; i++ {
			wg.Add(1)
			time.Sleep(10 * time.Millisecond)

			go func(i int) {
				defer wg.Done()
				res, err := es.Info()
				if err != nil {
					t.Errorf("Unexpected error: %s", err)
				} else {
					defer res.Body.Close()
				}
			}(i)
		}
		wg.Wait()
	})

	t.Run("WithContext", func(t *testing.T) {
		es, err := elasticsearch.NewDefaultClient()
		if err != nil {
			t.Fatalf("Error creating the client: %s", err)
		}

		ctx, cancel := context.WithTimeout(context.Background(), time.Nanosecond)
		defer cancel()

		res, err := es.Info(es.Info.WithContext(ctx))
		if err == nil {
			res.Body.Close()
			t.Fatal("Expected 'context deadline exceeded' error")
		}

		log.Printf("Request cancelled with %T", err)
	})

	t.Run("Configured", func(t *testing.T) {
		cfg := elasticsearch.Config{
			Transport: &http.Transport{
				MaxIdleConnsPerHost:   10,
				ResponseHeaderTimeout: time.Second,
				DialContext:           (&net.Dialer{Timeout: time.Nanosecond}).DialContext,
				TLSClientConfig: &tls.Config{
					MinVersion:         tls.VersionTLS12,
					InsecureSkipVerify: true,
				},
			},
		}

		es, err := elasticsearch.NewClient(cfg)
		if err != nil {
			t.Fatalf("Error creating the client: %s", err)
		}

		res, err := es.Info()
		if err == nil {
			t.Fatalf("Expected error, but got: %v", err)
		}
		if _, ok := err.(*net.OpError); !ok {
			t.Fatalf("Expected net.OpError, but got: %T", err)
		}

		if res != nil {
			t.Fatalf("Unexpected response: %+v", res)
		}
	})
}

type CustomTransport struct {
	client *http.Client
}

func (t *CustomTransport) RoundTrip(req *http.Request) (*http.Response, error) {
	req.Header.Set("X-Foo", "bar")
	log.Printf("> %s %s %s\n", req.Method, req.URL.String(), req.Header)
	return t.client.Do(req)
}

func TestClientCustomTransport(t *testing.T) {
	t.Run("Customized", func(t *testing.T) {
		cfg := elasticsearch.Config{
			Transport: &CustomTransport{
				client: http.DefaultClient,
			},
		}

		es, err := elasticsearch.NewClient(cfg)
		if err != nil {
			t.Fatalf("Error creating the client: %s", err)
		}

		for i := 0; i < 10; i++ {
			res, err := es.Info()
			if err != nil {
				t.Fatalf("Unexpected error: %s", err)
			}
			defer res.Body.Close()
		}
	})

	t.Run("Manual", func(t *testing.T) {
		tp, _ := elastictransport.New(elastictransport.Config{
			URLs: []*url.URL{
				{Scheme: "http", Host: "localhost:9200"},
			},
			Transport: http.DefaultTransport,
		})

		es := elasticsearch.Client{
			BaseClient: elasticsearch.BaseClient{
				Transport: tp,
			},
			API: esapi.New(tp),
		}

		for i := 0; i < 10; i++ {
			res, err := es.Info()
			if err != nil {
				t.Fatalf("Unexpected error: %s", err)
			}
			defer res.Body.Close()
		}
	})
}

type ReplacedTransport struct {
	counter uint64
}

func (t *ReplacedTransport) Perform(req *http.Request) (*http.Response, error) {
	req.URL.Scheme = "http"
	req.URL.Host = "localhost:9200"

	atomic.AddUint64(&t.counter, 1)

	return http.DefaultTransport.RoundTrip(req)
}

func (t *ReplacedTransport) Count() uint64 {
	return atomic.LoadUint64(&t.counter)
}

func TestClientReplaceTransport(t *testing.T) {
	t.Run("Replaced", func(t *testing.T) {
		tr := &ReplacedTransport{}
		es := elasticsearch.Client{
			BaseClient: elasticsearch.BaseClient{
				Transport: tr,
			},
			API: esapi.New(tr),
		}

		for i := 0; i < 10; i++ {
			res, err := es.Info()
			if err != nil {
				t.Fatalf("Unexpected error: %s", err)
			}
			defer res.Body.Close()
		}

		if tr.Count() != 10 {
			t.Errorf("Expected 10 requests, got=%d", tr.Count())
		}
	})
}

func TestClientAPI(t *testing.T) {
	t.Run("Info", func(t *testing.T) {
		es, err := elasticsearch.NewDefaultClient()
		if err != nil {
			log.Fatalf("Error creating the client: %s\n", err)
		}

		res, err := es.Info()
		if err != nil {
			log.Fatalf("Error getting the response: %s\n", err)
		}
		defer res.Body.Close()

		var d map[string]interface{}
		err = json.NewDecoder(res.Body).Decode(&d)
		if err != nil {
			log.Fatalf("Error parsing the response: %s\n", err)
		}

		fmt.Println(d["tagline"])
	})
}

func TestTypedClient(t *testing.T) {
	t.Run("Info", func(t *testing.T) {
		es, err := elasticsearch.NewTypedClient(elasticsearch.Config{})
		if err != nil {
			t.Fatalf("error creating the client: %s", err)
		}

		res, err := es.Info().Do(context.Background())
		if err != nil {
			t.Fatalf("error reading Info request: %s", err)
		}
		defer res.Body.Close()

		var d map[string]interface{}
		err = json.NewDecoder(res.Body).Decode(&d)
		if err != nil {
			log.Fatalf("Error parsing the response: %s\n", err)
		}

		if d["tagline"] != "You Know, for Search" {
			t.Errorf("invalid tagline, got: %s", d["tagline"])
		}
	})

	t.Run("Index & Search", func(t *testing.T) {
		es, err := elasticsearch.NewTypedClient(elasticsearch.Config{})

		u, _ := url.Parse("http://localhost:9200")
		tp, _ := elastictransport.New(
			elastictransport.Config{
				URLs:   []*url.URL{u},
				Logger: &elastictransport.ColorLogger{os.Stdout, true, true},
			},
		)
		es.Transport = tp
		if err != nil {
			t.Fatalf("error creating the client: %s", err)
		}

		// If the index doesn't exist we create it with a mapping.
		indexName := "test-index"
		if ok, err := es.Indices.Exists(indexName).IsSuccess(nil); !ok && err == nil {
			res, err := es.Indices.Create(indexName).
				Request(&create.Request{
					Mappings: &types.TypeMapping{
						Properties: map[types.PropertyName]types.Property{
							"price": types.NewIntegerNumberPropertyBuilder().Build(),
							"name":  types.NewKeywordPropertyBuilder().Build(),
						},
					},
				}).
				Do(nil)
			if err != nil {
				t.Fatalf("error creating index test-index: %s", err)
			}
			defer res.Body.Close()
		} else if err != nil {
			t.Error(err)
		}

		// Once everything is done we delete the index.
		defer func() {
			_, err := es.Indices.Delete(indexName).Do(context.Background())
			if err != nil {
				t.Error(err)
			}
		}()

		for _, document := range []struct {
			Id    int    `json:"id"`
			Name  string `json:"name"`
			Price int    `json:"price"`
		}{
			{
				Id:    1,
				Name:  "Foo",
				Price: 10,
			},
			{
				Id:    2,
				Name:  "Bar",
				Price: 12,
			},
			{
				Id:    3,
				Name:  "Baz",
				Price: 4,
			},
		} {
			indexResponse, err := es.Index(indexName).
				Request(document).
				Id(strconv.Itoa(document.Id)).
				Refresh(refresh.Waitfor).
				Do(context.Background())
			if err != nil {
				t.Fatalf("error indexing document: %s", err)
			}
			defer indexResponse.Body.Close()
		}

		if ok, err := es.Get(indexName, "1").IsSuccess(nil); !ok {
			t.Fatalf("could not retrieve document: %s", err)
		}

		res, err := es.Search().
			Index(indexName).
			Request(&search.Request{
				Query: &types.QueryContainer{
					Match: map[types.Field]types.MatchQuery{
						"name": {Query: "Foo"},
					},
				},
			}).Do(context.Background())

		if err != nil {
			t.Fatalf("error runnning search query: %s", err)
		}
		defer res.Body.Close()

		type SearchResult struct {
			Hits struct {
				Total struct {
					Value    int
					Relation string
				} `json:"total"`
				Hits []struct {
					Index  string `json:"_index"`
					Source struct {
						Id   int
						Name string
					} `json:"_source"`
				} `json:"hits"`
			} `json:"hits"`
		}

		sr := SearchResult{}

		err = json.NewDecoder(res.Body).Decode(&sr)
		if err != nil {
			t.Fatal(err)
		}

		if sr.Hits.Total.Value != 1 && sr.Hits.Hits[0].Source.Name != "Foo" {
			t.Fatalf("unexpected search result")
		}

		type PriceAggregation struct {
			Aggregations map[string]struct{ Value float64 } `json:"aggregations"`
		}

		pa := PriceAggregation{}

		size := 0
		field := types.Field("price")

		totalPricesAgg, err := es.Search().
			Index(indexName).
			Request(
				&search.Request{
					Size: &size,
					Aggregations: map[string]types.AggregationContainer{
						"total_prices": {
							Sum: &types.SumAggregation{
								Field: &field,
							},
						},
					},
				}).Do(context.Background())

		if err != nil {
			t.Fatal(err)
		}
		defer totalPricesAgg.Body.Close()

		err = json.NewDecoder(totalPricesAgg.Body).Decode(&pa)
		if err != nil {
			t.Fatal(err)
		}

		if pa.Aggregations["total_prices"].Value != 26. {
			t.Fatalf("error in aggregation, should be 26, got: %f", pa.Aggregations["total_prices"].Value)
		}
	})

	t.Run("Term query from JSON", func(t *testing.T) {
		searchRequest, err := search.NewRequestBuilder().FromJSON(`{
		  "query": {
			"term": {
			  "user.id": {
				"value": "kimchy",
				"boost": 1.0
			  }
			}
		  }
		}`)
		if err != nil {
			t.Fatal(err)
		}

		if searchRequest.Query.Term["user.id"].Value.(string) != "kimchy" {
			t.Fatalf("unexpected string in Query.Term.Value, expected kimchy, got: %s", searchRequest.Query.Term["user.id"].Value.(string))
		}
	})
}