badguardhome/internal/agherr/agherr_test.go

package agherr

import (
	"errors"
	"fmt"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestError_Error(t *testing.T) {
	testCases := []struct {
		err  error
		name string
		want string
	}{{
		err:  Many("a"),
		name: "simple",
		want: "a",
	}, {
		err:  Many("a", errors.New("b")),
		name: "wrapping",
		want: "a: b",
	}, {
		err:  Many("a", errors.New("b"), errors.New("c"), errors.New("d")),
		name: "wrapping several",
		want: "a: b (hidden: c, d)",
	}, {
		err:  Many("a", Many("b", errors.New("c"), errors.New("d"))),
		name: "wrapping wrapper",
		want: "a: b: c (hidden: d)",
	}}

	for _, tc := range testCases {
		assert.Equal(t, tc.want, tc.err.Error(), tc.name)
	}
}

func TestError_Unwrap(t *testing.T) {
	var _ wrapper = &manyError{}

	const (
		errSimple = iota
		errWrapped
		errNil
	)

	errs := []error{
		errSimple:  errors.New("a"),
		errWrapped: fmt.Errorf("err: %w", errors.New("nested")),
		errNil:     nil,
	}

	testCases := []struct {
		want    error
		wrapped error
		name    string
	}{{
		want:    errs[errSimple],
		wrapped: Many("a", errs[errSimple]),
		name:    "simple",
	}, {
		want:    errs[errWrapped],
		wrapped: Many("b", errs[errWrapped]),
		name:    "nested",
	}, {
		want:    errs[errNil],
		wrapped: Many("c", errs[errNil]),
		name:    "nil passed",
	}, {
		want:    nil,
		wrapped: Many("d"),
		name:    "nil not passed",
	}}

	for _, tc := range testCases {
		assert.Equal(t, tc.want, errors.Unwrap(tc.wrapped), tc.name)
	}
}

func TestAnnotate(t *testing.T) {
	const s = "1234"
	const wantMsg = `bad string "1234": test`

	// Don't use const, because we can't take a pointer of a constant.
	var errTest error = Error("test")

	t.Run("nil", func(t *testing.T) {
		var errPtr *error
		assert.NotPanics(t, func() {
			Annotate("bad string %q: %w", errPtr, s)
		})
	})

	t.Run("non_nil", func(t *testing.T) {
		errPtr := &errTest
		assert.NotPanics(t, func() {
			Annotate("bad string %q: %w", errPtr, s)
		})

		require.NotNil(t, errPtr)

		err := *errPtr
		require.NotNil(t, err)

		assert.Equal(t, wantMsg, err.Error())
	})

	t.Run("defer", func(t *testing.T) {
		f := func() (err error) {
			defer Annotate("bad string %q: %w", &errTest, s)

			return errTest
		}

		err := f()
		require.NotNil(t, err)

		assert.Equal(t, wantMsg, err.Error())
	})
}