events_test.go

package server_test

import (
	"context"
	"fmt"
	"testing"

	helmv2 "github.com/fluxcd/helm-controller/api/v2"
	kustomizev1 "github.com/fluxcd/kustomize-controller/api/v1"
	. "github.com/onsi/gomega"
	"google.golang.org/grpc/metadata"
	corev1 "k8s.io/api/core/v1"
	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/util/rand"
	"sigs.k8s.io/controller-runtime/pkg/client"

	pb "github.com/weaveworks/weave-gitops/pkg/api/core"
	"github.com/weaveworks/weave-gitops/pkg/kube"
)

func TestListEvents(t *testing.T) {
	g := NewGomegaWithT(t)

	ctx := context.Background()

	c := makeGRPCServer(ctx, k8sEnv.Rest, t)

	scheme, err := kube.CreateScheme()
	g.Expect(err).To(BeNil())

	k, err := client.New(k8sEnv.Rest, client.Options{
		Scheme: scheme,
	})
	g.Expect(err).NotTo(HaveOccurred())

	kustomizationObjectName := "my-kustomization"
	helmObjectName := "my-helmrelease"
	ns := newNamespace(ctx, k, g)

	// Kustomization
	kustomizeEvent := &corev1.Event{
		ObjectMeta: v1.ObjectMeta{
			Name:      fmt.Sprintf("%s.16da7d2e2c5b0930", kustomizationObjectName),
			Namespace: ns.Name,
		},
		InvolvedObject: corev1.ObjectReference{
			Kind:      kustomizev1.KustomizationKind,
			Namespace: ns.Name,
			Name:      kustomizationObjectName,
		},
		Source: corev1.EventSource{
			Component: "kustomize-controller",
		},
	}

	// HelmRelease
	helmEvent := &corev1.Event{
		ObjectMeta: v1.ObjectMeta{
			Name:      fmt.Sprintf("%s.16da7d2e2c5b0930", helmObjectName),
			Namespace: ns.Name,
		},
		InvolvedObject: corev1.ObjectReference{
			Kind:      helmv2.HelmReleaseKind,
			Namespace: ns.Name,
			Name:      helmObjectName,
		},
		Source: corev1.EventSource{
			Component: "helm-controller",
		},
	}

	// An event we don't care about. Shoud not show up in our response.
	otherEvent := &corev1.Event{
		ObjectMeta: v1.ObjectMeta{
			Name:      "someotherobject.16da7d2e2c5b0930",
			Namespace: ns.Name,
		},
		InvolvedObject: corev1.ObjectReference{
			Kind:      kustomizev1.KustomizationKind,
			Namespace: ns.Name,
			Name:      "someotherobject",
		},
		Source: corev1.EventSource{
			Component: "kustomize-controller",
		},
	}

	g.Expect(k.Create(ctx, kustomizeEvent)).To(Succeed())
	g.Expect(k.Create(ctx, helmEvent)).To(Succeed())
	g.Expect(k.Create(ctx, otherEvent)).To(Succeed())

	md := metadata.Pairs(MetadataUserKey, "anne", MetadataGroupsKey, "system:masters")
	outgoingCtx := metadata.NewOutgoingContext(ctx, md)

	// Get kustomization events
	res, err := c.ListEvents(outgoingCtx, &pb.ListEventsRequest{
		InvolvedObject: &pb.ObjectRef{
			Name:      kustomizationObjectName,
			Namespace: ns.Name,
			Kind:      kustomizev1.KustomizationKind,
		},
	})
	g.Expect(err).NotTo(HaveOccurred())

	g.Expect(res.Events).To(HaveLen(1))

	g.Expect(res.Events[0].Component).To(Equal(kustomizeEvent.Source.Component))

	// Get kustomization events, explicit cluster
	res, err = c.ListEvents(outgoingCtx, &pb.ListEventsRequest{
		InvolvedObject: &pb.ObjectRef{
			Name:        kustomizationObjectName,
			Namespace:   ns.Name,
			Kind:        kustomizev1.KustomizationKind,
			ClusterName: "Default",
		},
	})
	g.Expect(err).NotTo(HaveOccurred())

	g.Expect(res.Events).To(HaveLen(1))

	g.Expect(res.Events[0].Component).To(Equal(kustomizeEvent.Source.Component))

	// Get helmrelease events
	res, err = c.ListEvents(outgoingCtx, &pb.ListEventsRequest{
		InvolvedObject: &pb.ObjectRef{
			Name:      helmObjectName,
			Namespace: ns.Name,
			Kind:      helmv2.HelmReleaseKind,
		},
	})
	g.Expect(err).NotTo(HaveOccurred())

	g.Expect(res.Events).To(HaveLen(1))

	g.Expect(res.Events[0].Component).To(Equal(helmEvent.Source.Component))
}

func newNamespace(ctx context.Context, k client.Client, g *GomegaWithT) *corev1.Namespace {
	ns := &corev1.Namespace{}
	ns.Name = "kube-test-" + rand.String(5)

	g.Expect(k.Create(ctx, ns)).To(Succeed())

	return ns
}