1
0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2025-04-21 11:57:04 +02:00
opentelemetry-go/sdk/internal/state_locker_test.go
Gustavo Silva Paiva 288821cd22
Make histogram aggregator checkpoint consistent (#438)
* change the histogram aggregator to have a consistent but blocking Checkpoint()

* docs

* wrapping docs

* remove currentIdx from the 8bit alignment check

* stress test

* add export and move lockfreewrite algorithm to an external struct.

* move state locker to another package.

* add todos

* minimal tests

* renaming and docs

* change to context.Background()

* add link to algorithm and grammars

Co-authored-by: Joshua MacDonald <jmacd@users.noreply.github.com>
2020-03-11 11:12:23 -07:00

89 lines
1.8 KiB
Go

package internal
import (
"testing"
"time"
"github.com/stretchr/testify/require"
)
func TestInflightOperationMustEndBeforeSwap(t *testing.T) {
var swapped bool
ch := make(chan struct{})
l := StateLocker{}
op1 := l.Start()
go func() {
l.SwapActiveState(func() {})
swapped = true
ch <- struct{}{}
}()
require.False(t, swapped, "Swap should wait the end of the in-flight operation.")
l.End(op1)
select {
case <-ch:
require.True(t, swapped, "Swap should've been completed. ")
case <-time.After(50 * time.Millisecond):
t.Fatal("Swap was not concluded after 50 milliseconds.")
}
}
func TestEnsureIndexIsConsistent(t *testing.T) {
l := StateLocker{}
op1 := l.Start()
l.End(op1)
l.SwapActiveState(func() {})
op2 := l.Start()
l.End(op2)
op3 := l.Start()
l.End(op3)
l.SwapActiveState(func() {})
op4 := l.Start()
l.End(op4)
require.Equal(t, op1, op4, "two operations separated by two swaps should have the same index.")
require.Equal(t, op2, op3, "two operations with no swap in between should have the same index.")
require.Equal(t, 0, op1, "first index should be 0")
require.Equal(t, 1, op2, "second index should be 1")
}
func TestTwoSwapsCanHappenWithoutOperationsInBetween(t *testing.T) {
l := StateLocker{}
require.Equal(t, 1, l.ColdIdx(), "first cold index should be 1")
l.SwapActiveState(func() {})
require.Equal(t, 0, l.ColdIdx(), "second cold index should be 0")
l.SwapActiveState(func() {})
require.Equal(t, 1, l.ColdIdx(), "third cold index should be 1")
}
func BenchmarkStateLocker_StartEnd(b *testing.B) {
l := StateLocker{}
b.ReportAllocs()
for i := 0; i < b.N; i++ {
l.End(l.Start())
}
}
func BenchmarkStateLocker_SwapActiveState(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
l := StateLocker{}
l.SwapActiveState(func() {})
}
}