github.com/oam-dev/kubevela@v1.9.11/pkg/utils/helm/helm_helper_test.go

/*
Copyright 2021 The KubeVela 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 helm

import (
	"context"
	"fmt"
	"net/http"
	"net/http/httptest"
	"os"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"

	"github.com/google/go-cmp/cmp"
	v1 "k8s.io/api/core/v1"
	v12 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/apimachinery/pkg/types"
	"sigs.k8s.io/yaml"

	types2 "github.com/oam-dev/kubevela/apis/types"
	util2 "github.com/oam-dev/kubevela/pkg/oam/util"
	"github.com/oam-dev/kubevela/pkg/utils/util"
)

var _ = Describe("Test helm helper", func() {

	ctx := context.Background()

	It("Test LoadCharts ", func() {
		helper := NewHelper()
		chart, err := helper.LoadCharts("./testdata/autoscalertrait-0.1.0.tgz", nil)
		Expect(err).Should(BeNil())
		Expect(chart).ShouldNot(BeNil())
		Expect(chart.Metadata).ShouldNot(BeNil())
		Expect(cmp.Diff(chart.Metadata.Version, "0.1.0")).Should(BeEmpty())
	})

	It("Test UpgradeChart", func() {
		helper := NewHelper()
		chart, err := helper.LoadCharts("./testdata/autoscalertrait-0.1.0.tgz", nil)
		Expect(err).Should(BeNil())
		release, err := helper.UpgradeChart(chart, "autoscalertrait", "default", map[string]interface{}{
			"replicaCount": 2,
			"image.tag":    "0.1.0",
		}, UpgradeChartOptions{
			Config:  cfg,
			Detail:  false,
			Logging: util.IOStreams{Out: os.Stdout, ErrOut: os.Stderr},
			Wait:    false,
		})
		crds := GetCRDFromChart(release.Chart)
		Expect(cmp.Diff(len(crds), 1)).Should(BeEmpty())
		Expect(err).Should(BeNil())
		deployments := GetDeploymentsFromManifest(release.Manifest)
		Expect(cmp.Diff(len(deployments), 1)).Should(BeEmpty())
		Expect(cmp.Diff(*deployments[0].Spec.Replicas, int32(2))).Should(BeEmpty())
		containers := deployments[0].Spec.Template.Spec.Containers
		Expect(cmp.Diff(len(containers), 1)).Should(BeEmpty())

		// add new default value
		Expect(cmp.Diff(containers[0].Image, "ghcr.io/oam-dev/catalog/autoscalertrait:0.1.0")).Should(BeEmpty())

		chartNew, err := helper.LoadCharts("./testdata/autoscalertrait-0.2.0.tgz", nil)
		Expect(err).Should(BeNil())

		// the new custom values should override the last release custom values
		releaseNew, err := helper.UpgradeChart(chartNew, "autoscalertrait", "default", map[string]interface{}{
			"image.tag": "0.2.0",
		}, UpgradeChartOptions{
			Config:      cfg,
			Detail:      false,
			ReuseValues: true,
			Logging:     util.IOStreams{Out: os.Stdout, ErrOut: os.Stderr},
			Wait:        false,
		})
		Expect(err).Should(BeNil())
		deployments = GetDeploymentsFromManifest(releaseNew.Manifest)
		Expect(cmp.Diff(len(deployments), 1)).Should(BeEmpty())
		// keep the custom values
		Expect(cmp.Diff(*deployments[0].Spec.Replicas, int32(2))).Should(BeEmpty())
		containers = deployments[0].Spec.Template.Spec.Containers
		Expect(cmp.Diff(len(containers), 1)).Should(BeEmpty())

		// change the default value
		Expect(cmp.Diff(containers[0].Image, "ghcr.io/oam-dev/catalog/autoscalertrait:0.2.0")).Should(BeEmpty())

		// add new default value
		Expect(cmp.Diff(len(containers[0].Env), 1)).Should(BeEmpty())
		Expect(cmp.Diff(containers[0].Env[0].Name, "env1")).Should(BeEmpty())
	})

	It("Test UninstallRelease", func() {
		helper := NewHelper()
		err := helper.UninstallRelease("autoscalertrait", "default", cfg, false, util.IOStreams{Out: os.Stdout, ErrOut: os.Stderr})
		Expect(err).Should(BeNil())
	})

	It("Test ListVersions ", func() {
		helper := NewHelper()
		versions, err := helper.ListVersions("./testdata", "autoscalertrait", true, nil)
		Expect(err).Should(BeNil())
		Expect(cmp.Diff(len(versions), 3)).Should(BeEmpty())
	})

	It("Test getValues from chart", func() {
		helper := NewHelper()
		values, err := helper.GetValuesFromChart("./testdata", "autoscalertrait", "0.2.0", true, "helm", nil)
		Expect(err).Should(BeNil())
		Expect(values).ShouldNot(BeNil())
	})

	It("Test getValues from chart with uncompleted index.yaml url", func() {
		helper := NewHelper()
		values, err := helper.GetValuesFromChart("./testdata", "autoscalertrait", "0.2.0_p1", true, "helm", nil)
		Expect(err).Should(BeNil())
		Expect(values).ShouldNot(BeNil())
	})

	It("Test validate helm repo", func() {
		helper := NewHelper()
		helmRepo := &Repository{
			URL: "https://charts.kubevela.net/core",
		}
		ok, err := helper.ValidateRepo(ctx, helmRepo)
		Expect(err).Should(BeNil())
		Expect(ok).Should(BeTrue())
	})

	It("Test validate the corrupt helm repo", func() {
		helper := NewHelper()
		svr := httptest.NewTLSServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
			_, _ = fmt.Fprintf(writer, "corrupted")
		}))
		defer svr.Close()
		helmRepo := &Repository{URL: svr.URL}
		ok, err := helper.ValidateRepo(ctx, helmRepo)
		Expect(err).To(HaveOccurred())
		Expect(ok).Should(BeFalse())
	})
})

var _ = Describe("Test helm associated func", func() {
	ctx := context.Background()
	var aSec v1.Secret

	BeforeEach(func() {
		Expect(k8sClient.Create(ctx, &v1.Namespace{ObjectMeta: v12.ObjectMeta{Name: "vela-system"}})).Should(SatisfyAny(BeNil(), util2.AlreadyExistMatcher{}))
		aSec = v1.Secret{}
		Expect(yaml.Unmarshal([]byte(authSecret), &aSec)).Should(BeNil())
		Expect(k8sClient.Create(ctx, &aSec)).Should(SatisfyAny(BeNil(), util2.AlreadyExistMatcher{}))

		bSec := v1.Secret{}
		Expect(yaml.Unmarshal([]byte(caFileSecret), &bSec)).Should(BeNil())
		Expect(k8sClient.Create(ctx, &bSec)).Should(SatisfyAny(BeNil(), util2.AlreadyExistMatcher{}))
	})

	It("Test auth info secret func", func() {
		opts, err := SetHTTPOption(context.Background(), k8sClient, types.NamespacedName{Namespace: types2.DefaultKubeVelaNS, Name: "auth-secret"})
		Expect(err).Should(BeNil())
		Expect(opts.Username).Should(BeEquivalentTo("admin"))
		Expect(opts.Password).Should(BeEquivalentTo("admin"))
	})

	It("Test auth info secret func", func() {
		_, err := SetHTTPOption(context.Background(), k8sClient, types.NamespacedName{Namespace: types2.DefaultKubeVelaNS, Name: "auth-secret-1"})
		Expect(err).ShouldNot(BeNil())
	})

	It("Test ac secret func", func() {
		opts, err := SetHTTPOption(context.Background(), k8sClient, types.NamespacedName{Namespace: types2.DefaultKubeVelaNS, Name: "ca-secret"})
		Expect(err).Should(BeNil())
		Expect(opts.CaFile).Should(BeEquivalentTo("testfile"))
	})
})

var (
	authSecret = `
apiVersion: v1
kind: Secret
metadata:
  name: auth-secret
  namespace: vela-system
  labels:
    config.oam.dev/type: config-helm-repository
    config.oam.dev/project: my-project-1
stringData:
  url: https://kedacore.github.io/charts
  username: admin
  password: admin
type: Opaque
`
	caFileSecret = `
apiVersion: v1
kind: Secret
metadata:
  name: ca-secret
  namespace: vela-system
  labels:
    config.oam.dev/type: config-helm-repository
    config.oam.dev/project: my-project-1
stringData:
  url: https://kedacore.github.io/charts
  caFile: testfile
type: Opaque
`
)