Updated Testinfra binary to include one more command , so that block-sync can use it for copying files from default to pr namespace.

Signed-off-by: Kushal Shukla <kushalshukla110@gmail.com>

Author: kushalShukla-web

infra/infra.go

@@ -48,6 +48,9 @@ func main() {
 		PlaceHolder("service-account.json").
 		Short('a').
 		StringVar(&g.Auth)
+	k8sGKE.Command("copy", "Creating prombench-${PR_NUMBER} namespace and copy block-sync config and bucket-secret from default to prombench-${PR_NUMBER} namespace.").
+		Action(g.NewK8sProvider).
+		Action(g.CreateNamespace)
 
 	k8sGKE.Command("info", "gke info -v hashStable:COMMIT1 -v hashTesting:COMMIT2").
 		Action(g.GetDeploymentVars)
@@ -88,6 +91,10 @@ func main() {
 	k8sKIND := app.Command("kind", `Kubernetes In Docker (KIND) provider - https://kind.sigs.k8s.io/docs/user/quick-start/`).
 		Action(k.SetupDeploymentResources)
 
+	k8sKIND.Command("copy", "Creating prombench-${PR_NUMBER} namespace and copy block-sync config and bucket-secret from default to prombench-${PR_NUMBER} namespace.").
+		Action(k.NewK8sProvider).
+		Action(k.CreateNamespace)
+
 	k8sKIND.Command("info", "kind info -v hashStable:COMMIT1 -v hashTesting:COMMIT2").
 		Action(k.GetDeploymentVars)
 

pkg/provider/gke/gke.go

@@ -17,6 +17,9 @@ import (
 	"context"
 	"encoding/base64"
 	"fmt"
+	corev1 "k8s.io/api/core/v1"
+	k8serrors "k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"log"
 	"os"
 	"regexp"
@@ -576,6 +579,63 @@ func (c *GKE) NewK8sProvider(*kingpin.ParseContext) error {
 	return nil
 }
 
+func (c *GKE) CreateNamespace(*kingpin.ParseContext) error {
+	sourceNS := "default"
+	targetNS := "prombench-" + c.DeploymentVars["PR_NUMBER"]
+	configMapName := "blocksync-config"
+	secretName := "bucket-secret"
+
+	// Check if namespace exists
+	_, err := c.k8sProvider.Clt.CoreV1().Namespaces().Get(context.TODO(), targetNS, metav1.GetOptions{})
+	if err != nil {
+		if k8serrors.IsNotFound(err) {
+			ns := &corev1.Namespace{
+				ObjectMeta: metav1.ObjectMeta{
+					Name: targetNS,
+				},
+			}
+			_, err = c.k8sProvider.Clt.CoreV1().Namespaces().Create(context.TODO(), ns, metav1.CreateOptions{})
+			if err != nil {
+				return fmt.Errorf("error creating namespace: %w", err)
+			}
+		} else {
+			return fmt.Errorf("error checking namespace: %w", err)
+		}
+	}
+
+	// Copy ConfigMap
+	_, err = c.k8sProvider.Clt.CoreV1().ConfigMaps(targetNS).Get(context.TODO(), configMapName, metav1.GetOptions{})
+	if k8serrors.IsNotFound(err) {
+		cm, err := c.k8sProvider.Clt.CoreV1().ConfigMaps(sourceNS).Get(context.TODO(), configMapName, metav1.GetOptions{})
+		if err != nil {
+			return fmt.Errorf("error getting configmap: %w", err)
+		}
+		cm.ResourceVersion = ""
+		cm.Namespace = targetNS
+		_, err = c.k8sProvider.Clt.CoreV1().ConfigMaps(targetNS).Create(context.TODO(), cm, metav1.CreateOptions{})
+		if err != nil {
+			return fmt.Errorf("error creating configmap: %w", err)
+		}
+	}
+
+	// Copy Secret
+	_, err = c.k8sProvider.Clt.CoreV1().Secrets(targetNS).Get(context.TODO(), secretName, metav1.GetOptions{})
+	if k8serrors.IsNotFound(err) {
+		secret, err := c.k8sProvider.Clt.CoreV1().Secrets(sourceNS).Get(context.TODO(), secretName, metav1.GetOptions{})
+		if err != nil {
+			return fmt.Errorf("error getting secret: %w", err)
+		}
+		secret.ResourceVersion = ""
+		secret.Namespace = targetNS
+		_, err = c.k8sProvider.Clt.CoreV1().Secrets(targetNS).Create(context.TODO(), secret, metav1.CreateOptions{})
+		if err != nil {
+			return fmt.Errorf("error creating secret in target NS: %w", err)
+		}
+	}
+
+	return nil
+}
+
 // ResourceApply calls k8s.ResourceApply to apply the k8s objects in the manifest files.
 func (c *GKE) ResourceApply(*kingpin.ParseContext) error {
 	if err := c.k8sProvider.ResourceApply(c.k8sResources); err != nil {