107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package scrapererror
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestScrapeErrorsAddPartial(t *testing.T) {
|
|
err1 := errors.New("err 1")
|
|
err2 := errors.New("err 2")
|
|
expected := []error{
|
|
PartialScrapeError{error: err1, Failed: 1},
|
|
PartialScrapeError{error: err2, Failed: 10},
|
|
}
|
|
|
|
var errs ScrapeErrors
|
|
errs.AddPartial(1, err1)
|
|
errs.AddPartial(10, err2)
|
|
assert.Equal(t, expected, errs.errs)
|
|
}
|
|
|
|
func TestScrapeErrorsAdd(t *testing.T) {
|
|
err1 := errors.New("err a")
|
|
err2 := errors.New("err b")
|
|
expected := []error{err1, err2}
|
|
|
|
var errs ScrapeErrors
|
|
errs.Add(err1)
|
|
errs.Add(err2)
|
|
assert.Equal(t, expected, errs.errs)
|
|
}
|
|
|
|
func TestScrapeErrorsCombine(t *testing.T) {
|
|
testCases := []struct {
|
|
errs func() ScrapeErrors
|
|
expectedErr string
|
|
expectedFailedCount int
|
|
expectNil bool
|
|
expectedScrape bool
|
|
}{
|
|
{
|
|
errs: func() ScrapeErrors {
|
|
var errs ScrapeErrors
|
|
return errs
|
|
},
|
|
expectNil: true,
|
|
},
|
|
{
|
|
errs: func() ScrapeErrors {
|
|
var errs ScrapeErrors
|
|
errs.AddPartial(10, errors.New("bad scrapes"))
|
|
errs.AddPartial(1, fmt.Errorf("err: %w", errors.New("bad scrape")))
|
|
return errs
|
|
},
|
|
expectedErr: "bad scrapes; err: bad scrape",
|
|
expectedFailedCount: 11,
|
|
expectedScrape: true,
|
|
},
|
|
{
|
|
errs: func() ScrapeErrors {
|
|
var errs ScrapeErrors
|
|
errs.Add(errors.New("bad regular"))
|
|
errs.Add(fmt.Errorf("err: %w", errors.New("bad reg")))
|
|
return errs
|
|
},
|
|
expectedErr: "bad regular; err: bad reg",
|
|
},
|
|
{
|
|
errs: func() ScrapeErrors {
|
|
var errs ScrapeErrors
|
|
errs.AddPartial(2, errors.New("bad two scrapes"))
|
|
errs.AddPartial(10, fmt.Errorf("%d scrapes failed: %w", 10, errors.New("bad things happened")))
|
|
errs.Add(errors.New("bad event"))
|
|
errs.Add(fmt.Errorf("event: %w", errors.New("something happened")))
|
|
return errs
|
|
},
|
|
expectedErr: "bad two scrapes; 10 scrapes failed: bad things happened; bad event; event: something happened",
|
|
expectedFailedCount: 12,
|
|
expectedScrape: true,
|
|
},
|
|
}
|
|
|
|
for _, tt := range testCases {
|
|
scrapeErrs := tt.errs()
|
|
if tt.expectNil {
|
|
require.NoError(t, scrapeErrs.Combine())
|
|
continue
|
|
}
|
|
require.EqualError(t, scrapeErrs.Combine(), tt.expectedErr)
|
|
if tt.expectedScrape {
|
|
var partialScrapeErr PartialScrapeError
|
|
if !errors.As(scrapeErrs.Combine(), &partialScrapeErr) {
|
|
t.Errorf("%+v.Combine() = %q. Want: PartialScrapeError", scrapeErrs, scrapeErrs.Combine())
|
|
} else if tt.expectedFailedCount != partialScrapeErr.Failed {
|
|
t.Errorf("%+v.Combine().Failed. Got %d Failed count. Want: %d", scrapeErrs, partialScrapeErr.Failed, tt.expectedFailedCount)
|
|
}
|
|
}
|
|
}
|
|
}
|