github.com/opensearch-project/opensearch-go/v2@v2.3.0/opensearchapi/api.indices.datastream_test.go

// SPDX-License-Identifier: Apache-2.0
//
// The OpenSearch Contributors require contributions made to
// this file be licensed under the Apache-2.0 license or a
// compatible open source license.
//
// Modifications Copyright OpenSearch Contributors. See
// GitHub history for details.

//go:build integration
// +build integration

package opensearchapi_test

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"github.com/opensearch-project/opensearch-go/v2"
	"github.com/opensearch-project/opensearch-go/v2/opensearchapi"
	"github.com/stretchr/testify/require"
	"io/ioutil"
	"net/http"
	"strings"
	"testing"
	"time"
)

type DataStreamRequest interface {
	Do(context.Context, opensearchapi.Transport) (*opensearchapi.Response, error)
}

func TestIndicesDataStreams_Do(t *testing.T) {
	name := fmt.Sprintf("demo-%s", time.Now().Format("2006-01-02-15-04-05"))

	tests := []struct {
		name     string
		r        DataStreamRequest
		want     *opensearchapi.Response
		wantBody string
		wantErr  bool
	}{
		{
			name: "TestIndicesCreateDataStreamRequest_Do",
			r: opensearchapi.IndicesCreateDataStreamRequest{
				Name:       name,
				Pretty:     true,
				Human:      true,
				ErrorTrace: true,
				Header: map[string][]string{
					"Content-Type": {"application/json"},
				},
			},
			want: &opensearchapi.Response{
				StatusCode: 200,
				Header: http.Header{
					"Content-Type": []string{"application/json; charset=UTF-8"},
				},
			},
			wantBody: `{"acknowledged":true}`,
			wantErr:  false,
		},
		{
			name: "TestIndicesGetDataStreamRequest_Do",
			r: opensearchapi.IndicesGetDataStreamRequest{
				Name:       name,
				Pretty:     true,
				Human:      true,
				ErrorTrace: true,
				Header: map[string][]string{
					"Content-Type": {"application/json"},
				},
			},
			want: &opensearchapi.Response{
				StatusCode: 200,
				Header: http.Header{
					"Content-Type": []string{"application/json; charset=UTF-8"},
				},
			},
			wantErr: false,
		},
		{
			name: "TestIndicesGetAllDataStreamsRequest_Do",
			r: opensearchapi.IndicesGetDataStreamRequest{
				Pretty:     true,
				Human:      true,
				ErrorTrace: true,
				Header: map[string][]string{
					"Content-Type": {"application/json"},
				},
			},
			want: &opensearchapi.Response{
				StatusCode: 200,
				Header: http.Header{
					"Content-Type": []string{"application/json; charset=UTF-8"},
				},
			},
			wantErr: false,
		},
		{
			name: "TestIndicesGetStatsDataStreamRequest_Do",
			r: opensearchapi.IndicesGetDataStreamStatsRequest{
				Name:       name,
				Pretty:     true,
				Human:      true,
				ErrorTrace: true,
				Header: map[string][]string{
					"Content-Type": {"application/json"},
				},
			},
			want: &opensearchapi.Response{
				StatusCode: 200,
				Header: http.Header{
					"Content-Type": []string{"application/json; charset=UTF-8"},
				},
			},
			wantBody: fmt.Sprintf(`{"_shards":{"total":2,"successful":1,"failed":0},"data_stream_count":1,"backing_indices":1,"total_store_size":"208b","total_store_size_bytes":208,"data_streams":[{"data_stream":"%s","backing_indices":1,"store_size":"208b","store_size_bytes":208,"maximum_timestamp":0}]}`, name),
			wantErr:  false,
		},
		{
			name: "TestIndicesDeleteDataStreamRequest_Do",
			r: opensearchapi.IndicesDeleteDataStreamRequest{
				Name:       name,
				Pretty:     true,
				Human:      true,
				ErrorTrace: true,
				Header: map[string][]string{
					"Content-Type": {"application/json"},
				},
			},
			want: &opensearchapi.Response{
				StatusCode: 200,
				Header: http.Header{
					"Content-Type": []string{"application/json; charset=UTF-8"},
				},
			},
			wantBody: `{"acknowledged":true}`,
			wantErr:  false,
		},
	}

	client, err := opensearch.NewDefaultClient()
	require.NoError(t, err)

	iPut := opensearchapi.IndicesPutIndexTemplateRequest{
		Name:       fmt.Sprintf("demo-data-template"),
		Pretty:     true,
		Human:      true,
		ErrorTrace: true,
		Body:       strings.NewReader(fmt.Sprintf(`{"index_patterns": ["demo-*"], "data_stream": {}, "priority": 100} }`)),
	}

	iPutResponse, err := iPut.Do(context.Background(), client)
	require.NoError(t, err)
	require.Equalf(t, false, iPutResponse.IsError(),
		"Error when creating index template: %s", iPutResponse.String())

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := tt.r.Do(context.Background(), client)
			if (err != nil) != tt.wantErr {
				t.Errorf("Do() error = %v, wantErr %v", err, tt.wantErr)
				return
			}

			require.Equalf(t, got.IsError(), tt.wantErr, "Do() got = %v, want %v", got.IsError(), tt.wantErr)
			require.Equalf(t, got.StatusCode, tt.want.StatusCode, "Do() got = %v, want %v", got.StatusCode, tt.want.StatusCode)

			if tt.wantBody != "" {
				require.Equalf(t, got.Header, tt.want.Header, "Do() got = %v, want %v", got.Header, tt.want.Header)

				defer got.Body.Close()
				body, err := ioutil.ReadAll(got.Body)
				require.NoError(t, err)

				buffer := new(bytes.Buffer)
				if err := json.Compact(buffer, body); err != nil {
					fmt.Println(err)
				}

				require.Equalf(t, buffer.String(), tt.wantBody, "Do() got = %v, want %v", got.String(), tt.wantBody)
			}
		})
	}
}