Function TestClose

elasticsearch_internal_test.go:1474–1608

nolint:gocyclo

(t *testing.T)

//nolint:gocyclo
func TestClose(t *testing.T) {
	type closeable interface {
		esapi.Transport
		elastictransport.Closeable
	}

	tests := []struct {
		name string
		c    func() closeable
	}{
		{name: "BaseClient", c: func() closeable {
			c, _ := NewBase(WithTransportOptions(elastictransport.WithTransport(&mockTransp{})))
			return c
		}},
		{name: "Client", c: func() closeable {
			c, _ := New(WithTransportOptions(elastictransport.WithTransport(&mockTransp{})))
			return c
		}},
		{name: "TypedClient", c: func() closeable {
			c, _ := NewTyped(WithTransportOptions(elastictransport.WithTransport(&mockTransp{})))
			return c
		}},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Run("close", func(t *testing.T) {
				c := tt.c()
				err := c.Close(context.Background())
				if err != nil {
					t.Errorf("unexpected error: %v", err)
				}
			})

			t.Run("closed twice", func(t *testing.T) {
				c := tt.c()
				_ = c.Close(context.Background())
				err := c.Close(context.Background())
				if err != nil {
					if !errors.Is(err, ErrAlreadyClosed) {
						t.Errorf("unexpected error: %v", err)
					}
				}
			})

			t.Run("performing a request after close", func(t *testing.T) {
				c := tt.c()
				if err := c.Close(context.Background()); err != nil {
					t.Fatalf("unexpected error: %v", err)
				}
				req, err := http.NewRequest(http.MethodGet, "http://localhost:9200", nil)
				if err != nil {
					t.Fatalf("unexpected error: %v", err)
				}
				if _, err := c.Perform(req); err != nil {
					if !errors.Is(err, ErrClosed) {
						t.Fatalf("unexpected error: %v", err)
					}