|
4 | 4 | package pmetricutil_test
|
5 | 5 |
|
6 | 6 | import (
|
7 |
| - "path/filepath" |
8 | 7 | "testing"
|
9 | 8 |
|
10 | 9 | "github.com/stretchr/testify/assert"
|
11 |
| - "github.com/stretchr/testify/require" |
12 | 10 | "go.opentelemetry.io/collector/pdata/pmetric"
|
13 | 11 |
|
14 | 12 | "github.com/open-telemetry/opentelemetry-collector-contrib/connector/routingconnector/internal/pmetricutil"
|
15 |
| - "github.com/open-telemetry/opentelemetry-collector-contrib/pkg/golden" |
| 13 | + "github.com/open-telemetry/opentelemetry-collector-contrib/connector/routingconnector/internal/pmetricutiltest" |
16 | 14 | "github.com/open-telemetry/opentelemetry-collector-contrib/pkg/pdatatest/pmetrictest"
|
17 | 15 | )
|
18 | 16 |
|
19 | 17 | func TestMoveResourcesIf(t *testing.T) {
|
20 | 18 | testCases := []struct {
|
21 |
| - name string |
22 |
| - condition func(pmetric.ResourceMetrics) bool |
| 19 | + name string |
| 20 | + moveIf func(pmetric.ResourceMetrics) bool |
| 21 | + from pmetric.Metrics |
| 22 | + to pmetric.Metrics |
| 23 | + expectFrom pmetric.Metrics |
| 24 | + expectTo pmetric.Metrics |
23 | 25 | }{
|
24 | 26 | {
|
25 | 27 | name: "move_none",
|
26 |
| - condition: func(pmetric.ResourceMetrics) bool { |
| 28 | + moveIf: func(pmetric.ResourceMetrics) bool { |
27 | 29 | return false
|
28 | 30 | },
|
| 31 | + from: pmetricutiltest.NewMetrics("AB", "CD", "EF", "FG"), |
| 32 | + to: pmetric.NewMetrics(), |
| 33 | + expectFrom: pmetricutiltest.NewMetrics("AB", "CD", "EF", "FG"), |
| 34 | + expectTo: pmetric.NewMetrics(), |
29 | 35 | },
|
30 | 36 | {
|
31 | 37 | name: "move_all",
|
32 |
| - condition: func(pmetric.ResourceMetrics) bool { |
| 38 | + moveIf: func(pmetric.ResourceMetrics) bool { |
33 | 39 | return true
|
34 | 40 | },
|
| 41 | + from: pmetricutiltest.NewMetrics("AB", "CD", "EF", "FG"), |
| 42 | + to: pmetric.NewMetrics(), |
| 43 | + expectFrom: pmetric.NewMetrics(), |
| 44 | + expectTo: pmetricutiltest.NewMetrics("AB", "CD", "EF", "FG"), |
35 | 45 | },
|
36 | 46 | {
|
37 | 47 | name: "move_one",
|
38 |
| - condition: func(rl pmetric.ResourceMetrics) bool { |
| 48 | + moveIf: func(rl pmetric.ResourceMetrics) bool { |
39 | 49 | rname, ok := rl.Resource().Attributes().Get("resourceName")
|
40 | 50 | return ok && rname.AsString() == "resourceA"
|
41 | 51 | },
|
| 52 | + from: pmetricutiltest.NewMetrics("AB", "CD", "EF", "FG"), |
| 53 | + to: pmetric.NewMetrics(), |
| 54 | + expectFrom: pmetricutiltest.NewMetrics("B", "CD", "EF", "FG"), |
| 55 | + expectTo: pmetricutiltest.NewMetrics("A", "CD", "EF", "FG"), |
42 | 56 | },
|
43 | 57 | {
|
44 | 58 | name: "move_to_preexisting",
|
45 |
| - condition: func(rl pmetric.ResourceMetrics) bool { |
| 59 | + moveIf: func(rl pmetric.ResourceMetrics) bool { |
46 | 60 | rname, ok := rl.Resource().Attributes().Get("resourceName")
|
47 | 61 | return ok && rname.AsString() == "resourceB"
|
48 | 62 | },
|
| 63 | + from: pmetricutiltest.NewMetrics("AB", "CD", "EF", "FG"), |
| 64 | + to: pmetricutiltest.NewMetrics("1", "2", "3", "4"), |
| 65 | + expectFrom: pmetricutiltest.NewMetrics("A", "CD", "EF", "FG"), |
| 66 | + expectTo: func() pmetric.Metrics { |
| 67 | + move := pmetricutiltest.NewMetrics("B", "CD", "EF", "FG") |
| 68 | + moveTo := pmetricutiltest.NewMetrics("1", "2", "3", "4") |
| 69 | + move.ResourceMetrics().MoveAndAppendTo(moveTo.ResourceMetrics()) |
| 70 | + return moveTo |
| 71 | + }(), |
49 | 72 | },
|
50 | 73 | }
|
51 | 74 |
|
52 | 75 | for _, tt := range testCases {
|
53 | 76 | t.Run(tt.name, func(t *testing.T) {
|
54 |
| - // Load up a fresh copy of the input for each test, since it may be modified in place. |
55 |
| - from, err := golden.ReadMetrics(filepath.Join("testdata", "resource", tt.name, "from.yaml")) |
56 |
| - require.NoError(t, err) |
57 |
| - |
58 |
| - to, err := golden.ReadMetrics(filepath.Join("testdata", "resource", tt.name, "to.yaml")) |
59 |
| - require.NoError(t, err) |
60 |
| - |
61 |
| - fromModifed, err := golden.ReadMetrics(filepath.Join("testdata", "resource", tt.name, "from_modified.yaml")) |
62 |
| - require.NoError(t, err) |
63 |
| - |
64 |
| - toModified, err := golden.ReadMetrics(filepath.Join("testdata", "resource", tt.name, "to_modified.yaml")) |
65 |
| - require.NoError(t, err) |
66 |
| - |
67 |
| - pmetricutil.MoveResourcesIf(from, to, tt.condition) |
68 |
| - |
69 |
| - assert.NoError(t, pmetrictest.CompareMetrics(fromModifed, from), "from not modified as expected") |
70 |
| - assert.NoError(t, pmetrictest.CompareMetrics(toModified, to), "to not as expected") |
| 77 | + pmetricutil.MoveResourcesIf(tt.from, tt.to, tt.moveIf) |
| 78 | + assert.NoError(t, pmetrictest.CompareMetrics(tt.expectFrom, tt.from), "from not modified as expected") |
| 79 | + assert.NoError(t, pmetrictest.CompareMetrics(tt.expectTo, tt.to), "to not as expected") |
71 | 80 | })
|
72 | 81 | }
|
73 | 82 | }
|
0 commit comments