github.com/olli-ai/jx/v2@v2.0.400-0.20210921045218-14731b4dd448/pkg/cmd/step/step_tag_test.go

// +build unit

package step

import (
	"path/filepath"
	"strings"
	"testing"

	"github.com/olli-ai/jx/v2/pkg/cmd/opts/step"

	"io/ioutil"
	"os"
	"path"

	"github.com/olli-ai/jx/v2/pkg/cmd/opts"
	"github.com/olli-ai/jx/v2/pkg/gits"
	"github.com/olli-ai/jx/v2/pkg/util"
	"github.com/stretchr/testify/assert"
	"k8s.io/helm/pkg/chartutil"
)

func TestStepTagCharts(t *testing.T) {
	t.Parallel()
	f, err := ioutil.TempDir("", "test-step-tag-charts")
	assert.NoError(t, err)
	defer os.RemoveAll(f)

	testData := path.Join("test_data", "step_tag_project")
	_, err = os.Stat(testData)
	assert.NoError(t, err)

	err = util.CopyDir(testData, f, true)
	assert.NoError(t, err)

	expectedVersion := "1.2.3"
	expectedImageName := "docker.io/jenkinsxio/awesome"

	chartsDir := filepath.Join(f, "charts", "mydemo")
	chartFile := filepath.Join(chartsDir, "Chart.yaml")
	valuesFile := filepath.Join(chartsDir, "values.yaml")

	o := StepTagOptions{
		StepOptions: step.StepOptions{
			CommonOptions: &opts.CommonOptions{},
		},
	}
	//o.Out = tests.Output()
	o.Flags.ChartsDir = chartsDir
	o.Flags.Version = expectedVersion
	o.Flags.ChartValueRepository = expectedImageName
	o.SetGit(&gits.GitFake{})
	err = o.Run()
	assert.NoError(t, err)

	// root file
	chart, err := chartutil.LoadChartfile(chartFile)
	assert.NoError(t, err, "failed to load file %s", chartFile)

	assert.Equal(t, expectedVersion, chart.Version, "replaced chart version")
	assert.Equal(t, expectedVersion, chart.AppVersion, "replaced chart appVersion")

	data, err := ioutil.ReadFile(valuesFile)
	assert.NoError(t, err, "failed to load file %s", valuesFile)
	lines := strings.Split(string(data), "\n")

	foundRepo := false
	foundVersion := false
	anotherImage := false
	for _, line := range lines {
		if strings.HasPrefix(line, "anotherImage:") {
			anotherImage = true
		}
		if strings.HasPrefix(line, ValuesYamlRepositoryPrefix) {
			value := strings.TrimSpace(strings.TrimPrefix(line, ValuesYamlRepositoryPrefix))
			if anotherImage {
				assert.Equal(t, "anotherImageRepoValue", value, "values.yaml anotherImage.repository: attribute")
			} else {
				foundRepo = true
				assert.Equal(t, expectedImageName, value, "values.yaml repository: attribute")
			}
		} else if strings.HasPrefix(line, ValuesYamlTagPrefix) {
			value := strings.TrimSpace(strings.TrimPrefix(line, ValuesYamlTagPrefix))
			if anotherImage {
				assert.Equal(t, "anotherImageTagValue", value, "values.yaml anotherImage.tag: attribute")
			} else {
				foundVersion = true
				assert.Equal(t, expectedVersion, value, "values.yaml tag: attribute")
			}
		}
	}

	assert.True(t, foundRepo, "Failed to find tag '%s' in file %s", ValuesYamlRepositoryPrefix, valuesFile)
	assert.True(t, foundVersion, "Failed to find tag '%s' in file %s", ValuesYamlTagPrefix, valuesFile)
}

