Skip to content

Commit 1c8cf8b

Browse files
committed
Add temporary collector tests
1 parent 2187af7 commit 1c8cf8b

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

internal/mode/static/telemetry/collector_test.go

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ import (
99

1010
. "github.com/onsi/ginkgo/v2"
1111
. "github.com/onsi/gomega"
12+
"github.com/spf13/pflag"
1213
appsv1 "k8s.io/api/apps/v1"
1314
v1 "k8s.io/api/core/v1"
1415
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -79,6 +80,7 @@ var _ = Describe("Collector", Ordered, func() {
7980
kubeNamespace *v1.Namespace
8081

8182
baseGetCalls getCallsFunc
83+
flagset *pflag.FlagSet
8284
)
8385

8486
BeforeAll(func() {
@@ -129,16 +131,19 @@ var _ = Describe("Collector", Ordered, func() {
129131

130132
BeforeEach(func() {
131133
expData = telemetry.Data{
132-
ProjectMetadata: telemetry.ProjectMetadata{Name: "NGF", Version: version},
133-
NodeCount: 0,
134-
NGFResourceCounts: telemetry.NGFResourceCounts{},
135-
NGFReplicaCount: 1,
136-
ClusterID: string(kubeNamespace.GetUID()),
137-
ImageSource: "local",
138-
Arch: runtime.GOARCH,
139-
DeploymentID: string(ngfReplicaSet.ObjectMeta.OwnerReferences[0].UID),
134+
ProjectMetadata: telemetry.ProjectMetadata{Name: "NGF", Version: version},
135+
NodeCount: 0,
136+
NGFResourceCounts: telemetry.NGFResourceCounts{},
137+
NGFReplicaCount: 1,
138+
ClusterID: string(kubeNamespace.GetUID()),
139+
ImageSource: "local",
140+
Arch: runtime.GOARCH,
141+
DeploymentID: string(ngfReplicaSet.ObjectMeta.OwnerReferences[0].UID),
142+
DeploymentFlagOptions: telemetry.DeploymentFlagOptions{FlagKeys: []string{}, FlagValues: []string{}},
140143
}
141144

145+
flagset = pflag.NewFlagSet("flagset", 0)
146+
142147
k8sClientReader = &eventsfakes.FakeReader{}
143148
fakeGraphGetter = &telemetryfakes.FakeGraphGetter{}
144149
fakeConfigurationGetter = &telemetryfakes.FakeConfigurationGetter{}
@@ -153,6 +158,7 @@ var _ = Describe("Collector", Ordered, func() {
153158
Version: version,
154159
PodNSName: podNSName,
155160
ImageSource: "local",
161+
Flags: flagset,
156162
})
157163

158164
baseGetCalls = createGetCallsFunc(ngfPod, ngfReplicaSet, kubeNamespace)

0 commit comments

Comments
 (0)