FilesExpand file tree

main

/

import_observability_test.go

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

125 lines (117 loc) · 3.11 KB

main

/

import_observability_test.go

Copy path

Top

File metadata and controls

• Code
• Blame

125 lines (117 loc) · 3.11 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

package parser

import (

"encoding/json"

"testing"

"github.com/stretchr/testify/assert"

"github.com/stretchr/testify/require"

)

// TestExtractOTLPAttributesFromObsMap verifies extractOTLPAttributesFromObsMap

// correctly extracts string-valued OTLP custom attributes from a raw

// observability map, and returns nil for absent/malformed/empty input.

func TestExtractOTLPAttributesFromObsMap(t *testing.T) {

t.Parallel()

tests := []struct {

name string

obs map[string]any

expected map[string]string

}{

{

name: "nil map returns nil",

obs: nil,

expected: nil,

},

{

name: "empty map returns nil",

obs: map[string]any{},

expected: nil,

},

{

name: "missing otlp key returns nil",

obs: map[string]any{"other": "value"},

expected: nil,

},

{

name: "otlp value not a map returns nil",

obs: map[string]any{"otlp": "not-a-map"},

expected: nil,

},

{

name: "otlp map missing attributes returns nil",

obs: map[string]any{"otlp": map[string]any{"other": "value"}},

expected: nil,

},

{

name: "attributes value not a map returns nil",

obs: map[string]any{"otlp": map[string]any{"attributes": "not-a-map"}},

expected: nil,

},

{

name: "empty attributes map returns empty result",

obs: map[string]any{"otlp": map[string]any{"attributes": map[string]any{}}},

expected: map[string]string{},

},

{

name: "string attributes are extracted",

obs: map[string]any{

"otlp": map[string]any{

"attributes": map[string]any{

"env": "prod",

"service": "gh-aw",

},

},

},

expected: map[string]string{"env": "prod", "service": "gh-aw"},

},

{

name: "non-string attribute values are silently ignored",

obs: map[string]any{

"otlp": map[string]any{

"attributes": map[string]any{

"env": "prod",

"count": 42,

"enabled": true,

"nested": map[string]any{"a": "b"},

"list": []any{"x", "y"},

"missing": nil,

},

},

},

expected: map[string]string{"env": "prod"},

},

{

name: "empty string key is ignored",

obs: map[string]any{

"otlp": map[string]any{

"attributes": map[string]any{

"": "should-be-ignored",

"real": "value",

},

},

},

expected: map[string]string{"real": "value"},

},

}

for _, tt := range tests {

t.Run(tt.name, func(t *testing.T) {

t.Parallel()

got := extractOTLPAttributesFromObsMap(tt.obs)

assert.Equal(t, tt.expected, got)

})

}

}

func TestMergeObservabilityConfigsIncludesResourceAttributes(t *testing.T) {

configs := []string{

`{"otlp":{"resource-attributes":{"shared.key":"first","first.only":"one"}}}`,

`{"otlp":{"resource-attributes":{"shared.key":"second","second.only":"two"}}}`,

}

merged := mergeObservabilityConfigs(configs)

require.NotEmpty(t, merged)

var result map[string]any

require.NoError(t, json.Unmarshal([]byte(merged), &result))

otlp := result["otlp"].(map[string]any)

assert.Equal(t, map[string]any{

"shared.key": "first",

"first.only": "one",

"second.only": "two",

}, otlp["resource-attributes"])

}