opentelemetry-go/internal/metric/registry/registry_test.go

// Copyright The OpenTelemetry Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package registry_test

import (
	"context"
	"errors"
	"testing"

	"github.com/stretchr/testify/require"

	"go.opentelemetry.io/otel/internal/metric/registry"
	"go.opentelemetry.io/otel/metric"
	"go.opentelemetry.io/otel/metric/metrictest"
	"go.opentelemetry.io/otel/metric/sdkapi"
)

type (
	newFunc func(m metric.Meter, name string) (sdkapi.InstrumentImpl, error)
)

var (
	allNew = map[string]newFunc{
		"counter.int64": func(m metric.Meter, name string) (sdkapi.InstrumentImpl, error) {
			return unwrap(m.NewInt64Counter(name))
		},
		"counter.float64": func(m metric.Meter, name string) (sdkapi.InstrumentImpl, error) {
			return unwrap(m.NewFloat64Counter(name))
		},
		"histogram.int64": func(m metric.Meter, name string) (sdkapi.InstrumentImpl, error) {
			return unwrap(m.NewInt64Histogram(name))
		},
		"histogram.float64": func(m metric.Meter, name string) (sdkapi.InstrumentImpl, error) {
			return unwrap(m.NewFloat64Histogram(name))
		},
		"gaugeobserver.int64": func(m metric.Meter, name string) (sdkapi.InstrumentImpl, error) {
			return unwrap(m.NewInt64GaugeObserver(name, func(context.Context, metric.Int64ObserverResult) {}))
		},
		"gaugeobserver.float64": func(m metric.Meter, name string) (sdkapi.InstrumentImpl, error) {
			return unwrap(m.NewFloat64GaugeObserver(name, func(context.Context, metric.Float64ObserverResult) {}))
		},
	}
)

func unwrap(impl interface{}, err error) (sdkapi.InstrumentImpl, error) {
	if impl == nil {
		return nil, err
	}
	if s, ok := impl.(interface {
		SyncImpl() sdkapi.SyncImpl
	}); ok {
		return s.SyncImpl(), err
	}
	if a, ok := impl.(interface {
		AsyncImpl() sdkapi.AsyncImpl
	}); ok {
		return a.AsyncImpl(), err
	}
	return nil, err
}

func testMeterWithRegistry(name string) metric.Meter {
	return metric.WrapMeterImpl(
		registry.NewUniqueInstrumentMeterImpl(
			metrictest.NewMeterProvider().Meter(name).MeterImpl(),
		),
	)
}

func TestRegistrySameInstruments(t *testing.T) {
	for _, nf := range allNew {
		meter := testMeterWithRegistry("meter")
		inst1, err1 := nf(meter, "this")
		inst2, err2 := nf(meter, "this")

		require.NoError(t, err1)
		require.NoError(t, err2)
		require.Equal(t, inst1, inst2)
	}
}

func TestRegistryDifferentNamespace(t *testing.T) {
	for _, nf := range allNew {
		provider := metrictest.NewMeterProvider()

		meter1 := provider.Meter("meter1")
		meter2 := provider.Meter("meter2")
		inst1, err1 := nf(meter1, "this")
		inst2, err2 := nf(meter2, "this")

		require.NoError(t, err1)
		require.NoError(t, err2)
		require.NotEqual(t, inst1, inst2)
	}
}

func TestRegistryDiffInstruments(t *testing.T) {
	for origName, origf := range allNew {
		meter := testMeterWithRegistry("meter")

		_, err := origf(meter, "this")
		require.NoError(t, err)

		for newName, nf := range allNew {
			if newName == origName {
				continue
			}

			other, err := nf(meter, "this")
			require.Error(t, err)
			require.NotNil(t, other)
			require.True(t, errors.Is(err, registry.ErrMetricKindMismatch))
		}
	}
}