versions_test.go

package main

import (
	"reflect"
	"testing"

	"github.com/blang/semver/v4"
)

func Test_getTags(t *testing.T) {
	type args struct {
		imageName  string
		registries []string
		versions   []string
		fullname   bool
	}
	tests := []struct {
		name     string
		args     args
		wantTags []string
	}{
		{
			name: "single registry (full name)",
			args: args{
				imageName:  "name",
				registries: []string{"registry1"},
				versions:   []string{"1.0.1"},
				fullname:   true,
			},
			wantTags: []string{"registry1/name:1.0.1"},
		},
		{
			name: "multiple registries (full name)",
			args: args{
				imageName:  "name",
				registries: []string{"registry1", "registry2"},
				versions:   []string{"1.0.1"},
				fullname:   true,
			},
			wantTags: []string{"registry1/name:1.0.1", "registry2/name:1.0.1"},
		},
		{
			name: "multiple versions (full name)",
			args: args{
				imageName:  "name",
				registries: []string{"registry1"},
				versions:   []string{"1.0.1", "latest"},
				fullname:   true,
			},
			wantTags: []string{"registry1/name:1.0.1", "registry1/name:latest"},
		},
		{
			name: "multiple versions (mo full name)",
			args: args{
				imageName:  "name",
				registries: []string{"registry1"},
				versions:   []string{"1.0.1", "latest"},
				fullname:   false,
			},
			wantTags: []string{"1.0.1", "latest"},
		},
		{
			name: "multiple both (full name)",
			args: args{
				imageName:  "name",
				registries: []string{"registry1", "registry2"},
				versions:   []string{"1.0.1", "latest"},
				fullname:   true,
			},
			wantTags: []string{"registry1/name:1.0.1", "registry1/name:latest", "registry2/name:1.0.1", "registry2/name:latest"},
		},
		{
			name: "multiple both (no full name)",
			args: args{
				imageName:  "name",
				registries: []string{"registry1", "registry2"},
				versions:   []string{"1.0.1", "latest"},
				fullname:   false,
			},
			wantTags: []string{"1.0.1", "latest", "1.0.1", "latest"},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if gotTags := getTags(tt.args.imageName, tt.args.registries, tt.args.versions, tt.args.fullname); !reflect.DeepEqual(gotTags, tt.wantTags) {
				t.Errorf("getTags() = %v, want %v", gotTags, tt.wantTags)
			}
		})
	}
}

func Test_refToVersion(t *testing.T) {
	type args struct {
		ref string
		sha string
	}
	tests := []struct {
		name string
		args args
		want string
	}{
		{
			name: "v1.0.0",
			args: args{
				ref: "refs/tags/v1.0.0",
				sha: "abc123",
			},
			want: "1.0.0",
		},
		{
			name: "1.0.0",
			args: args{
				ref: "refs/tags/1.0.0",
				sha: "abc123",
			},
			want: "1.0.0",
		},
		{
			name: "v1.0.0.0.0.0",
			args: args{
				ref: "refs/tags/v1.0.0.0.0.0",
				sha: "abc123",
			},
			want: "unknown",
		},
		{
			name: "1.0",
			args: args{
				ref: "refs/tags/1.0",
				sha: "abc123",
			},
			want: "unknown",
		},
		{
			name: "v1.0",
			args: args{
				ref: "refs/tags/v1.0",
				sha: "abc123",
			},
			want: "unknown",
		},
		{
			name: "v1",
			args: args{
				ref: "refs/tags/v1",
				sha: "abc123",
			},
			want: "unknown",
		},
		{
			name: "1",
			args: args{
				ref: "refs/tags/1",
				sha: "abc123",
			},
			want: "unknown",
		},
		{
			name: "master",
			args: args{
				ref: "refs/heads/master",
				sha: "abc123",
			},
			want: "abc123",
		},
		{
			name: "main",
			args: args{
				ref: "refs/heads/main",
				sha: "abc123",
			},
			want: "abc123",
		},
		{
			name: "non numeric number",
			args: args{
				ref: "refs/tags/v1.a",
				sha: "abc123",
			},
			want: "unknown",
		},
		{
			name: "invalid ref",
			args: args{
				ref: "refs/heads/dev",
				sha: "abc123",
			},
			want: "unknown",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := refToVersion(tt.args.ref, tt.args.sha); got != tt.want {
				t.Errorf("refToVersion() = %v, want %v", got, tt.want)
			}
		})
	}
}

func Test_refToVersions(t *testing.T) {
	tests := []struct {
		name          string
		input         string
		latestVersion semver.Version
		wantVersions  []string
	}{
		{
			name:          "v1.0.0",
			input:         "refs/tags/v1.0.0",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  []string{"latest", "1.0.0", "1.0", "1"},
		},
		{
			name:          "1.0.0",
			input:         "refs/tags/v1.0.0",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  []string{"latest", "1.0.0", "1.0", "1"},
		},
		{
			name:          "v1.0.0.0.0.0",
			input:         "refs/tags/v1.0.0.0.0.0",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  nil,
		},
		{
			name:          "1.0",
			input:         "refs/tags/1.0",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  nil,
		},
		{
			name:          "v1.0",
			input:         "refs/tags/v1.0",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  nil,
		},
		{
			name:          "1",
			input:         "refs/tags/1",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  nil,
		},
		{
			name:          "v1",
			input:         "refs/tags/v1",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  nil,
		},
		{
			name:          "master",
			input:         "refs/heads/master",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  []string{"dev"},
		},
		{
			name:          "main",
			input:         "refs/heads/main",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  []string{"dev"},
		},
		{
			name:          "non numeric number",
			input:         "refs/tags/v1.a",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  nil,
		},
		{
			name:          "invalid ref",
			input:         "refs/heads/dev",
			latestVersion: semver.MustParse("1.0.0"),
			wantVersions:  nil,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if gotVersions := refToVersions(tt.input, &tt.latestVersion); !reflect.DeepEqual(gotVersions, tt.wantVersions) {
				t.Errorf("refToVersions() = %v, want %v", gotVersions, tt.wantVersions)
			}
		})
	}
}

func Test_getLatestVersion(t *testing.T) {
	tests := []struct {
		name     string
		versions []string
		want     string
		wantErr  bool
	}{
		{
			name:     "Normal",
			versions: []string{"7.0.0", "1.0.0"},
			want:     "7.0.0",
			wantErr:  false,
		},
		{
			name:     "has Prerelease",
			versions: []string{"7.0.0-alpha", "1.0.0"},
			want:     "1.0.0",
			wantErr:  false,
		},
		{
			name:     "has Build",
			versions: []string{"7.0.0+1", "1.0.0"},
			want:     "7.0.0+1",
			wantErr:  false,
		},
		{
			name:     "has Build",
			versions: []string{"7.0.0+1", "7.0.0", "1.0.0"},
			want:     "7.0.0+1",
			wantErr:  false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := getLatestVersion(tt.versions)
			wantSemVer := semver.MustParse(tt.want)
			if !reflect.DeepEqual(got, &wantSemVer) {
				t.Errorf("getLatestVersion() got = %#v, want %#v", got, tt.want)
			}
		})
	}
}