package version

import (
	"testing"
)

func TestNormalizeVersion(t *testing.T) {
	tests := []struct {
		name    string
		version string
		want    string
	}{
		{
			name:    "version with v prefix",
			version: "v1.2.3",
			want:    "v1.2.3",
		},
		{
			name:    "version without v prefix",
			version: "1.2.3",
			want:    "",
		},
		{
			name:    "version with additional text",
			version: "v1.2.3-beta",
			want:    "v1.2.3",
		},
		{
			name:    "version with multiple dots",
			version: "v1.2.3.4",
			want:    "v1.2.3",
		},
		{
			name:    "version with single number",
			version: "v1",
			want:    "v1",
		},
		{
			name:    "version with two numbers",
			version: "v1.2",
			want:    "v1.2",
		},
		{
			name:    "no version pattern",
			version: "no version here",
			want:    "",
		},
		{
			name:    "empty string",
			version: "",
			want:    "",
		},
		{
			name:    "just v",
			version: "v",
			want:    "",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := NormalizeVersion(tt.version); got != tt.want {
				t.Errorf("NormalizeVersion() = %v, want %v", got, tt.want)
			}
		})
	}
}
