Function TestPluginUninstallCleansUpVersionedFiles

pkg/cmd/plugin_uninstall_test.go:29–146

(t *testing.T)

)

func TestPluginUninstallCleansUpVersionedFiles(t *testing.T) {
	ensure.HelmHome(t)

	// Create a fake plugin directory structure in a temp directory
	pluginsDir := t.TempDir()
	t.Setenv("HELM_PLUGINS", pluginsDir)

	// Create a new settings instance that will pick up the environment variable
	testSettings := cli.New()
	pluginName := "test-plugin"

	// Create plugin directory
	pluginDir := filepath.Join(pluginsDir, pluginName)
	if err := os.MkdirAll(pluginDir, 0755); err != nil {
		t.Fatal(err)
	}

	// Create plugin.yaml
	pluginYAML := `name: test-plugin
version: 1.2.3
description: Test plugin
command: $HELM_PLUGIN_DIR/test-plugin
`
	if err := os.WriteFile(filepath.Join(pluginDir, "plugin.yaml"), []byte(pluginYAML), 0644); err != nil {
		t.Fatal(err)
	}

	// Create versioned tarball and provenance files
	tarballFile := filepath.Join(pluginsDir, "test-plugin-1.2.3.tgz")
	provFile := filepath.Join(pluginsDir, "test-plugin-1.2.3.tgz.prov")
	otherVersionTarball := filepath.Join(pluginsDir, "test-plugin-2.0.0.tgz")

	if err := os.WriteFile(tarballFile, []byte("fake tarball"), 0644); err != nil {
		t.Fatal(err)
	}
	if err := os.WriteFile(provFile, []byte("fake provenance"), 0644); err != nil {
		t.Fatal(err)
	}
	// Create another version that should NOT be removed
	if err := os.WriteFile(otherVersionTarball, []byte("other version"), 0644); err != nil {
		t.Fatal(err)
	}

	// Load the plugin
	p, err := plugin.LoadDir(pluginDir)
	if err != nil {
		t.Fatal(err)
	}

	// Create a test uninstall function that uses our test settings
	testUninstallPlugin := func(plugin plugin.Plugin) error {
		if err := os.RemoveAll(plugin.Dir()); err != nil {
			return err
		}

		// Clean up versioned tarball and provenance files from test HELM_PLUGINS directory
		pluginName := plugin.Metadata().Name
		pluginVersion := plugin.Metadata().Version