feat: add subscriber k8s fuzz test

chaoscenter/subscriber/pkg/k8s/client.go

@@ -1,6 +1,8 @@
 package k8s
 
 import (
+	"fmt"
+
 	wfclientset "github.com/argoproj/argo-workflows/v3/pkg/client/clientset/versioned"
 	v1alpha12 "github.com/argoproj/argo-workflows/v3/pkg/client/clientset/versioned/typed/workflow/v1alpha1"
 	"k8s.io/client-go/discovery"
@@ -14,11 +16,19 @@ var KubeConfig *string
 
 // getKubeConfig setup the config for access cluster resource
 func (k8s *k8sSubscriber) GetKubeConfig() (*rest.Config, error) {
-	// Use in-cluster config if kubeconfig path is not specified
+	if KubeConfig == nil {
+		return rest.InClusterConfig()
+	}
+
 	if *KubeConfig == "" {
 		return rest.InClusterConfig()
 	}
-	return clientcmd.BuildConfigFromFlags("", *KubeConfig)
+
+	config, err := clientcmd.BuildConfigFromFlags("", *KubeConfig)
+	if err != nil {
+		return nil, fmt.Errorf("failed to build config from kubeconfig: %v", err)
+	}
+	return config, nil
 }
 
 func (k8s *k8sSubscriber) GetGenericK8sClient() (*kubernetes.Clientset, error) {

chaoscenter/subscriber/pkg/k8s/fuzz/client_fuzz_test.go

@@ -0,0 +1,38 @@
+package fuzz
+
+import (
+	"subscriber/pkg/graphql"
+	"subscriber/pkg/k8s"
+	"testing"
+
+	fuzz "github.com/AdaLogics/go-fuzz-headers"
+)
+
+func FuzzGenerateArgoClient(f *testing.F) {
+	f.Fuzz(func(t *testing.T, data []byte) {
+		fuzzConsumer := fuzz.NewConsumer(data)
+		targetStruct := &struct {
+			namespace string
+		}{}
+		err := fuzzConsumer.GenerateStruct(targetStruct)
+		if err != nil {
+			return
+		}
+
+		if targetStruct.namespace == "" {
+			targetStruct.namespace = "default-namespace"
+		}
+		subscriberGraphql := graphql.NewSubscriberGql()
+		subscriberK8s := k8s.NewK8sSubscriber(subscriberGraphql)
+
+		wfClient, err := subscriberK8s.GenerateArgoClient(targetStruct.namespace)
+		if err != nil {
+			t.Logf("Unexpected error: %v", err)
+			return
+		}
+
+		if wfClient == nil {
+			t.Errorf("Returned payload is nil")
+		}
+	})
+}

chaoscenter/subscriber/pkg/k8s/fuzz/log_fuzz_test.go

@@ -0,0 +1,108 @@
+package fuzz
+
+import (
+	"subscriber/pkg/graphql"
+	"subscriber/pkg/k8s"
+	"subscriber/pkg/types"
+	"testing"
+
+	fuzz "github.com/AdaLogics/go-fuzz-headers"
+)
+
+func FuzzGetLogs(f *testing.F) {
+	f.Fuzz(func(t *testing.T, data []byte) {
+		fuzzConsumer := fuzz.NewConsumer(data)
+		targetStruct := &struct {
+			PodName   string
+			Namespace string
+			Container string
+		}{}
+
+		err := fuzzConsumer.GenerateStruct(targetStruct)
+		if err != nil {
+			t.Skip("Failed to generate struct:", err)
+			return
+		}
+
+		subscriberGraphql := graphql.NewSubscriberGql()
+		if subscriberGraphql == nil {
+			t.Fatalf("Failed to create a GraphQL subscriber")
+		}
+
+		subscriberK8s := k8s.NewK8sSubscriber(subscriberGraphql)
+		if subscriberK8s == nil {
+			t.Fatalf("Failed to create a Kubernetes subscriber")
+		}
+
+		response, err := subscriberK8s.GetLogs(targetStruct.PodName, targetStruct.Namespace, targetStruct.Container)
+		if err != nil {
+			t.Logf("Unexpected error: %v", err)
+			return
+		}
+		if response == "" {
+			t.Errorf("Returned payload is nil")
+		}
+	})
+}
+
+func FuzzCreatePodLogs(f *testing.F) {
+	f.Fuzz(func(t *testing.T, data []byte) {
+		fuzzConsumer := fuzz.NewConsumer(data)
+		targetStruct := &struct {
+			podLog types.PodLogRequest
+		}{}
+		err := fuzzConsumer.GenerateStruct(targetStruct)
+		if err != nil {
+			return
+		}
+		subscriberGraphql := graphql.NewSubscriberGql()
+		subscriberK8s := k8s.NewK8sSubscriber(subscriberGraphql)
+		_, err = subscriberK8s.CreatePodLog(targetStruct.podLog)
+		if err != nil {
+			t.Errorf("Unexpected error: %v", err)
+		}
+	})
+}
+
+func FuzzSendPodLogs(f *testing.F) {
+	f.Fuzz(func(t *testing.T, data []byte) {
+		fuzzConsumer := fuzz.NewConsumer(data)
+		targetStruct := &struct {
+			infraData map[string]string
+			podLog    types.PodLogRequest
+		}{}
+		err := fuzzConsumer.GenerateStruct(targetStruct)
+		if err != nil {
+			return
+		}
+		subscriberGraphql := graphql.NewSubscriberGql()
+		subscriberK8s := k8s.NewK8sSubscriber(subscriberGraphql)
+		subscriberK8s.SendPodLogs(targetStruct.infraData, targetStruct.podLog)
+	})
+}
+
+func FuzzGenerateLogPayload(f *testing.F) {
+	f.Fuzz(func(t *testing.T, data []byte) {
+		fuzzConsumer := fuzz.NewConsumer(data)
+		targetStruct := &struct {
+			cid       string
+			accessKey string
+			version   string
+			podLog    types.PodLogRequest
+		}{}
+		err := fuzzConsumer.GenerateStruct(targetStruct)
+		if err != nil {
+			return
+		}
+		subscriberGraphql := graphql.NewSubscriberGql()
+		subscriberK8s := k8s.NewK8sSubscriber(subscriberGraphql)
+		response, err := subscriberK8s.GenerateLogPayload(targetStruct.cid, targetStruct.accessKey, targetStruct.version, targetStruct.podLog)
+		if err != nil {
+			t.Errorf("Unexpected error: %v", err)
+		}
+		if response == nil {
+			t.Errorf("Returned payload is nil")
+		}
+
+	})
+}