threshold/internal/daemon/metric/initialize_metrics_test.go

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

package metric

import (
	"testing"

	"github.com/hashicorp/boundary/internal/gen/testing/protooptions"
	"github.com/stretchr/testify/assert"
	"google.golang.org/protobuf/reflect/protoreflect"
)

// As is, this test will currently fail if the services and methods in the protooptions
// package are changed. Please use temporary testcases to manually debug.
func Test_AppendServicesAndMethods(t *testing.T) {
	t.Parallel()

	cases := []struct {
		name     string
		pkg      protoreflect.FileDescriptor
		filter   func(string, string) bool
		expected map[string][]string
	}{
		{
			name:     "basic",
			pkg:      protooptions.File_testing_options_v1_service_proto,
			filter:   func(string, string) bool { return false },
			expected: map[string][]string{"testing.options.v1.TestService": {"TestMethod"}},
		},
		{
			name: "filter-out",
			pkg:  protooptions.File_testing_options_v1_service_proto,
			filter: func(_ string, m string) bool {
				return m == "TestMethod"
			},
			expected: map[string][]string{},
		},
		{
			name: "filter-allow",
			pkg:  protooptions.File_testing_options_v1_service_proto,
			filter: func(_ string, m string) bool {
				return m != "TestMethod"
			},
			expected: map[string][]string{"testing.options.v1.TestService": {"TestMethod"}},
		},
	}
	for _, tc := range cases {
		m := make(map[string][]string, 0)
		appendServicesAndMethods(m, tc.pkg, tc.filter)
		assert.Equal(t, tc.expected, m)
	}
}