From a4e2414ce27189e69f2dbb345cddaf5654da8b00 Mon Sep 17 00:00:00 2001
From: Alex Kattathra Johnson <alex.kattathra.johnson@gmail.com>
Date: Tue, 17 Dec 2024 09:22:41 -0600
Subject: [PATCH] Add tests to test TakeOwnership

Signed-off-by: Alex Kattathra Johnson <alex.kattathra.johnson@gmail.com>
---
 pkg/action/install_test.go | 43 ++++++++++++++++++++++++++++++++++++++
 pkg/kube/fake/fake.go      |  6 +++++-
 2 files changed, 48 insertions(+), 1 deletion(-)

diff --git a/pkg/action/install_test.go b/pkg/action/install_test.go
index d11b04011..09b43e9e5 100644
--- a/pkg/action/install_test.go
+++ b/pkg/action/install_test.go
@@ -798,3 +798,46 @@ func TestInstallWithSystemLabels(t *testing.T) {
 
 	is.Equal(fmt.Errorf("user supplied labels contains system reserved label name. System labels: %+v", driver.GetSystemLabels()), err)
 }
+
+func TestInstallWithTakeOwnershipOption(t *testing.T) {
+	is := assert.New(t)
+	req := require.New(t)
+
+	instAction := installAction(t)
+	instAction.ClientOnly = false
+	instAction.IsUpgrade = false
+	instAction.TakeOwnership = true
+	failer := instAction.cfg.KubeClient.(*kubefake.FailingKubeClient)
+	failer.BuildDummy = true
+	instAction.cfg.KubeClient = failer
+
+	vals := map[string]interface{}{}
+	ctx, done := context.WithCancel(context.Background())
+	res, err := instAction.RunWithContext(ctx, buildChart(), vals)
+	if err != nil {
+		t.Fatalf("Failed install: %s", err)
+	}
+	is.Equal(res.Name, "test-install-release", "Expected release name.")
+	is.Equal(res.Namespace, "spaced")
+
+	rel, err := instAction.cfg.Releases.Get(res.Name, res.Version)
+	is.NoError(err)
+
+	is.Len(rel.Hooks, 1)
+	is.Equal(rel.Hooks[0].Manifest, manifestWithHook)
+	is.Equal(rel.Hooks[0].Events[0], release.HookPostInstall)
+	is.Equal(rel.Hooks[0].Events[1], release.HookPreDelete, "Expected event 0 is pre-delete")
+
+	is.NotEqual(len(res.Manifest), 0)
+	is.NotEqual(len(rel.Manifest), 0)
+	is.Contains(rel.Manifest, "---\n# Source: hello/templates/hello\nhello: world")
+	is.Equal(rel.Info.Description, "Install complete")
+
+	// Detecting previous bug where context termination after successful release
+	// caused release to fail.
+	done()
+	time.Sleep(time.Millisecond * 100)
+	lastRelease, err := instAction.cfg.Releases.Last(rel.Name)
+	req.NoError(err)
+	is.Equal(lastRelease.Info.Status, release.StatusDeployed)
+}
diff --git a/pkg/kube/fake/fake.go b/pkg/kube/fake/fake.go
index 267020d57..391c56075 100644
--- a/pkg/kube/fake/fake.go
+++ b/pkg/kube/fake/fake.go
@@ -22,8 +22,10 @@ import (
 	"time"
 
 	v1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/api/meta"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/apimachinery/pkg/runtime"
+	"k8s.io/apimachinery/pkg/runtime/schema"
 	"k8s.io/cli-runtime/pkg/resource"
 
 	"helm.sh/helm/v3/pkg/kube"
@@ -153,8 +155,10 @@ func createDummyResourceList() kube.ResourceList {
 	var resInfo resource.Info
 	resInfo.Name = "dummyName"
 	resInfo.Namespace = "dummyNamespace"
+	resInfo.Mapping = &meta.RESTMapping{
+		Resource: schema.GroupVersionResource{Version: "v1", Resource: "pods"},
+	}
 	var resourceList kube.ResourceList
 	resourceList.Append(&resInfo)
 	return resourceList
-
 }