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