mirror of
https://github.com/argoproj/argo-cd
synced 2026-04-21 17:07:16 +00:00
122 lines
3.5 KiB
Go
122 lines
3.5 KiB
Go
package helm
|
|
|
|
import (
|
|
"bytes"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"gopkg.in/yaml.v2"
|
|
|
|
"github.com/argoproj/argo-cd/v2/util/io"
|
|
)
|
|
|
|
type fakeIndexCache struct {
|
|
data []byte
|
|
}
|
|
|
|
func (f *fakeIndexCache) SetHelmIndex(_ string, indexData []byte) error {
|
|
f.data = indexData
|
|
return nil
|
|
}
|
|
|
|
func (f *fakeIndexCache) GetHelmIndex(_ string, indexData *[]byte) error {
|
|
*indexData = f.data
|
|
return nil
|
|
}
|
|
|
|
func TestIndex(t *testing.T) {
|
|
t.Run("Invalid", func(t *testing.T) {
|
|
client := NewClient("", Creds{}, false, "")
|
|
_, err := client.GetIndex(false)
|
|
assert.Error(t, err)
|
|
})
|
|
t.Run("Stable", func(t *testing.T) {
|
|
client := NewClient("https://argoproj.github.io/argo-helm", Creds{}, false, "")
|
|
index, err := client.GetIndex(false)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, index)
|
|
})
|
|
t.Run("BasicAuth", func(t *testing.T) {
|
|
client := NewClient("https://argoproj.github.io/argo-helm", Creds{
|
|
Username: "my-password",
|
|
Password: "my-username",
|
|
}, false, "")
|
|
index, err := client.GetIndex(false)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, index)
|
|
})
|
|
|
|
t.Run("Cached", func(t *testing.T) {
|
|
fakeIndex := Index{Entries: map[string]Entries{"fake": {}}}
|
|
data := bytes.Buffer{}
|
|
err := yaml.NewEncoder(&data).Encode(fakeIndex)
|
|
require.NoError(t, err)
|
|
|
|
client := NewClient("https://argoproj.github.io/argo-helm", Creds{}, false, "", WithIndexCache(&fakeIndexCache{data: data.Bytes()}))
|
|
index, err := client.GetIndex(false)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, fakeIndex, *index)
|
|
})
|
|
|
|
}
|
|
|
|
func Test_nativeHelmChart_ExtractChart(t *testing.T) {
|
|
client := NewClient("https://argoproj.github.io/argo-helm", Creds{}, false, "")
|
|
path, closer, err := client.ExtractChart("argo-cd", "0.7.1", false)
|
|
assert.NoError(t, err)
|
|
defer io.Close(closer)
|
|
info, err := os.Stat(path)
|
|
assert.NoError(t, err)
|
|
assert.True(t, info.IsDir())
|
|
}
|
|
|
|
func Test_nativeHelmChart_ExtractChart_insecure(t *testing.T) {
|
|
client := NewClient("https://argoproj.github.io/argo-helm", Creds{InsecureSkipVerify: true}, false, "")
|
|
path, closer, err := client.ExtractChart("argo-cd", "0.7.1", false)
|
|
assert.NoError(t, err)
|
|
defer io.Close(closer)
|
|
info, err := os.Stat(path)
|
|
assert.NoError(t, err)
|
|
assert.True(t, info.IsDir())
|
|
}
|
|
|
|
func Test_normalizeChartName(t *testing.T) {
|
|
t.Run("Test non-slashed name", func(t *testing.T) {
|
|
n := normalizeChartName("mychart")
|
|
assert.Equal(t, n, "mychart")
|
|
})
|
|
t.Run("Test single-slashed name", func(t *testing.T) {
|
|
n := normalizeChartName("myorg/mychart")
|
|
assert.Equal(t, n, "mychart")
|
|
})
|
|
t.Run("Test chart name with suborg", func(t *testing.T) {
|
|
n := normalizeChartName("myorg/mysuborg/mychart")
|
|
assert.Equal(t, n, "mychart")
|
|
})
|
|
t.Run("Test double-slashed name", func(t *testing.T) {
|
|
n := normalizeChartName("myorg//mychart")
|
|
assert.Equal(t, n, "mychart")
|
|
})
|
|
t.Run("Test invalid chart name - ends with slash", func(t *testing.T) {
|
|
n := normalizeChartName("myorg/")
|
|
assert.Equal(t, n, "myorg/")
|
|
})
|
|
t.Run("Test invalid chart name - is dot", func(t *testing.T) {
|
|
n := normalizeChartName("myorg/.")
|
|
assert.Equal(t, n, "myorg/.")
|
|
})
|
|
t.Run("Test invalid chart name - is two dots", func(t *testing.T) {
|
|
n := normalizeChartName("myorg/..")
|
|
assert.Equal(t, n, "myorg/..")
|
|
})
|
|
}
|
|
|
|
func TestIsHelmOciRepo(t *testing.T) {
|
|
assert.True(t, IsHelmOciRepo("demo.goharbor.io"))
|
|
assert.True(t, IsHelmOciRepo("demo.goharbor.io:8080"))
|
|
assert.False(t, IsHelmOciRepo("https://demo.goharbor.io"))
|
|
assert.False(t, IsHelmOciRepo("https://demo.goharbor.io:8080"))
|
|
}
|