Function TestExtractDigestFromImageRef

internal/validate/vsa/vsa_test.go:1372–1422

TestExtractDigestFromImageRef tests the ExtractDigestFromImageRef function

(t *testing.T)

// TestExtractDigestFromImageRef tests the ExtractDigestFromImageRef function
func TestExtractDigestFromImageRef(t *testing.T) {
	tests := []struct {
		name        string
		imageRef    string
		expected    string
		expectError bool
	}{
		{
			name:        "tag reference",
			imageRef:    "registry.io/image:latest",
			expected:    "registry.io/image:latest",
			expectError: false,
		},
		{
			name:        "docker hub tag",
			imageRef:    "nginx:1.21",
			expected:    "nginx:1.21",
			expectError: false,
		},
		{
			name:        "quay reference",
			imageRef:    "quay.io/namespace/repo:tag",
			expected:    "quay.io/namespace/repo:tag",
			expectError: false,
		},
		{
			name:        "invalid reference",
			imageRef:    "invalid:reference:with:colons",
			expected:    "",
			expectError: true,
		},
		{
			name:        "empty reference",
			imageRef:    "",
			expected:    "",
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result, err := ExtractDigestFromImageRef(tt.imageRef)
			if tt.expectError {
				require.Error(t, err)
			} else {
				require.NoError(t, err)
				assert.Equal(t, tt.expected, result)
			}
		})
	}
}