vault/audit/sink_metric_labeler_test.go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1

package audit

import (
	"errors"
	"testing"

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

// TestMetricLabelerAuditSink_Label ensures we always get the right label based
// on the input value of the error.
func TestMetricLabelerAuditSink_Label(t *testing.T) {
	t.Parallel()

	tests := map[string]struct {
		err      error
		expected []string
	}{
		"nil": {
			err:      nil,
			expected: []string{"audit", "sink", "success"},
		},
		"error": {
			err:      errors.New("I am an error"),
			expected: []string{"audit", "sink", "failure"},
		},
	}

	for name, tc := range tests {
		name := name
		tc := tc
		t.Run(name, func(t *testing.T) {
			t.Parallel()

			m := &metricLabelerAuditSink{}
			result := m.Labels(nil, tc.err)
			assert.Equal(t, tc.expected, result)
		})
	}
}