mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Dmitry AnoshinandGitHub
6491891215
- `NewEntity(t string) Entity`: creates a standalone Entity with the given type - `Entity.CopyToResource(res pcommon.Resource)`: copies the entity's identity and descriptive attributes into a resource's entity refs - `EntityAttributeMap.All() iter.Seq2[string, pcommon.Value]`: iterates over the attribute key-value pairs belonging to this map's key set These are needed for the new mdatagen entity-scoped MetricsBuilder API in https://github.com/open-telemetry/opentelemetry-collector/pull/14660
126 lines
3.2 KiB
Go
126 lines
3.2 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package entity
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"go.opentelemetry.io/collector/pdata/pcommon"
|
|
)
|
|
|
|
func TestNewEntity(t *testing.T) {
|
|
e := NewEntity("service")
|
|
assert.Equal(t, "service", e.Type())
|
|
assert.Empty(t, e.SchemaURL())
|
|
}
|
|
|
|
func TestEntity_CopyToResource(t *testing.T) {
|
|
e := NewEntity("service")
|
|
e.IdentifyingAttributes().PutStr("service.name", "my-service")
|
|
e.DescriptiveAttributes().PutStr("service.version", "1.0.0")
|
|
|
|
res := pcommon.NewResource()
|
|
e.CopyToResource(res)
|
|
|
|
entities := ResourceEntities(res)
|
|
copied, ok := entities.Get("service")
|
|
assert.True(t, ok)
|
|
|
|
idVal, ok := copied.IdentifyingAttributes().Get("service.name")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "my-service", idVal.Str())
|
|
|
|
descVal, ok := copied.DescriptiveAttributes().Get("service.version")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "1.0.0", descVal.Str())
|
|
}
|
|
|
|
func TestEntity_Type(t *testing.T) {
|
|
em := NewEntityMap()
|
|
e := em.PutEmpty("service")
|
|
|
|
assert.Equal(t, "service", e.Type())
|
|
}
|
|
|
|
func TestEntity_SchemaURL(t *testing.T) {
|
|
em := NewEntityMap()
|
|
e := em.PutEmpty("service")
|
|
|
|
assert.Empty(t, e.SchemaURL())
|
|
|
|
e.SetSchemaURL("https://opentelemetry.io/schemas/1.0.0")
|
|
assert.Equal(t, "https://opentelemetry.io/schemas/1.0.0", e.SchemaURL())
|
|
|
|
e.SetSchemaURL("https://opentelemetry.io/schemas/1.1.0")
|
|
assert.Equal(t, "https://opentelemetry.io/schemas/1.1.0", e.SchemaURL())
|
|
}
|
|
|
|
func TestEntity_IdentifyingAttributes(t *testing.T) {
|
|
em := NewEntityMap()
|
|
e := em.PutEmpty("service")
|
|
|
|
idAttrs := e.IdentifyingAttributes()
|
|
idAttrs.PutStr("key1", "value1")
|
|
|
|
val, ok := e.IdentifyingAttributes().Get("key1")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "value1", val.Str())
|
|
}
|
|
|
|
func TestEntity_DescriptiveAttributes(t *testing.T) {
|
|
em := NewEntityMap()
|
|
e := em.PutEmpty("service")
|
|
|
|
descAttrs := e.DescriptiveAttributes()
|
|
descAttrs.PutStr("key1", "value1")
|
|
|
|
val, ok := e.DescriptiveAttributes().Get("key1")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "value1", val.Str())
|
|
}
|
|
|
|
func TestEntity_IdAndDescriptionAttributes_Isolated(t *testing.T) {
|
|
em := NewEntityMap()
|
|
e := em.PutEmpty("service")
|
|
|
|
e.IdentifyingAttributes().PutStr("id.key", "id-value")
|
|
e.DescriptiveAttributes().PutStr("desc.key", "desc-value")
|
|
|
|
val, ok := e.IdentifyingAttributes().Get("id.key")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "id-value", val.Str())
|
|
|
|
_, ok = e.IdentifyingAttributes().Get("desc.key")
|
|
assert.False(t, ok)
|
|
|
|
val, ok = e.DescriptiveAttributes().Get("desc.key")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "desc-value", val.Str())
|
|
|
|
_, ok = e.DescriptiveAttributes().Get("id.key")
|
|
assert.False(t, ok)
|
|
}
|
|
|
|
func TestEntity_IdAndDescriptionAttributes_CanPut(t *testing.T) {
|
|
em := NewEntityMap()
|
|
e := em.PutEmpty("service")
|
|
|
|
e.IdentifyingAttributes().PutStr("shared.key", "id-value")
|
|
|
|
assert.True(t, e.IdentifyingAttributes().CanPut("shared.key"))
|
|
assert.False(t, e.DescriptiveAttributes().CanPut("shared.key"))
|
|
|
|
e.DescriptiveAttributes().PutStr("shared.key", "desc-value")
|
|
|
|
val, ok := e.IdentifyingAttributes().Get("shared.key")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "desc-value", val.Str())
|
|
|
|
val, ok = e.DescriptiveAttributes().Get("shared.key")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "desc-value", val.Str())
|
|
}
|