mirror of
https://github.com/open-telemetry/opentelemetry-go.git
synced 2024-12-14 10:13:10 +02:00
245 lines
7.1 KiB
Go
245 lines
7.1 KiB
Go
|
package internal_test
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"io"
|
||
|
"io/ioutil"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/stretchr/testify/require"
|
||
|
|
||
|
"go.opentelemetry.io/otel/api/core"
|
||
|
"go.opentelemetry.io/otel/api/global"
|
||
|
"go.opentelemetry.io/otel/api/global/internal"
|
||
|
"go.opentelemetry.io/otel/api/key"
|
||
|
"go.opentelemetry.io/otel/exporter/metric/stdout"
|
||
|
metrictest "go.opentelemetry.io/otel/internal/metric"
|
||
|
"go.opentelemetry.io/otel/sdk/metric/batcher/ungrouped"
|
||
|
"go.opentelemetry.io/otel/sdk/metric/controller/push"
|
||
|
"go.opentelemetry.io/otel/sdk/metric/selector/simple"
|
||
|
)
|
||
|
|
||
|
func TestDirect(t *testing.T) {
|
||
|
internal.ResetForTest()
|
||
|
|
||
|
ctx := context.Background()
|
||
|
meter1 := global.MeterProvider().Meter("test1")
|
||
|
meter2 := global.MeterProvider().Meter("test2")
|
||
|
lvals1 := key.String("A", "B")
|
||
|
labels1 := meter1.Labels(lvals1)
|
||
|
lvals2 := key.String("C", "D")
|
||
|
labels2 := meter1.Labels(lvals2)
|
||
|
lvals3 := key.String("E", "F")
|
||
|
labels3 := meter2.Labels(lvals3)
|
||
|
|
||
|
counter := meter1.NewInt64Counter("test.counter")
|
||
|
counter.Add(ctx, 1, labels1)
|
||
|
counter.Add(ctx, 1, labels1)
|
||
|
|
||
|
gauge := meter1.NewInt64Gauge("test.gauge")
|
||
|
gauge.Set(ctx, 1, labels2)
|
||
|
gauge.Set(ctx, 2, labels2)
|
||
|
|
||
|
measure := meter1.NewFloat64Measure("test.measure")
|
||
|
measure.Record(ctx, 1, labels1)
|
||
|
measure.Record(ctx, 2, labels1)
|
||
|
|
||
|
second := meter2.NewFloat64Measure("test.second")
|
||
|
second.Record(ctx, 1, labels3)
|
||
|
second.Record(ctx, 2, labels3)
|
||
|
|
||
|
sdk := metrictest.NewProvider()
|
||
|
global.SetMeterProvider(sdk)
|
||
|
|
||
|
counter.Add(ctx, 1, labels1)
|
||
|
gauge.Set(ctx, 3, labels2)
|
||
|
measure.Record(ctx, 3, labels1)
|
||
|
second.Record(ctx, 3, labels3)
|
||
|
|
||
|
mock := sdk.Meter("test1").(*metrictest.Meter)
|
||
|
require.Equal(t, 3, len(mock.MeasurementBatches))
|
||
|
|
||
|
require.Equal(t, map[core.Key]core.Value{
|
||
|
lvals1.Key: lvals1.Value,
|
||
|
}, mock.MeasurementBatches[0].LabelSet.Labels)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches[0].Measurements))
|
||
|
require.Equal(t, core.NewInt64Number(1),
|
||
|
mock.MeasurementBatches[0].Measurements[0].Number)
|
||
|
require.Equal(t, "test.counter",
|
||
|
mock.MeasurementBatches[0].Measurements[0].Instrument.Name)
|
||
|
|
||
|
require.Equal(t, map[core.Key]core.Value{
|
||
|
lvals2.Key: lvals2.Value,
|
||
|
}, mock.MeasurementBatches[1].LabelSet.Labels)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches[1].Measurements))
|
||
|
require.Equal(t, core.NewInt64Number(3),
|
||
|
mock.MeasurementBatches[1].Measurements[0].Number)
|
||
|
require.Equal(t, "test.gauge",
|
||
|
mock.MeasurementBatches[1].Measurements[0].Instrument.Name)
|
||
|
|
||
|
require.Equal(t, map[core.Key]core.Value{
|
||
|
lvals1.Key: lvals1.Value,
|
||
|
}, mock.MeasurementBatches[2].LabelSet.Labels)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches[2].Measurements))
|
||
|
require.Equal(t, core.NewFloat64Number(3),
|
||
|
mock.MeasurementBatches[2].Measurements[0].Number)
|
||
|
require.Equal(t, "test.measure",
|
||
|
mock.MeasurementBatches[2].Measurements[0].Instrument.Name)
|
||
|
|
||
|
// This tests the second Meter instance
|
||
|
mock = sdk.Meter("test2").(*metrictest.Meter)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches))
|
||
|
|
||
|
require.Equal(t, map[core.Key]core.Value{
|
||
|
lvals3.Key: lvals3.Value,
|
||
|
}, mock.MeasurementBatches[0].LabelSet.Labels)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches[0].Measurements))
|
||
|
require.Equal(t, core.NewFloat64Number(3),
|
||
|
mock.MeasurementBatches[0].Measurements[0].Number)
|
||
|
require.Equal(t, "test.second",
|
||
|
mock.MeasurementBatches[0].Measurements[0].Instrument.Name)
|
||
|
}
|
||
|
|
||
|
func TestBound(t *testing.T) {
|
||
|
internal.ResetForTest()
|
||
|
|
||
|
// Note: this test uses oppsite Float64/Int64 number kinds
|
||
|
// vs. the above, to cover all the instruments.
|
||
|
ctx := context.Background()
|
||
|
glob := global.MeterProvider().Meter("test")
|
||
|
lvals1 := key.String("A", "B")
|
||
|
labels1 := glob.Labels(lvals1)
|
||
|
lvals2 := key.String("C", "D")
|
||
|
labels2 := glob.Labels(lvals2)
|
||
|
|
||
|
counter := glob.NewFloat64Counter("test.counter")
|
||
|
boundC := counter.AcquireHandle(labels1)
|
||
|
boundC.Add(ctx, 1)
|
||
|
boundC.Add(ctx, 1)
|
||
|
|
||
|
gauge := glob.NewFloat64Gauge("test.gauge")
|
||
|
boundG := gauge.AcquireHandle(labels2)
|
||
|
boundG.Set(ctx, 1)
|
||
|
boundG.Set(ctx, 2)
|
||
|
|
||
|
measure := glob.NewInt64Measure("test.measure")
|
||
|
boundM := measure.AcquireHandle(labels1)
|
||
|
boundM.Record(ctx, 1)
|
||
|
boundM.Record(ctx, 2)
|
||
|
|
||
|
sdk := metrictest.NewProvider()
|
||
|
global.SetMeterProvider(sdk)
|
||
|
|
||
|
boundC.Add(ctx, 1)
|
||
|
boundG.Set(ctx, 3)
|
||
|
boundM.Record(ctx, 3)
|
||
|
|
||
|
mock := sdk.Meter("test").(*metrictest.Meter)
|
||
|
require.Equal(t, 3, len(mock.MeasurementBatches))
|
||
|
|
||
|
require.Equal(t, map[core.Key]core.Value{
|
||
|
lvals1.Key: lvals1.Value,
|
||
|
}, mock.MeasurementBatches[0].LabelSet.Labels)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches[0].Measurements))
|
||
|
require.Equal(t, core.NewFloat64Number(1),
|
||
|
mock.MeasurementBatches[0].Measurements[0].Number)
|
||
|
require.Equal(t, "test.counter",
|
||
|
mock.MeasurementBatches[0].Measurements[0].Instrument.Name)
|
||
|
|
||
|
require.Equal(t, map[core.Key]core.Value{
|
||
|
lvals2.Key: lvals2.Value,
|
||
|
}, mock.MeasurementBatches[1].LabelSet.Labels)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches[1].Measurements))
|
||
|
require.Equal(t, core.NewFloat64Number(3),
|
||
|
mock.MeasurementBatches[1].Measurements[0].Number)
|
||
|
require.Equal(t, "test.gauge",
|
||
|
mock.MeasurementBatches[1].Measurements[0].Instrument.Name)
|
||
|
|
||
|
require.Equal(t, map[core.Key]core.Value{
|
||
|
lvals1.Key: lvals1.Value,
|
||
|
}, mock.MeasurementBatches[2].LabelSet.Labels)
|
||
|
require.Equal(t, 1, len(mock.MeasurementBatches[2].Measurements))
|
||
|
require.Equal(t, core.NewInt64Number(3),
|
||
|
mock.MeasurementBatches[2].Measurements[0].Number)
|
||
|
require.Equal(t, "test.measure",
|
||
|
mock.MeasurementBatches[2].Measurements[0].Instrument.Name)
|
||
|
|
||
|
boundC.Release()
|
||
|
boundG.Release()
|
||
|
boundM.Release()
|
||
|
}
|
||
|
|
||
|
func TestRelease(t *testing.T) {
|
||
|
// Tests Release with SDK never installed.
|
||
|
internal.ResetForTest()
|
||
|
|
||
|
glob := global.MeterProvider().Meter("test")
|
||
|
lvals1 := key.New("A").String("B")
|
||
|
labels1 := glob.Labels(lvals1)
|
||
|
lvals2 := key.New("C").String("D")
|
||
|
labels2 := glob.Labels(lvals2)
|
||
|
|
||
|
counter := glob.NewFloat64Counter("test.counter")
|
||
|
boundC := counter.AcquireHandle(labels1)
|
||
|
|
||
|
gauge := glob.NewFloat64Gauge("test.gauge")
|
||
|
boundG := gauge.AcquireHandle(labels2)
|
||
|
|
||
|
measure := glob.NewInt64Measure("test.measure")
|
||
|
boundM := measure.AcquireHandle(labels1)
|
||
|
|
||
|
boundC.Release()
|
||
|
boundG.Release()
|
||
|
boundM.Release()
|
||
|
}
|
||
|
|
||
|
func TestDefaultSDK(t *testing.T) {
|
||
|
internal.ResetForTest()
|
||
|
|
||
|
ctx := context.Background()
|
||
|
meter1 := global.MeterProvider().Meter("builtin")
|
||
|
lvals1 := key.String("A", "B")
|
||
|
labels1 := meter1.Labels(lvals1)
|
||
|
|
||
|
counter := meter1.NewInt64Counter("test.builtin")
|
||
|
counter.Add(ctx, 1, labels1)
|
||
|
counter.Add(ctx, 1, labels1)
|
||
|
|
||
|
in, out := io.Pipe()
|
||
|
// TODO this should equal a stdout.NewPipeline(), use it.
|
||
|
// Consider also moving the io.Pipe() and go func() call
|
||
|
// below into a test helper somewhere.
|
||
|
sdk := func(options stdout.Options) *push.Controller {
|
||
|
selector := simple.NewWithInexpensiveMeasure()
|
||
|
exporter, err := stdout.New(options)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
batcher := ungrouped.New(selector, true)
|
||
|
pusher := push.New(batcher, exporter, time.Second)
|
||
|
pusher.Start()
|
||
|
|
||
|
return pusher
|
||
|
}(stdout.Options{
|
||
|
File: out,
|
||
|
DoNotPrintTime: true,
|
||
|
})
|
||
|
|
||
|
global.SetMeterProvider(sdk)
|
||
|
|
||
|
counter.Add(ctx, 1, labels1)
|
||
|
|
||
|
ch := make(chan string)
|
||
|
go func() {
|
||
|
data, _ := ioutil.ReadAll(in)
|
||
|
ch <- string(data)
|
||
|
}()
|
||
|
|
||
|
sdk.Stop()
|
||
|
out.Close()
|
||
|
|
||
|
require.Equal(t, `{"updates":[{"name":"test.builtin{A=B}","sum":1}]}
|
||
|
`, <-ch)
|
||
|
}
|