From a24318c7f12410e760f2607bf2b12c046b57bb3b Mon Sep 17 00:00:00 2001
From: Terry Howe <terrylhowe@gmail.com>
Date: Thu, 2 Oct 2025 12:50:37 -0600
Subject: [PATCH] fix: flake upgrade test

Signed-off-by: Terry Howe <terrylhowe@gmail.com>
---
 pkg/action/upgrade_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/action/upgrade_test.go b/pkg/action/upgrade_test.go
index 0a436534f..80e3070e5 100644
--- a/pkg/action/upgrade_test.go
+++ b/pkg/action/upgrade_test.go
@@ -60,9 +60,9 @@ func TestUpgradeRelease_Success(t *testing.T) {
 
 	ctx, done := context.WithCancel(t.Context())
 	res, err := upAction.RunWithContext(ctx, rel.Name, buildChart(), vals)
-	done()
 	req.NoError(err)
 	is.Equal(res.Info.Status, release.StatusDeployed)
+	done()
 
 	// Detecting previous bug where context termination after successful release
 	// caused release to fail.