@@ -7,9 +7,12 @@ import (
7
7
"testing"
8
8
9
9
metricspb "github.com/census-instrumentation/opencensus-proto/gen-go/metrics/v1"
10
+ "github.com/stretchr/testify/assert"
10
11
"github.com/stretchr/testify/require"
11
12
"go.uber.org/zap"
12
13
corev1 "k8s.io/api/core/v1"
14
+ "k8s.io/apimachinery/pkg/api/resource"
15
+ metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
13
16
14
17
"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/k8sclusterreceiver/internal/constants"
15
18
"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/k8sclusterreceiver/internal/testutils"
@@ -140,3 +143,63 @@ func TestNodeConditionValue(t *testing.T) {
140
143
})
141
144
}
142
145
}
146
+
147
+ func TestTransform (t * testing.T ) {
148
+ originalNode := & corev1.Node {
149
+ ObjectMeta : metav1.ObjectMeta {
150
+ Name : "my-node" ,
151
+ UID : "my-node-uid" ,
152
+ Labels : map [string ]string {
153
+ "node-role" : "worker" ,
154
+ },
155
+ },
156
+ Status : corev1.NodeStatus {
157
+ Conditions : []corev1.NodeCondition {
158
+ {
159
+ Type : corev1 .NodeReady ,
160
+ Status : corev1 .ConditionTrue ,
161
+ },
162
+ },
163
+ Capacity : corev1.ResourceList {
164
+ corev1 .ResourceCPU : resource .MustParse ("8" ),
165
+ corev1 .ResourceMemory : resource .MustParse ("16Gi" ),
166
+ },
167
+ Allocatable : corev1.ResourceList {
168
+ corev1 .ResourceCPU : resource .MustParse ("4" ),
169
+ corev1 .ResourceMemory : resource .MustParse ("8Gi" ),
170
+ },
171
+ Addresses : []corev1.NodeAddress {
172
+ {
173
+ Type : corev1 .NodeHostName ,
174
+ Address : "my-node-hostname" ,
175
+ },
176
+ {
177
+ Type : corev1 .NodeInternalIP ,
178
+ Address : "192.168.1.100" ,
179
+ },
180
+ },
181
+ },
182
+ }
183
+ wantNode := & corev1.Node {
184
+ ObjectMeta : metav1.ObjectMeta {
185
+ Name : "my-node" ,
186
+ UID : "my-node-uid" ,
187
+ Labels : map [string ]string {
188
+ "node-role" : "worker" ,
189
+ },
190
+ },
191
+ Status : corev1.NodeStatus {
192
+ Conditions : []corev1.NodeCondition {
193
+ {
194
+ Type : corev1 .NodeReady ,
195
+ Status : corev1 .ConditionTrue ,
196
+ },
197
+ },
198
+ Allocatable : corev1.ResourceList {
199
+ corev1 .ResourceCPU : resource .MustParse ("4" ),
200
+ corev1 .ResourceMemory : resource .MustParse ("8Gi" ),
201
+ },
202
+ },
203
+ }
204
+ assert .Equal (t , wantNode , Transform (originalNode ))
205
+ }
0 commit comments