Expose logs/resources after test run

Signed-off-by: Joaquim Moreno Prusi <joaquim@redhat.com>

Author: jmprusi

Makefile

@@ -12,6 +12,10 @@ export RUKPAK_VERSION=$(shell go list -mod=mod -m -f "{{.Version}}" github.com/o
 export WAIT_TIMEOUT ?= 60s
 IMG?=$(IMAGE_REPO):$(IMAGE_TAG)
 
+# ARTIFACT_PATH is the absolute path to the directory where the operator-controller e2e tests will store the artifacts
+# for example: ARTIFACT_PATH=/tmp/artifacts make test 
+export ARTIFACT_PATH ?= 
+
 OPERATOR_CONTROLLER_NAMESPACE ?= operator-controller-system
 KIND_CLUSTER_NAME ?= operator-controller
 

go.mod

@@ -11,6 +11,8 @@ require (
 	github.com/operator-framework/operator-registry v1.26.3
 	github.com/operator-framework/rukpak v0.12.0
 	go.uber.org/zap v1.24.0
+	gopkg.in/yaml.v2 v2.4.0
+	k8s.io/api v0.26.1
 	k8s.io/apimachinery v0.26.1
 	k8s.io/client-go v0.26.1
 	k8s.io/utils v0.0.0-20221128185143-99ec85e7a448
@@ -67,9 +69,7 @@ require (
 	google.golang.org/appengine v1.6.7 // indirect
 	google.golang.org/protobuf v1.28.1 // indirect
 	gopkg.in/inf.v0 v0.9.1 // indirect
-	gopkg.in/yaml.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
-	k8s.io/api v0.26.1 // indirect
 	k8s.io/apiextensions-apiserver v0.26.1 // indirect
 	k8s.io/component-base v0.26.1 // indirect
 	k8s.io/klog/v2 v2.80.1 // indirect

test/e2e/e2e_suite_test.go

@@ -6,6 +6,8 @@ import (
 
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
+	appsv1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/client-go/rest"
 	ctrl "sigs.k8s.io/controller-runtime"
@@ -40,6 +42,13 @@ var _ = BeforeSuite(func() {
 
 	err = catalogd.AddToScheme(scheme)
 	Expect(err).ToNot(HaveOccurred())
+
+	err = appsv1.AddToScheme(scheme)
+	Expect(err).ToNot(HaveOccurred())
+
+	err = corev1.AddToScheme(scheme)
+	Expect(err).ToNot(HaveOccurred())
+
 	c, err = client.New(cfg, client.Options{Scheme: scheme})
 	Expect(err).To(Not(HaveOccurred()))
 })

test/e2e/install_test.go

@@ -3,22 +3,34 @@ package e2e
 import (
 	"context"
 	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
 	"time"
 
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 	catalogd "github.com/operator-framework/catalogd/pkg/apis/core/v1beta1"
 	operatorv1alpha1 "github.com/operator-framework/operator-controller/api/v1alpha1"
 	rukpakv1alpha1 "github.com/operator-framework/rukpak/api/v1alpha1"
+	"gopkg.in/yaml.v2"
+	appsv1 "k8s.io/api/apps/v1"
+	corev1 "k8s.io/api/core/v1"
+	v1 "k8s.io/api/core/v1"
 	apimeta "k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/types"
 	"k8s.io/apimachinery/pkg/util/rand"
+	kubeclient "k8s.io/client-go/kubernetes"
+	"k8s.io/utils/env"
+	"sigs.k8s.io/controller-runtime/pkg/client"
 )
 
 const (
 	defaultTimeout = 30 * time.Second
 	defaultPoll    = 1 * time.Second
+	artifactName   = "operator-controller-e2e"
 )
 
 var _ = Describe("Operator Install", func() {
@@ -104,10 +116,174 @@ var _ = Describe("Operator Install", func() {
 
 		})
 		AfterEach(func() {
+			if basePath := env.GetString("ARTIFACT_PATH", ""); basePath != "" && CurrentSpecReport().Failed() {
+				// get all the artifacts from the test run and save them to the artifact path
+				getArtifactsOutput(ctx, basePath)
+			}
 			err := c.Delete(ctx, operatorCatalog)
 			Expect(err).ToNot(HaveOccurred())
 			err = c.Delete(ctx, operator)
 			Expect(err).ToNot(HaveOccurred())
 		})
 	})
 })
+
+// getArtifactsOutput gets all the artifacts from the test run and saves them to the artifact path.
+// right now it will save:
+// - operators
+// - pods logs
+// - deployments
+// - bundle
+// - bundledeployments
+// - catalogsources
+
+func getArtifactsOutput(ctx context.Context, basePath string) {
+	kubeClient, err := kubeclient.NewForConfig(cfg)
+	Expect(err).To(Not(HaveOccurred()))
+
+	// sanitize the artifact name for use as a directory name
+	testName := strings.ReplaceAll(strings.ToLower(CurrentSpecReport().LeafNodeText), " ", "-")
+	// Get the test description and sanitize it for use as a directory name
+	artifactPath := filepath.Join(basePath, artifactName, fmt.Sprint(time.Now().UnixNano()), testName)
+
+	// Create the full artifact path
+	err = os.MkdirAll(artifactPath, 0755)
+	Expect(err).To(Not(HaveOccurred()))
+
+	// Get all namespaces
+	namespaces := corev1.NamespaceList{}
+	if err := c.List(ctx, &namespaces); err != nil {
+		GinkgoWriter.Printf("Failed to list namespaces %w", err)
+	}
+
+	// get all operators save them to the artifact path.
+	operators := operatorv1alpha1.OperatorList{}
+	if err := c.List(ctx, &operators, client.InNamespace("")); err != nil {
+		GinkgoWriter.Printf("Failed to list operators %w", err)
+	}
+	for _, operator := range operators.Items {
+		// Save operator to artifact path
+		operatorYaml, err := yaml.Marshal(operator)
+		if err != nil {
+			GinkgoWriter.Printf("Failed to marshal operator %w", err)
+			continue
+		}
+		if err := os.WriteFile(filepath.Join(artifactPath, operator.Name+"-operator.yaml"), operatorYaml, 0644); err != nil {
+			GinkgoWriter.Printf("Failed to write operator to file %w", err)
+		}
+	}
+
+	// get all catalogsources save them to the artifact path.
+	catalogsources := catalogd.CatalogList{}
+	if err := c.List(ctx, &catalogsources, client.InNamespace("")); err != nil {
+		GinkgoWriter.Printf("Failed to list catalogsources %w", err)
+	}
+	for _, catalogsource := range catalogsources.Items {
+		// Save catalogsource to artifact path
+		catalogsourceYaml, err := yaml.Marshal(catalogsource)
+		if err != nil {
+			GinkgoWriter.Printf("Failed to marshal catalogsource %w", err)
+			continue
+		}
+		if err := os.WriteFile(filepath.Join(artifactPath, catalogsource.Name+"-catalogsource.yaml"), catalogsourceYaml, 0644); err != nil {
+			GinkgoWriter.Printf("Failed to write catalogsource to file %w", err)
+		}
+	}
+
+	// Get all Bundles in the namespace and save them to the artifact path.
+	bundles := rukpakv1alpha1.BundleList{}
+	if err := c.List(ctx, &bundles, client.InNamespace("")); err != nil {
+		GinkgoWriter.Printf("Failed to list bundles %w", err)
+	}
+	for _, bundle := range bundles.Items {
+		// Save bundle to artifact path
+		bundleYaml, err := yaml.Marshal(bundle)
+		if err != nil {
+			GinkgoWriter.Printf("Failed to marshal bundle %w", err)
+			continue
+		}
+		if err := os.WriteFile(filepath.Join(artifactPath, bundle.Name+"-bundle.yaml"), bundleYaml, 0644); err != nil {
+			GinkgoWriter.Printf("Failed to write bundle to file %w", err)
+		}
+	}
+
+	// Get all BundleDeployments in the namespace and save them to the artifact path.
+	bundleDeployments := rukpakv1alpha1.BundleDeploymentList{}
+	if err := c.List(ctx, &bundleDeployments, client.InNamespace("")); err != nil {
+		GinkgoWriter.Printf("Failed to list bundleDeployments %w", err)
+	}
+	for _, bundleDeployment := range bundleDeployments.Items {
+		// Save bundleDeployment to artifact path
+		bundleDeploymentYaml, err := yaml.Marshal(bundleDeployment)
+		if err != nil {
+			GinkgoWriter.Printf("Failed to marshal bundleDeployment %w", err)
+			continue
+		}
+		if err := os.WriteFile(filepath.Join(artifactPath, bundleDeployment.Name+"-bundleDeployment.yaml"), bundleDeploymentYaml, 0644); err != nil {
+			GinkgoWriter.Printf("Failed to write bundleDeployment to file %w", err)
+		}
+	}
+
+	for _, namespace := range namespaces.Items {
+		// let's ignore kube-* namespaces.
+		if strings.Contains(namespace.Name, "kube-") {
+			continue
+		}
+
+		namespacedArtifactPath := filepath.Join(artifactPath, namespace.Name)
+		if err := os.Mkdir(namespacedArtifactPath, 0755); err != nil {
+			GinkgoWriter.Printf("Failed to create namespaced artifact path %w", err)
+			continue
+		}
+
+		// get all deployments in the namespace and save them to the artifact path.
+		deployments := appsv1.DeploymentList{}
+		if err := c.List(ctx, &deployments, client.InNamespace(namespace.Name)); err != nil {
+			GinkgoWriter.Printf("Failed to list deployments %w in namespace: %q", err, namespace.Name)
+			continue
+		}
+
+		for _, deployment := range deployments.Items {
+			// Save deployment to artifact path
+			deploymentYaml, err := yaml.Marshal(deployment)
+			if err != nil {
+				GinkgoWriter.Printf("Failed to marshal deployment %w", err)
+				continue
+			}
+			if err := os.WriteFile(filepath.Join(namespacedArtifactPath, deployment.Name+"-deployment.yaml"), deploymentYaml, 0644); err != nil {
+				GinkgoWriter.Printf("Failed to write deployment to file %w", err)
+			}
+		}
+
+		// Get logs from all pods in all namespaces
+		pods := corev1.PodList{}
+		if err := c.List(ctx, &pods, client.InNamespace(namespace.Name)); err != nil {
+			GinkgoWriter.Printf("Failed to list pods %w in namespace: %q", err, namespace.Name)
+		}
+		for _, pod := range pods.Items {
+			if pod.Status.Phase != v1.PodRunning && pod.Status.Phase != v1.PodSucceeded && pod.Status.Phase != v1.PodFailed {
+				continue
+			}
+			for _, container := range pod.Spec.Containers {
+				logs, err := kubeClient.CoreV1().Pods(namespace.Name).GetLogs(pod.Name, &v1.PodLogOptions{Container: container.Name}).Stream(ctx)
+				if err != nil {
+					GinkgoWriter.Printf("Failed to get logs for pod %q in namespace %q: %w", pod.Name, namespace.Name, err)
+					continue
+				}
+				defer logs.Close()
+
+				outFile, err := os.Create(filepath.Join(namespacedArtifactPath, pod.Name+"-"+container.Name+"-logs.txt"))
+				if err != nil {
+					GinkgoWriter.Printf("Failed to create file for pod %q in namespace %q: %w", pod.Name, namespace.Name, err)
+					continue
+				}
+				defer outFile.Close()
+
+				if _, err := io.Copy(outFile, logs); err != nil {
+					GinkgoWriter.Printf("Failed to copy logs for pod %q in namespace %q: %w", pod.Name, namespace.Name, err)
+					continue
+				}
+			}
+		}
+	}
+}