Function TestBulkIndexer

esutil/bulk_indexer_internal_test.go:65–1659

nolint:gocyclo

(t *testing.T)

//nolint:gocyclo
func TestBulkIndexer(t *testing.T) {
	t.Run("Basic", func(t *testing.T) {
		tests := []struct {
			name       string
			makeClient func(cfg elasticsearch.Config) (esapi.Transport, error)
		}{
			{
				name: "Client",
				makeClient: func(cfg elasticsearch.Config) (esapi.Transport, error) {
					return elasticsearch.NewClient(cfg)
				},
			},
			{
				name: "TypedClient",
				makeClient: func(cfg elasticsearch.Config) (esapi.Transport, error) {
					return elasticsearch.NewTypedClient(cfg)
				},
			},
		}
		for _, tt := range tests {
			tt := tt

			t.Run(tt.name, func(t *testing.T) {
				var (
					wg sync.WaitGroup

					countReqs int
					testfile  string
					numItems  = 6
				)

				es, err := tt.makeClient(elasticsearch.Config{Transport: &mockTransport{
					RoundTripFunc: func(*http.Request) (*http.Response, error) {
						countReqs++
						switch countReqs {
						case 1:
							testfile = "testdata/bulk_response_1a.json"
						case 2:
							testfile = "testdata/bulk_response_1b.json"
						case 3:
							testfile = "testdata/bulk_response_1c.json"
						}
						bodyContent, err := os.ReadFile(testfile)
						if err != nil {
							t.Fatalf("Unexpected error: %s", err)
						}
						return &http.Response{
							Body:   io.NopCloser(bytes.NewBuffer(bodyContent)),
							Header: http.Header{"X-Elastic-Product": []string{"Elasticsearch"}},
						}, nil
					},
				}})

				if err != nil {
					t.Errorf("Unexpected error: %s", err)
				}

				cfg := BulkIndexerConfig{