92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package state
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/helmfile/helmfile/pkg/filesystem"
|
|
"github.com/helmfile/helmfile/pkg/helmexec"
|
|
"github.com/helmfile/helmfile/pkg/remote"
|
|
)
|
|
|
|
func TestStorage_resolveFile(t *testing.T) {
|
|
type args struct {
|
|
missingFileHandler *string
|
|
title string
|
|
path string
|
|
}
|
|
|
|
cacheDir := remote.CacheDir()
|
|
infoHandler := MissingFileHandlerInfo
|
|
errorHandler := MissingFileHandlerError
|
|
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantFiles []string
|
|
wantSkipped bool
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "non existing file in repo produce skip",
|
|
args: args{
|
|
path: "git::https://github.com/helmfile/helmfile.git@examples/values/non-existing-file.yaml?ref=v0.145.2",
|
|
title: "values",
|
|
missingFileHandler: &infoHandler,
|
|
},
|
|
wantSkipped: true,
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "non existing file in repo produce skip",
|
|
args: args{
|
|
path: "git::https://github.com/helmfile/helmfile.git@examples/values/non-existing-file.yaml?ref=v0.145.2",
|
|
title: "values",
|
|
missingFileHandler: &errorHandler,
|
|
},
|
|
wantSkipped: false,
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "existing remote value fetched",
|
|
args: args{
|
|
path: "git::https://github.com/helmfile/helmfile.git@examples/values/replica-values.yaml?ref=v0.145.2",
|
|
title: "values",
|
|
missingFileHandler: &infoHandler,
|
|
},
|
|
wantFiles: []string{fmt.Sprintf("%s/%s", cacheDir, "values/https_github_com_helmfile_helmfile_git.ref=v0.145.2/examples/values/replica-values.yaml")},
|
|
wantSkipped: false,
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "non existing remote repo produce an error",
|
|
args: args{
|
|
path: "https://github.com/helmfile/helmfiles.git@examples/values/replica-values.yaml?ref=v0.145.2",
|
|
title: "values",
|
|
missingFileHandler: &infoHandler,
|
|
},
|
|
wantSkipped: false,
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
st := NewStorage(cacheDir, helmexec.NewLogger(io.Discard, "debug"), filesystem.DefaultFileSystem())
|
|
|
|
files, skipped, err := st.resolveFile(tt.args.missingFileHandler, tt.args.title, tt.args.path)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("resolveFile() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(files, tt.wantFiles) {
|
|
t.Errorf("resolveFile() files = %v, want %v", files, tt.wantFiles)
|
|
}
|
|
if skipped != tt.wantSkipped {
|
|
t.Errorf("resolveFile() skipped = %v, want %v", skipped, tt.wantSkipped)
|
|
}
|
|
})
|
|
}
|
|
}
|