func TestDefaultChartValueRepositoryValidName(t *testing.T) {
	f, err := ioutil.TempDir("", "test-step-tag-valid-name")
	assert.NoError(t, err)
	defer os.RemoveAll(f)

	if app, clean := os.LookupEnv("APP_NAME"); clean {
		assert.NoError(t, os.Unsetenv("APP_NAME"))
		defer os.Setenv("APP_NAME", app)
	}
	if app, clean := os.LookupEnv("REPO_NAME"); clean {
		assert.NoError(t, os.Unsetenv("REPO_NAME"))
		defer os.Setenv("REPO_NAME", app)
	}

	testData := path.Join("test_data", "step_tag_project")
	_, err = os.Stat(testData)
	assert.NoError(t, err)

	err = util.CopyDir(testData, f, true)
	assert.NoError(t, err)

	expectedImageName := "docker.io/jenkinsxio/my-awesome-project-org"

	o := StepTagOptions{
		StepOptions: step.StepOptions{
			CommonOptions: &opts.CommonOptions{},
		},
	}
	chartsDir := filepath.Join(f, "charts", "mydemo")
	o.Flags.Dir = f
	o.Flags.ChartsDir = chartsDir
	git := gits.GitFake{}
	git.SetRemoteURL("", "origin", "https://github.com/My_Self/My_Awesome_Project.org")
	o.SetGit(&git)

	assert.Equal(t, expectedImageName, o.defaultChartValueRepository())
}

func TestFindChartsDir(t *testing.T) {
	testData := path.Join("test_data", "step_tag_find_chart")
	dir, err := os.Getwd()
	assert.NoError(t, err)
	defer os.Chdir(dir)
	assert.NoError(t, os.Chdir(testData))

	if val, ok := os.LookupEnv("APP_NAME"); ok {
		assert.NoError(t, os.Unsetenv("APP_NAME"))
		defer os.Setenv("APP_NAME", val)
	} else {
		defer os.Unsetenv("APP_NAME")
	}
	if val, ok := os.LookupEnv("REPO_NAME"); ok {
		assert.NoError(t, os.Unsetenv("REPO_NAME"))
		defer os.Setenv("REPO_NAME", val)
	} else {
		defer os.Unsetenv("REPO_NAME")
	}

	chartsDir := filepath.Join("charts", "my-project") + "/"
	o := StepTagOptions{}

	assert.NoError(t, os.Setenv("APP_NAME", "My.Project"))
	val, err := o.findChartsDir()
	assert.NoError(t, err)
	assert.Equal(t, chartsDir, val)
	assert.NoError(t, os.Unsetenv("APP_NAME"))

	assert.NoError(t, os.Setenv("REPO_NAME", "my-project"))
	val, err = o.findChartsDir()
	assert.NoError(t, err)
	assert.Equal(t, chartsDir, val)
	assert.NoError(t, os.Unsetenv("REPO_NAME"))
}

func TestFindBestFitChart(t *testing.T) {
	testData := path.Join("test_data", "step_tag_find_best_fit_chart")
	dir, err := os.Getwd()
	assert.NoError(t, err)
	defer os.Chdir(dir)
	assert.NoError(t, os.Chdir(testData))

	if val, ok := os.LookupEnv("APP_NAME"); ok {
		assert.NoError(t, os.Unsetenv("APP_NAME"))
		defer os.Setenv("APP_NAME", val)
	} else {
		defer os.Unsetenv("APP_NAME")
	}
	if val, ok := os.LookupEnv("REPO_NAME"); ok {
		assert.NoError(t, os.Unsetenv("REPO_NAME"))
		defer os.Setenv("REPO_NAME", val)
	} else {
		defer os.Unsetenv("REPO_NAME")
	}

	chartsDir := filepath.Join("charts", "my-project") + "/"
	o := StepTagOptions{}

	assert.NoError(t, os.Setenv("APP_NAME", "My.Project"))
	val, err := o.findChartsDir()
	assert.NoError(t, err)
	assert.Equal(t, chartsDir, val)
	assert.NoError(t, os.Unsetenv("APP_NAME"))

	assert.NoError(t, os.Setenv("REPO_NAME", "my-project"))
	val, err = o.findChartsDir()
	assert.NoError(t, err)
	assert.Equal(t, chartsDir, val)
	assert.NoError(t, os.Unsetenv("REPO_NAME"))
}