Add deployment replica count tests

Author: bjee19

internal/mode/static/telemetry/collector_test.go

@@ -7,6 +7,7 @@ import (
 
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
+	appsv1 "k8s.io/api/apps/v1"
 	v1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
@@ -39,6 +40,38 @@ func createListCallsFunc(nodes []v1.Node) func(
 	}
 }
 
+func createGetCallsFunc() func(
+	ctx context.Context,
+	key client.ObjectKey,
+	object client.Object,
+	option ...client.GetOption,
+) error {
+	return func(ctx context.Context, key client.ObjectKey, object client.Object, option ...client.GetOption) error {
+		Expect(option).To(BeEmpty())
+
+		switch typedObj := object.(type) {
+		case *v1.Pod:
+			typedObj.ObjectMeta = metav1.ObjectMeta{
+				Name: "pod1",
+				OwnerReferences: []metav1.OwnerReference{
+					{
+						Kind: "ReplicaSet",
+						Name: "replicaset1",
+					},
+				},
+			}
+		case *appsv1.ReplicaSet:
+			replicas := int32(1)
+			typedObj.Spec = appsv1.ReplicaSetSpec{
+				Replicas: &replicas,
+			}
+		default:
+			Fail(fmt.Sprintf("unknown type: %T", typedObj))
+		}
+		return nil
+	}
+}
+
 var _ = Describe("Collector", Ordered, func() {
 	var (
 		k8sClientReader         *eventsfakes.FakeReader
@@ -61,7 +94,7 @@ var _ = Describe("Collector", Ordered, func() {
 			ProjectMetadata:   telemetry.ProjectMetadata{Name: "NGF", Version: version},
 			NodeCount:         0,
 			NGFResourceCounts: telemetry.NGFResourceCounts{},
-			NGFReplicaCount:   0,
+			NGFReplicaCount:   1,
 		}
 
 		k8sClientReader = &eventsfakes.FakeReader{}
@@ -82,6 +115,7 @@ var _ = Describe("Collector", Ordered, func() {
 			Version:             version,
 			PodNSName:           podNSName,
 		})
+		k8sClientReader.GetCalls(createGetCallsFunc())
 	})
 
 	Describe("Normal case", func() {
@@ -370,4 +404,106 @@ var _ = Describe("Collector", Ordered, func() {
 			})
 		})
 	})
+
+	Describe("NGF replica count collector", func() {
+		When("collecting NGF replica count", func() {
+			When("it encounters an error while collecting data", func() {
+				BeforeEach(func() {
+					k8sClientReader.GetReturns(nil)
+					k8sClientReader.GetCalls(createGetCallsFunc())
+				})
+
+				It("should error if the kubernetes client errored when getting the Pod", func() {
+					k8sClientReader.GetReturnsOnCall(0, errors.New("there was an error"))
+
+					_, err := dataCollector.Collect(ctx)
+					Expect(err).To(HaveOccurred())
+				})
+
+				It("should error if the Pod's owner reference is nil", func() {
+					k8sClientReader.GetCalls(
+						func(ctx context.Context, key client.ObjectKey, object client.Object, option ...client.GetOption) error {
+							Expect(option).To(BeEmpty())
+
+							switch typedObj := object.(type) {
+							case *v1.Pod:
+								typedObj.ObjectMeta = metav1.ObjectMeta{
+									Name:            "pod1",
+									OwnerReferences: nil,
+								}
+							default:
+								Fail(fmt.Sprintf("unknown type: %T", typedObj))
+							}
+							return nil
+						})
+
+					_, err := dataCollector.Collect(ctx)
+					Expect(err).To(HaveOccurred())
+				})
+
+				It("should error if the Pod has multiple owner references", func() {
+					k8sClientReader.GetCalls(
+						func(ctx context.Context, key client.ObjectKey, object client.Object, option ...client.GetOption) error {
+							Expect(option).To(BeEmpty())
+
+							switch typedObj := object.(type) {
+							case *v1.Pod:
+								typedObj.ObjectMeta = metav1.ObjectMeta{
+									Name: "pod1",
+									OwnerReferences: []metav1.OwnerReference{
+										{
+											Kind: "ReplicaSet",
+											Name: "replicaset1",
+										},
+										{
+											Kind: "ReplicaSet",
+											Name: "replicaset2",
+										},
+									},
+								}
+							default:
+								Fail(fmt.Sprintf("unknown type: %T", typedObj))
+							}
+							return nil
+						})
+
+					_, err := dataCollector.Collect(ctx)
+					Expect(err).To(HaveOccurred())
+				})
+
+				It("should error if the Pod's owner reference is not a ReplicaSet", func() {
+					k8sClientReader.GetCalls(
+						func(ctx context.Context, key client.ObjectKey, object client.Object, option ...client.GetOption) error {
+							Expect(option).To(BeEmpty())
+
+							switch typedObj := object.(type) {
+							case *v1.Pod:
+								typedObj.ObjectMeta = metav1.ObjectMeta{
+									Name: "pod1",
+									OwnerReferences: []metav1.OwnerReference{
+										{
+											Kind: "Deployment",
+											Name: "deployment1",
+										},
+									},
+								}
+							default:
+								Fail(fmt.Sprintf("unknown type: %T", typedObj))
+							}
+							return nil
+						})
+
+					_, err := dataCollector.Collect(ctx)
+					Expect(err).To(HaveOccurred())
+				})
+
+				It("should error if the kubernetes client errored when getting the ReplicaSet", func() {
+					k8sClientReader.GetReturnsOnCall(1, errors.New("there was an error"))
+
+					_, err := dataCollector.Collect(ctx)
+					Expect(err).To(HaveOccurred())
+				})
+			})
+		})
+	})
 })