package state

import (
	"testing"
)

func TestIsLocalChart(t *testing.T) {
	testcases := []struct {
		input    string
		expected bool
	}{
		{
			input:    "mychart",
			expected: true,
		},
		{
			input:    "stable/mysql",
			expected: false,
		},
		{
			input:    "center/stable/mysql",
			expected: false,
		},
		{
			input:    "./charts/myapp",
			expected: true,
		},
		{
			input:    "center/stable/myapp",
			expected: false,
		},
		{
			input:    "./charts/mysubsystem/myapp",
			expected: true,
		},
		{
			input:    "/charts/mysubsystem/myapp",
			expected: true,
		},
		{
			// Regression test case for:
			// * https://github.com/roboll/helmfile/issues/675
			// * https://github.com/roboll/helmfile/issues/687
			input:    "https://github.com/arangodb/kube-arangodb/releases/download/0.3.11/kube-arangodb-crd.tgz",
			expected: false,
		},
		{
			input:    "https://example.com/bar.tgz",
			expected: false,
		},
	}

	for i := range testcases {
		testcase := testcases[i]

		actual := isLocalChart(testcase.input)

		if testcase.expected != actual {
			t.Errorf("unexpected result: isLocalChart(\"%s\"): expected=%v, got=%v", testcase.input, testcase.expected, actual)
		}
	}
}

func TestResolveRemortChart(t *testing.T) {
	testcases := []struct {
		input  string
		repo   string
		chart  string
		remote bool
	}{
		{
			input:  "mychart",
			remote: false,
		},
		{
			input:  "stable/mysql",
			repo:   "stable",
			chart:  "mysql",
			remote: true,
		},
		{
			input:  "./charts/myapp",
			remote: false,
		},
		{
			input:  "center/stable/myapp",
			repo:   "center",
			chart:  "stable/myapp",
			remote: true,
		},
		{
			input:  "./charts/mysubsystem/myapp",
			remote: false,
		},
		{
			input:  "/charts/mysubsystem/myapp",
			remote: false,
		},
		{
			// Regression test case for:
			// * https://github.com/roboll/helmfile/issues/675
			// * https://github.com/roboll/helmfile/issues/687
			input:  "https://github.com/arangodb/kube-arangodb/releases/download/0.3.11/kube-arangodb-crd.tgz",
			remote: false,
		},
		{
			input:  "https://example.com/bar.tgz",
			remote: false,
		},
	}

	for i := range testcases {
		testcase := testcases[i]

		repo, chart, actual := resolveRemoteChart(testcase.input)

		if testcase.remote != actual {
			t.Fatalf("unexpected result: reolveRemoteChart(\"%s\"): expected=%v, got=%v", testcase.input, testcase.remote, actual)
		}

		if testcase.repo != repo {
			t.Errorf("unexpected repo: %s: expected=%v, got=%v", testcase.input, testcase.repo, repo)
		}

		if testcase.chart != chart {
			t.Errorf("unexpected chart: %s: expected=%v, got=%v", testcase.input, testcase.chart, chart)
		}
	}
}

func TestNormalizeChart(t *testing.T) {
	testcases := []struct {
		input    string
		expected string
	}{
		{
			input:    "mychart",
			expected: "/path/to/mychart",
		},
		{
			input:    "/charts/mychart",
			expected: "/charts/mychart",
		},
	}

	for i := range testcases {
		testcase := testcases[i]

		actual := normalizeChart("/path/to", testcase.input)

		if testcase.expected != actual {
			t.Fatalf("unexpected result: normalizeChart(\"/path/to\", \"%s\"): expected=%v, got=%v", testcase.input, testcase.expected, actual)
		}
	}
}
func TestSafeVersionPath(t *testing.T) {
	testcases := []struct {
		input    string
		expected string
	}{
		{
			input:    "=1.2.3",
			expected: "_1.2.3",
		},
		{
			input:    ">1.0.0",
			expected: "_1.0.0",
		},
		{
			input:    "<=2.3.4",
			expected: "__2.3.4",
		},
		{
			input:    "!3.4.5",
			expected: "_3.4.5",
		},
		{
			input:    "|4.5.6",
			expected: "_4.5.6",
		},
		{
			input:    "~5.6.7",
			expected: "_5.6.7",
		},
		{
			input:    "^6.7.8",
			expected: "_6.7.8",
		},
		{
			input:    " 7.8.9",
			expected: "_7.8.9",
		},
		{
			input:    ",9.10.11",
			expected: "_9.10.11",
		},
		{
			input:    "<= 1.2.3, >= 1.4",
			expected: "___1.2.3_____1.4",
		},
		{
			input:    "*",
			expected: "_",
		},
	}

	for i := range testcases {
		testcase := testcases[i]

		actual := safeVersionPath(testcase.input)

		if testcase.expected != actual {
			t.Errorf("unexpected result: safeVersionPath(\"%s\"): expected=%v, got=%v", testcase.input, testcase.expected, actual)
		}
	}
}