mirror of https://github.com/helm/helm
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.1 KiB
85 lines
2.1 KiB
/*
|
|
Copyright The Helm Authors.
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package plugin
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.yaml.in/yaml/v3"
|
|
|
|
"helm.sh/helm/v4/internal/plugin/schema"
|
|
)
|
|
|
|
func mockSubprocessCLIPluginErrorExit(t *testing.T, pluginName string, exitCode uint8) *SubprocessPluginRuntime {
|
|
t.Helper()
|
|
|
|
rc := RuntimeConfigSubprocess{
|
|
PlatformCommand: []PlatformCommand{
|
|
{Command: "sh", Args: []string{"-c", fmt.Sprintf("echo \"mock plugin $@\"; exit %d", exitCode)}},
|
|
},
|
|
}
|
|
|
|
pluginDir := t.TempDir()
|
|
|
|
md := Metadata{
|
|
Name: pluginName,
|
|
Version: "v0.1.2",
|
|
Type: "cli/v1",
|
|
APIVersion: "v1",
|
|
Runtime: "subprocess",
|
|
Config: &schema.ConfigCLIV1{
|
|
Usage: "Mock plugin",
|
|
ShortHelp: "Mock plugin",
|
|
LongHelp: "Mock plugin for testing",
|
|
IgnoreFlags: false,
|
|
},
|
|
RuntimeConfig: &rc,
|
|
}
|
|
|
|
data, err := yaml.Marshal(md)
|
|
require.NoError(t, err)
|
|
os.WriteFile(filepath.Join(pluginDir, "plugin.yaml"), data, 0o644)
|
|
|
|
return &SubprocessPluginRuntime{
|
|
metadata: md,
|
|
pluginDir: pluginDir,
|
|
RuntimeConfig: rc,
|
|
}
|
|
}
|
|
|
|
func TestSubprocessPluginRuntime(t *testing.T) {
|
|
p := mockSubprocessCLIPluginErrorExit(t, "foo", 56)
|
|
|
|
output, err := p.Invoke(t.Context(), &Input{
|
|
Message: schema.InputMessageCLIV1{
|
|
ExtraArgs: []string{"arg1", "arg2"},
|
|
//Env: []string{"FOO=bar"},
|
|
},
|
|
})
|
|
|
|
require.Error(t, err)
|
|
ieerr, ok := err.(*InvokeExecError)
|
|
require.True(t, ok, "expected InvokeExecError, got %T", err)
|
|
assert.Equal(t, 56, ieerr.ExitCode)
|
|
|
|
assert.Nil(t, output)
|
|
}
|