opentelemetry-collector/pdata/pprofile/generated_attributetablesli...

149 lines
4.3 KiB
Go

// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
// Code generated by "internal/cmd/pdatagen/main.go". DO NOT EDIT.
// To regenerate this file run "make genpdata".
package pprofile
import (
"testing"
"github.com/stretchr/testify/assert"
"go.opentelemetry.io/collector/pdata/internal"
v1 "go.opentelemetry.io/collector/pdata/internal/data/protogen/common/v1"
)
func TestAttributeTableSlice(t *testing.T) {
es := NewAttributeTableSlice()
assert.Equal(t, 0, es.Len())
state := internal.StateMutable
es = newAttributeTableSlice(&[]v1.KeyValue{}, &state)
assert.Equal(t, 0, es.Len())
emptyVal := NewAttribute()
testVal := generateTestAttribute()
for i := 0; i < 7; i++ {
es.AppendEmpty()
assert.Equal(t, emptyVal, es.At(i))
internal.FillOrigTestKeyValue(&(*es.orig)[i])
assert.Equal(t, testVal, es.At(i))
}
assert.Equal(t, 7, es.Len())
}
func TestAttributeTableSliceReadOnly(t *testing.T) {
sharedState := internal.StateReadOnly
es := newAttributeTableSlice(&[]v1.KeyValue{}, &sharedState)
assert.Equal(t, 0, es.Len())
assert.Panics(t, func() { es.AppendEmpty() })
assert.Panics(t, func() { es.EnsureCapacity(2) })
es2 := NewAttributeTableSlice()
es.CopyTo(es2)
assert.Panics(t, func() { es2.CopyTo(es) })
assert.Panics(t, func() { es.MoveAndAppendTo(es2) })
assert.Panics(t, func() { es2.MoveAndAppendTo(es) })
}
func TestAttributeTableSlice_CopyTo(t *testing.T) {
dest := NewAttributeTableSlice()
src := generateTestAttributeTableSlice()
src.CopyTo(dest)
assert.Equal(t, generateTestAttributeTableSlice(), dest)
}
func TestAttributeTableSlice_EnsureCapacity(t *testing.T) {
es := generateTestAttributeTableSlice()
// Test ensure smaller capacity.
const ensureSmallLen = 4
es.EnsureCapacity(ensureSmallLen)
assert.Less(t, ensureSmallLen, es.Len())
assert.Equal(t, es.Len(), cap(*es.orig))
assert.Equal(t, generateTestAttributeTableSlice(), es)
// Test ensure larger capacity
const ensureLargeLen = 9
es.EnsureCapacity(ensureLargeLen)
assert.Less(t, generateTestAttributeTableSlice().Len(), ensureLargeLen)
assert.Equal(t, ensureLargeLen, cap(*es.orig))
assert.Equal(t, generateTestAttributeTableSlice(), es)
}
func TestAttributeTableSlice_MoveAndAppendTo(t *testing.T) {
// Test MoveAndAppendTo to empty
expectedSlice := generateTestAttributeTableSlice()
dest := NewAttributeTableSlice()
src := generateTestAttributeTableSlice()
src.MoveAndAppendTo(dest)
assert.Equal(t, generateTestAttributeTableSlice(), dest)
assert.Equal(t, 0, src.Len())
assert.Equal(t, expectedSlice.Len(), dest.Len())
// Test MoveAndAppendTo empty slice
src.MoveAndAppendTo(dest)
assert.Equal(t, generateTestAttributeTableSlice(), dest)
assert.Equal(t, 0, src.Len())
assert.Equal(t, expectedSlice.Len(), dest.Len())
// Test MoveAndAppendTo not empty slice
generateTestAttributeTableSlice().MoveAndAppendTo(dest)
assert.Equal(t, 2*expectedSlice.Len(), dest.Len())
for i := 0; i < expectedSlice.Len(); i++ {
assert.Equal(t, expectedSlice.At(i), dest.At(i))
assert.Equal(t, expectedSlice.At(i), dest.At(i+expectedSlice.Len()))
}
dest.MoveAndAppendTo(dest)
assert.Equal(t, 2*expectedSlice.Len(), dest.Len())
for i := 0; i < expectedSlice.Len(); i++ {
assert.Equal(t, expectedSlice.At(i), dest.At(i))
assert.Equal(t, expectedSlice.At(i), dest.At(i+expectedSlice.Len()))
}
}
func TestAttributeTableSlice_RemoveIf(t *testing.T) {
// Test RemoveIf on empty slice
emptySlice := NewAttributeTableSlice()
emptySlice.RemoveIf(func(el Attribute) bool {
t.Fail()
return false
})
// Test RemoveIf
filtered := generateTestAttributeTableSlice()
pos := 0
filtered.RemoveIf(func(el Attribute) bool {
pos++
return pos%3 == 0
})
assert.Equal(t, 5, filtered.Len())
}
func TestAttributeTableSlice_RemoveIfAll(t *testing.T) {
got := generateTestAttributeTableSlice()
got.RemoveIf(func(el Attribute) bool {
return true
})
assert.Equal(t, 0, got.Len())
}
func TestAttributeTableSliceAll(t *testing.T) {
ms := generateTestAttributeTableSlice()
assert.NotEmpty(t, ms.Len())
var c int
for i, v := range ms.All() {
assert.Equal(t, ms.At(i), v, "element should match")
c++
}
assert.Equal(t, ms.Len(), c, "All elements should have been visited")
}
func generateTestAttributeTableSlice() AttributeTableSlice {
ms := NewAttributeTableSlice()
*ms.orig = internal.GenerateOrigTestKeyValueSlice()
return ms
}