/*
Copyright 2018 The Kubernetes 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 version

import (
	"testing"
)

func TestTruncate(t *testing.T) {
	t.Parallel()
	cases := []struct {
		Value     string
		MaxLength int
		Expected  string
	}{
		{
			Value:     "A Really Long String",
			MaxLength: 1,
			Expected:  "A",
		},
		{
			Value:     "A Short String",
			MaxLength: 10,
			Expected:  "A Short St",
		},
		{
			Value:     "Under Max Length String",
			MaxLength: 1000,
			Expected:  "Under Max Length String",
		},
	}
	for _, tc := range cases {
		tc := tc // capture range variable
		t.Run(tc.Value, func(t *testing.T) {
			t.Parallel()
			result := truncate(tc.Value, tc.MaxLength)
			// sanity check length
			if len(result) > tc.MaxLength {
				t.Errorf("Result %q longer than Max Length %d!", result, tc.MaxLength)
			}
			if tc.Expected != result {
				t.Errorf("Strings did not match!")
				t.Errorf("Expected: %q", tc.Expected)
				t.Errorf("But got: %q", result)
			}
		})
	}
}

func TestVersion(t *testing.T) {
	tests := []struct {
		name              string
		versionPreRelease string
		gitCommit         string
		gitCommitCount    string
		want              string
	}{
		{
			name:              "With git commit count and with commit hash",
			versionPreRelease: "alpha",
			gitCommit:         "mocked-hash",
			gitCommitCount:    "mocked-count",
			want:              versionCore + "-" + "alpha" + "." + "mocked-count" + "+" + "mocked-hash",
		},
		{
			name:              "Without git commit count and and with hash",
			versionPreRelease: "beta",
			gitCommit:         "mocked-hash",
			gitCommitCount:    "",
			want:              versionCore + "-" + "beta" + "+" + "mocked-hash",
		},
		{
			name:              "Without git commit hash and with commit count",
			versionPreRelease: "alpha",
			gitCommit:         "",
			gitCommitCount:    "mocked-count",
			want:              versionCore + "-" + "alpha" + "." + "mocked-count",
		},
		{
			name:              "Without git commit hash and without commit count",
			versionPreRelease: "alpha",
			gitCommit:         "",
			gitCommitCount:    "",
			want:              versionCore + "-" + "alpha",
		},
		{
			name:              "Without pre release version",
			versionPreRelease: "",
			gitCommit:         "",
			gitCommitCount:    "",
			want:              versionCore,
		},
		{
			name:              "Without pre release version and with git commit hash and count",
			versionPreRelease: "",
			gitCommit:         "mocked-commit",
			gitCommitCount:    "mocked-count",
			want:              versionCore,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if tt.gitCommit != "" {
				gitCommitBackup := gitCommit
				gitCommit = tt.gitCommit
				defer func() {
					gitCommit = gitCommitBackup
				}()
			}

			if tt.gitCommitCount != "" {
				gitCommitCountBackup := gitCommitCount
				gitCommitCount = tt.gitCommitCount
				defer func() {
					gitCommitCount = gitCommitCountBackup
				}()
			}

			versionPreReleaseBackup := versionPreRelease
			versionPreRelease = tt.versionPreRelease
			defer func() {
				versionPreRelease = versionPreReleaseBackup
			}()
			if got := Version(); got != tt.want {
				t.Errorf("Version() = %v, want %v", got, tt.want)
			}
		})
	}
}
