1// Copyright The OpenTelemetry Authors
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15package basic_test
16
17import (
18	"context"
19	"runtime"
20	"testing"
21	"time"
22
23	"github.com/stretchr/testify/require"
24
25	"go.opentelemetry.io/otel/attribute"
26	"go.opentelemetry.io/otel/metric"
27	export "go.opentelemetry.io/otel/sdk/export/metric"
28	controller "go.opentelemetry.io/otel/sdk/metric/controller/basic"
29	"go.opentelemetry.io/otel/sdk/metric/controller/controllertest"
30	processor "go.opentelemetry.io/otel/sdk/metric/processor/basic"
31	"go.opentelemetry.io/otel/sdk/metric/processor/processortest"
32	"go.opentelemetry.io/otel/sdk/resource"
33)
34
35func TestPullNoCollect(t *testing.T) {
36	puller := controller.New(
37		processor.New(
38			processortest.AggregatorSelector(),
39			export.CumulativeExportKindSelector(),
40			processor.WithMemory(true),
41		),
42		controller.WithCollectPeriod(0),
43		controller.WithResource(resource.Empty()),
44	)
45
46	ctx := context.Background()
47	meter := puller.MeterProvider().Meter("nocache")
48	counter := metric.Must(meter).NewInt64Counter("counter.sum")
49
50	counter.Add(ctx, 10, attribute.String("A", "B"))
51
52	require.NoError(t, puller.Collect(ctx))
53	records := processortest.NewOutput(attribute.DefaultEncoder())
54	require.NoError(t, puller.ForEach(export.CumulativeExportKindSelector(), records.AddRecord))
55
56	require.EqualValues(t, map[string]float64{
57		"counter.sum/A=B/": 10,
58	}, records.Map())
59
60	counter.Add(ctx, 10, attribute.String("A", "B"))
61
62	require.NoError(t, puller.Collect(ctx))
63	records = processortest.NewOutput(attribute.DefaultEncoder())
64	require.NoError(t, puller.ForEach(export.CumulativeExportKindSelector(), records.AddRecord))
65
66	require.EqualValues(t, map[string]float64{
67		"counter.sum/A=B/": 20,
68	}, records.Map())
69}
70
71func TestPullWithCollect(t *testing.T) {
72	puller := controller.New(
73		processor.New(
74			processortest.AggregatorSelector(),
75			export.CumulativeExportKindSelector(),
76			processor.WithMemory(true),
77		),
78		controller.WithCollectPeriod(time.Second),
79		controller.WithResource(resource.Empty()),
80	)
81	mock := controllertest.NewMockClock()
82	puller.SetClock(mock)
83
84	ctx := context.Background()
85	meter := puller.MeterProvider().Meter("nocache")
86	counter := metric.Must(meter).NewInt64Counter("counter.sum")
87
88	counter.Add(ctx, 10, attribute.String("A", "B"))
89
90	require.NoError(t, puller.Collect(ctx))
91	records := processortest.NewOutput(attribute.DefaultEncoder())
92	require.NoError(t, puller.ForEach(export.CumulativeExportKindSelector(), records.AddRecord))
93
94	require.EqualValues(t, map[string]float64{
95		"counter.sum/A=B/": 10,
96	}, records.Map())
97
98	counter.Add(ctx, 10, attribute.String("A", "B"))
99
100	// Cached value!
101	require.NoError(t, puller.Collect(ctx))
102	records = processortest.NewOutput(attribute.DefaultEncoder())
103	require.NoError(t, puller.ForEach(export.CumulativeExportKindSelector(), records.AddRecord))
104
105	require.EqualValues(t, map[string]float64{
106		"counter.sum/A=B/": 10,
107	}, records.Map())
108
109	mock.Add(time.Second)
110	runtime.Gosched()
111
112	// Re-computed value!
113	require.NoError(t, puller.Collect(ctx))
114	records = processortest.NewOutput(attribute.DefaultEncoder())
115	require.NoError(t, puller.ForEach(export.CumulativeExportKindSelector(), records.AddRecord))
116
117	require.EqualValues(t, map[string]float64{
118		"counter.sum/A=B/": 20,
119	}, records.Map())
120
121}
122