adoption_controller_test.go

package operators

import (
	"context"
	"fmt"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
	"github.com/operator-framework/operator-lifecycle-manager/pkg/controller/install"
	appsv1 "k8s.io/api/apps/v1"
	corev1 "k8s.io/api/core/v1"
	rbacv1 "k8s.io/api/rbac/v1"
	apiextensionsv1 "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
	apierrors "k8s.io/apimachinery/pkg/api/errors"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/tools/reference"
	apiregistrationv1 "k8s.io/kube-aggregator/pkg/apis/apiregistration/v1"
	"sigs.k8s.io/controller-runtime/pkg/client"

	operatorsv1alpha1 "github.com/operator-framework/api/pkg/operators/v1alpha1"
	"github.com/operator-framework/operator-lifecycle-manager/pkg/controller/operators/decorators"
	"github.com/operator-framework/operator-lifecycle-manager/pkg/lib/ownerutil"
	"github.com/operator-framework/operator-lifecycle-manager/pkg/lib/testobj"
)

var _ = Describe("Adoption Controller", func() {
	var (
		ctx context.Context
	)

	BeforeEach(func() {
		ctx = context.Background()
	})

	Describe("Component label generation", func() {
		var (
			created []client.Object
		)

		BeforeEach(func() {
			created = []client.Object{}
		})

		JustAfterEach(func() {
			for _, obj := range created {
				Eventually(func() error {
					err := k8sClient.Delete(ctx, obj)
					if apierrors.IsNotFound(err) {
						return nil
					}

					return err
				}).Should(Succeed())
			}
		})

		Context("a subscription", func() {
			var (
				ns  *corev1.Namespace
				sub *operatorsv1alpha1.Subscription
			)

			BeforeEach(func() {
				ns = &corev1.Namespace{}
				ns.SetName(genName("operators-"))

				Eventually(func() error {
					return k8sClient.Create(ctx, ns)
				}, timeout, interval).Should(Succeed())
				created = append(created, ns)
			})

			Context("with a package", func() {
				var (
					componentLabelKey string
					installed         *operatorsv1alpha1.ClusterServiceVersion
				)

				BeforeEach(func() {
					sub = &operatorsv1alpha1.Subscription{
						Spec: &operatorsv1alpha1.SubscriptionSpec{
							Package: "poultry",
						},
					}
					sub.SetNamespace(ns.GetName())
					sub.SetName(sub.Spec.Package)

					Eventually(func() error {
						return k8sClient.Create(ctx, sub)
					}, timeout, interval).Should(Succeed())
					created = append(created, sub)

					// Set the Subscription's status separately
					Eventually(func() error {
						if err := k8sClient.Get(ctx, testobj.NamespacedName(sub), sub); err != nil {
							return err
						}
						sub.Status = operatorsv1alpha1.SubscriptionStatus{
							InstalledCSV: "turkey",
							LastUpdated:  metav1.Now(),
						}
						return k8sClient.Status().Update(ctx, sub)
					}, timeout, interval).Should(Succeed())

					componentLabelKey = fmt.Sprintf("%s%s.%s", decorators.ComponentLabelKeyPrefix, sub.Spec.Package, sub.GetNamespace())
				})

				Context("that references an existing installplan", func() {
					var (
						ip *operatorsv1alpha1.InstallPlan
					)

					BeforeEach(func() {
						ip = fixtures.Fill(&operatorsv1alpha1.InstallPlan{}).(*operatorsv1alpha1.InstallPlan)
						ip.SetNamespace(sub.GetNamespace())
						ip.SetName(genName("poultry-"))

						Eventually(func() error {
							owned := testobj.WithOwner(sub, ip)
							return k8sClient.Create(ctx, owned)
						}).Should(Succeed())
						created = append(created, ip)

						ref, err := reference.GetReference(scheme, ip)
						Expect(err).ToNot(HaveOccurred())

						// Set the Subscription's status separately
						status := sub.DeepCopy().Status
						status.InstallPlanRef = ref
						Eventually(func() error {
							if err := k8sClient.Get(ctx, testobj.NamespacedName(sub), sub); err != nil {
								return err
							}
							sub.Status = status

							return k8sClient.Status().Update(ctx, sub)
						}).Should(Succeed())
					})

					Context("and has other, non-latest, adopted installplans", func() {
						var (
							ips []*operatorsv1alpha1.InstallPlan
						)

						BeforeEach(func() {
							for i := 0; i < 4; i++ {
								ip := fixtures.Fill(&operatorsv1alpha1.InstallPlan{}).(*operatorsv1alpha1.InstallPlan)
								ip.SetNamespace(sub.GetNamespace())
								ip.SetName(genName(""))
								ip.SetLabels(map[string]string{
									componentLabelKey: "",
								})

								Eventually(func() error {
									return k8sClient.Create(ctx, ip)
								}).Should(Succeed())

								created = append(created, ip)
								ips = append(ips, ip)
							}

						})

						Specify("correct component labels", func() {
							installPlan := ip.DeepCopy()
							Eventually(func() (map[string]string, error) {
								err := k8sClient.Get(ctx, testobj.NamespacedName(ip), installPlan)
								return installPlan.GetLabels(), err
							}).Should(HaveKey(componentLabelKey))

							for _, ip := range ips {
								Eventually(func() (map[string]string, error) {
									err := k8sClient.Get(ctx, testobj.NamespacedName(ip), ip)
									return ip.GetLabels(), err
								}, timeout, interval).ShouldNot(HaveKey(componentLabelKey))
							}

						})

					})
				})

				Context("that has an existing installed csv", func() {
					var (
						providedCRD *apiextensionsv1.CustomResourceDefinition
					)

					BeforeEach(func() {
						providedCRD = fixtures.Fill(&apiextensionsv1.CustomResourceDefinition{}).(*apiextensionsv1.CustomResourceDefinition)
						installed = &operatorsv1alpha1.ClusterServiceVersion{
							Spec: operatorsv1alpha1.ClusterServiceVersionSpec{
								CustomResourceDefinitions: operatorsv1alpha1.CustomResourceDefinitions{
									Owned: []operatorsv1alpha1.CRDDescription{
										{
											Name:    providedCRD.GetName(),
											Kind:    providedCRD.Spec.Names.Kind,
											Version: providedCRD.Spec.Versions[0].Name,
										},
									},
								},
								InstallStrategy: operatorsv1alpha1.NamedInstallStrategy{
									StrategyName: operatorsv1alpha1.InstallStrategyNameDeployment,
									StrategySpec: operatorsv1alpha1.StrategyDetailsDeployment{
										DeploymentSpecs: []operatorsv1alpha1.StrategyDeploymentSpec{},
									},
								},
							},
						}
						installed.SetNamespace(ns.GetName())
						installed.SetName(sub.Status.InstalledCSV)

						Eventually(func() error {
							return k8sClient.Create(ctx, installed)
						}, timeout, interval).Should(Succeed())
						created = append(created, installed)
					})

					Context("that has no resources owned by the installed csv", func() {
						Specify("a component label", func() {
							Eventually(func() (map[string]string, error) {
								latest := &operatorsv1alpha1.ClusterServiceVersion{}
								err := k8sClient.Get(ctx, testobj.NamespacedName(installed), latest)

								return latest.GetLabels(), err
							}, timeout, interval).Should(HaveKey(componentLabelKey))
						})
					})

					Context("with an existing provided CRD", func() {

						BeforeEach(func() {
							Eventually(func() error {
								return k8sClient.Create(ctx, providedCRD)
							}, timeout, interval).Should(Succeed())
							created = append(created, providedCRD)

							Eventually(func() (map[string]string, error) {
								latest := &apiextensionsv1.CustomResourceDefinition{}
								err := k8sClient.Get(ctx, testobj.NamespacedName(providedCRD), latest)

								return latest.GetLabels(), err
							}, timeout, interval).Should(HaveKey(componentLabelKey))
						})

						Context("when its component label is removed", func() {

							BeforeEach(func() {
								Eventually(func() error {
									latest := &apiextensionsv1.CustomResourceDefinition{}
									if err := k8sClient.Get(ctx, testobj.NamespacedName(providedCRD), latest); err != nil {
										return err
									}

									if len(latest.GetLabels()) == 0 {
										return nil
									}
									delete(latest.GetLabels(), componentLabelKey)

									return k8sClient.Update(ctx, latest)
								}, timeout, interval).Should(Succeed())
							})

							It("should be relabelled", func() {
								Eventually(func() (map[string]string, error) {
									latest := &apiextensionsv1.CustomResourceDefinition{}
									err := k8sClient.Get(ctx, testobj.NamespacedName(providedCRD), latest)

									return latest.GetLabels(), err
								}, timeout, interval).Should(HaveKey(componentLabelKey))
							})
						})
					})

					Context("that has resources owned by the installed csv", func() {
						var (
							components []testobj.RuntimeMetaObject
						)

						BeforeEach(func() {
							Eventually(func() error {
								return k8sClient.Get(ctx, testobj.NamespacedName(installed), installed)
							}, timeout, interval).Should(Succeed())

							namespace := installed.GetNamespace()
							ownerLabels := map[string]string{
								ownerutil.OwnerKind:         operatorsv1alpha1.ClusterServiceVersionKind,
								ownerutil.OwnerNamespaceKey: namespace,
								ownerutil.OwnerKey:          installed.GetName(),
							}
							components = []testobj.RuntimeMetaObject{
								testobj.WithOwner(
									installed,
									testobj.WithNamespace(
										namespace,
										fixtures.Fill(&appsv1.Deployment{}),
									),
								),
								testobj.WithOwner(
									installed,
									testobj.WithNamespace(
										namespace,
										fixtures.Fill(&corev1.Service{}),
									),
								),
								testobj.WithOwner(
									installed,
									testobj.WithNamespace(
										namespace,
										fixtures.Fill(&corev1.ServiceAccount{}),
									),
								),
								testobj.WithOwner(
									installed,
									testobj.WithNamespace(
										namespace,
										fixtures.Fill(&corev1.Secret{}),
									),
								),
								testobj.WithOwner(
									installed,
									testobj.WithNamespace(
										namespace,
										fixtures.Fill(&corev1.ConfigMap{}),
									),
								),
								testobj.WithOwner(
									installed,
									testobj.WithNamespace(
										namespace,
										fixtures.Fill(&rbacv1.Role{}),
									),
								),
								testobj.WithOwner(
									installed,
									testobj.WithNamespace(
										namespace,
										fixtures.Fill(&rbacv1.RoleBinding{}),
									),
								),
								testobj.WithLabels(
									ownerLabels,
									fixtures.Fill(&rbacv1.ClusterRole{}),
								),
								testobj.WithLabels(
									ownerLabels,
									fixtures.Fill(&rbacv1.ClusterRoleBinding{}),
								),
								testobj.WithLabels(
									ownerLabels,
									fixtures.Fill(&apiregistrationv1.APIService{}),
								),
							}
							for _, component := range components {
								labels := component.GetLabels()
								if labels == nil {
									labels = map[string]string{}
								}
								labels[install.OLMManagedLabelKey] = install.OLMManagedLabelValue
								component.SetLabels(labels)
								Eventually(func() error {
									return k8sClient.Create(ctx, component)
								}, timeout, interval).Should(Succeed())
								created = append(created, component)
							}
						})

						Specify("a component label", func() {
							for _, component := range components {
								Eventually(func() (map[string]string, error) {
									err := k8sClient.Get(ctx, testobj.NamespacedName(component), component)
									return component.GetLabels(), err
								}, timeout, interval).Should(HaveKey(componentLabelKey))
							}
						})

					})

				})
			})

		})
	})
})