From 461276004b6f4f3759b0702b07440fc032e4fa91 Mon Sep 17 00:00:00 2001
From: natasha41575 <natashasarkar@google.com>
Date: Wed, 11 Oct 2023 20:16:11 +0000
Subject: [PATCH] add test

Signed-off-by: natasha41575 <natashasarkar@google.com>
---
 porch/test/e2e/e2e_test.go | 100 +++++++++++++++++++++++++++++++++++++
 1 file changed, 100 insertions(+)

diff --git a/porch/test/e2e/e2e_test.go b/porch/test/e2e/e2e_test.go
index e751d21fe..aa939014d 100644
--- a/porch/test/e2e/e2e_test.go
+++ b/porch/test/e2e/e2e_test.go
@@ -1115,6 +1115,106 @@ func (t *PorchSuite) TestDeleteAndRecreate(ctx context.Context) {
 	t.Logf("successfully recreated package revision %q", packageName)
 }
 
+func (t *PorchSuite) TestDeleteFromMain(ctx context.Context) {
+	const (
+		repository        = "delete-main"
+		packageNameFirst  = "test-delete-main-first"
+		packageNameSecond = "test-delete-main-second"
+		workspace         = "workspace"
+	)
+
+	// Register the repository
+	t.registerMainGitRepositoryF(ctx, repository)
+
+	// Create the first draft package
+	createdFirst := t.createPackageDraftF(ctx, repository, packageNameFirst, workspace)
+
+	// Check the package exists
+	var pkgFirst porchapi.PackageRevision
+	t.mustExist(ctx, client.ObjectKey{Namespace: t.namespace, Name: createdFirst.Name}, &pkgFirst)
+
+	// Propose the package revision to be finalized
+	pkgFirst.Spec.Lifecycle = porchapi.PackageRevisionLifecycleProposed
+	t.UpdateF(ctx, &pkgFirst)
+
+	pkgFirst.Spec.Lifecycle = porchapi.PackageRevisionLifecyclePublished
+	t.UpdateApprovalF(ctx, &pkgFirst, metav1.UpdateOptions{})
+
+	t.mustExist(ctx, client.ObjectKey{Namespace: t.namespace, Name: createdFirst.Name}, &pkgFirst)
+
+	// Create the second draft package
+	createdSecond := t.createPackageDraftF(ctx, repository, packageNameSecond, workspace)
+
+	// Check the package exists
+	var pkgSecond porchapi.PackageRevision
+	t.mustExist(ctx, client.ObjectKey{Namespace: t.namespace, Name: createdSecond.Name}, &pkgSecond)
+
+	// Propose the package revision to be finalized
+	pkgSecond.Spec.Lifecycle = porchapi.PackageRevisionLifecycleProposed
+	t.UpdateF(ctx, &pkgSecond)
+
+	pkgSecond.Spec.Lifecycle = porchapi.PackageRevisionLifecyclePublished
+	t.UpdateApprovalF(ctx, &pkgSecond, metav1.UpdateOptions{})
+
+	t.mustExist(ctx, client.ObjectKey{Namespace: t.namespace, Name: createdSecond.Name}, &pkgSecond)
+
+	// We need to wait for the sync for the "main" revisions to get created
+	time.Sleep(75 * time.Second)
+
+	var list porchapi.PackageRevisionList
+	t.ListE(ctx, &list, client.InNamespace(t.namespace))
+
+	var firstPkgRevFromMain porchapi.PackageRevision
+	var secondPkgRevFromMain porchapi.PackageRevision
+
+	for _, pkgrev := range list.Items {
+		if pkgrev.Spec.PackageName == packageNameFirst && pkgrev.Spec.Revision == "main" {
+			firstPkgRevFromMain = pkgrev
+		}
+		if pkgrev.Spec.PackageName == packageNameSecond && pkgrev.Spec.Revision == "main" {
+			secondPkgRevFromMain = pkgrev
+		}
+	}
+
+	// Propose deletion of both main packages
+	firstPkgRevFromMain.Spec.Lifecycle = porchapi.PackageRevisionLifecycleDeletionProposed
+	t.UpdateApprovalF(ctx, &firstPkgRevFromMain, metav1.UpdateOptions{})
+	secondPkgRevFromMain.Spec.Lifecycle = porchapi.PackageRevisionLifecycleDeletionProposed
+	t.UpdateApprovalF(ctx, &secondPkgRevFromMain, metav1.UpdateOptions{})
+
+	// Delete the first package revision from main
+	t.DeleteE(ctx, &porchapi.PackageRevision{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: t.namespace,
+			Name:      firstPkgRevFromMain.Name,
+		},
+	})
+
+	// We need to wait for the sync
+	time.Sleep(75 * time.Second)
+
+	// Delete the second package revision from main
+	t.DeleteE(ctx, &porchapi.PackageRevision{
+		ObjectMeta: metav1.ObjectMeta{
+			Namespace: t.namespace,
+			Name:      secondPkgRevFromMain.Name,
+		},
+	})
+
+	// Propose and delete the original package revisions (cleanup)
+	t.ListE(ctx, &list, client.InNamespace(t.namespace))
+	for _, pkgrev := range list.Items {
+		pkgrev.Spec.Lifecycle = porchapi.PackageRevisionLifecycleDeletionProposed
+		t.UpdateApprovalF(ctx, &pkgrev, metav1.UpdateOptions{})
+		t.DeleteE(ctx, &porchapi.PackageRevision{
+			ObjectMeta: metav1.ObjectMeta{
+				Namespace: t.namespace,
+				Name:      pkgrev.Name,
+			},
+		})
+	}
+}
+
 func (t *PorchSuite) TestCloneLeadingSlash(ctx context.Context) {
 	const (
 		repository  = "clone-ls"