From 6fee8986a985f4e1df8f1249e4cf6b20483c08ac Mon Sep 17 00:00:00 2001
From: Mikalai Radchuk <mradchuk@redhat.com>
Date: Thu, 25 Apr 2024 16:56:25 +0200
Subject: [PATCH] Add `ClusterExtension` package uniqueness test

Signed-off-by: Mikalai Radchuk <mradchuk@redhat.com>
---
 test/e2e/cluster_extension_admission_test.go | 75 ++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100644 test/e2e/cluster_extension_admission_test.go

diff --git a/test/e2e/cluster_extension_admission_test.go b/test/e2e/cluster_extension_admission_test.go
new file mode 100644
index 000000000..cc6b9eaea
--- /dev/null
+++ b/test/e2e/cluster_extension_admission_test.go
@@ -0,0 +1,75 @@
+package e2e
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"k8s.io/apimachinery/pkg/api/errors"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	"k8s.io/apimachinery/pkg/types"
+
+	ocv1alpha1 "github.com/operator-framework/operator-controller/api/v1alpha1"
+)
+
+func TestClusterExtensionPackageUniqueness(t *testing.T) {
+	ctx := context.Background()
+
+	deleteClusterExtension := func(clusterExtension *ocv1alpha1.ClusterExtension) {
+		require.NoError(t, c.Delete(ctx, clusterExtension))
+		require.Eventually(t, func() bool {
+			err := c.Get(ctx, types.NamespacedName{Name: clusterExtension.Name}, &ocv1alpha1.ClusterExtension{})
+			return errors.IsNotFound(err)
+		}, pollDuration, pollInterval)
+	}
+
+	const firstResourceName = "test-extension-first"
+	const firstResourcePackageName = "package1"
+
+	t.Log("create first resource")
+	clusterExtension1 := &ocv1alpha1.ClusterExtension{
+		ObjectMeta: metav1.ObjectMeta{
+			Name: firstResourceName,
+		},
+		Spec: ocv1alpha1.ClusterExtensionSpec{
+			PackageName: firstResourcePackageName,
+		},
+	}
+	require.NoError(t, c.Create(ctx, clusterExtension1))
+	defer deleteClusterExtension(clusterExtension1)
+
+	t.Log("create second resource with the same package as the first resource")
+	clusterExtension2 := &ocv1alpha1.ClusterExtension{
+		ObjectMeta: metav1.ObjectMeta{
+			GenerateName: "test-extension-",
+		},
+		Spec: ocv1alpha1.ClusterExtensionSpec{
+			PackageName: firstResourcePackageName,
+		},
+	}
+	err := c.Create(ctx, clusterExtension2)
+	require.ErrorContains(t, err, fmt.Sprintf("Package %q is already installed via ClusterExtension %q", firstResourcePackageName, firstResourceName))
+
+	t.Log("create second resource with different package")
+	clusterExtension2 = &ocv1alpha1.ClusterExtension{
+		ObjectMeta: metav1.ObjectMeta{
+			GenerateName: "test-extension-",
+		},
+		Spec: ocv1alpha1.ClusterExtensionSpec{
+			PackageName: "package2",
+		},
+	}
+	require.NoError(t, c.Create(ctx, clusterExtension2))
+	defer deleteClusterExtension(clusterExtension2)
+
+	t.Log("update second resource with package which already exists on the cluster")
+	clusterExtension2.Spec.PackageName = firstResourcePackageName
+	err = c.Update(ctx, clusterExtension2)
+	require.ErrorContains(t, err, fmt.Sprintf("Package %q is already installed via ClusterExtension %q", firstResourcePackageName, firstResourceName))
+
+	t.Log("update second resource with package which does not exist on the cluster")
+	require.NoError(t, c.Get(ctx, types.NamespacedName{Name: clusterExtension2.Name}, clusterExtension2))
+	clusterExtension2.Spec.PackageName = "package3"
+	require.NoError(t, c.Update(ctx, clusterExtension2))
+}