enhancement(5235): added detectFlavor mock setup in upgrade error handling tests

Kaan Yalti · Kaan Yalti · commit 381d5a56d024 · 2025-09-04T16:49:42.000+03:00
diff --git a/internal/pkg/agent/application/upgrade/upgrade_test.go b/internal/pkg/agent/application/upgrade/upgrade_test.go
@@ -1369,6 +1369,9 @@ func TestUpgradeErrorHandling(t *testing.T) {
 						hash:     metadata.hash,
 					}
 				}
+				upgrader.detectFlavor = func(topDir string, defaultFlavor string) (string, error) {
+					return "flavor", nil
+				}
 				upgrader.unpacker = &mockUnpacker{
 					returnPackageMetadata: packageMetadata{
 						manifest: &v1.PackageManifest{},

Original file line number	Diff line number	Diff line change
`@@ -1369,6 +1369,9 @@ func TestUpgradeErrorHandling(t *testing.T) {`
`1369`	`1369`	`hash: metadata.hash,`
`1370`	`1370`	`}`
`1371`	`1371`	`}`
	`1372`	`+ upgrader.detectFlavor = func(topDir string, defaultFlavor string) (string, error) {`
	`1373`	`+ return "flavor", nil`
	`1374`	`+ }`
`1372`	`1375`	`upgrader.unpacker = &mockUnpacker{`
`1373`	`1376`	`returnPackageMetadata: packageMetadata{`
`1374`	`1377`	`manifest: &v1.PackageManifest{},`