119 lines
2.9 KiB
Go
119 lines
2.9 KiB
Go
package state
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestParseOCIChartRef(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
chartURL string
|
|
expectedBase string
|
|
expectedVer string
|
|
expectedDigest string
|
|
}{
|
|
{
|
|
name: "plain OCI URL",
|
|
chartURL: "oci://registry/chart",
|
|
expectedBase: "oci://registry/chart",
|
|
expectedVer: "",
|
|
expectedDigest: "",
|
|
},
|
|
{
|
|
name: "OCI URL with version",
|
|
chartURL: "oci://registry/chart:2.0.0",
|
|
expectedBase: "oci://registry/chart",
|
|
expectedVer: "2.0.0",
|
|
expectedDigest: "",
|
|
},
|
|
{
|
|
name: "OCI URL with digest",
|
|
chartURL: "oci://registry/chart@sha256:abc",
|
|
expectedBase: "oci://registry/chart",
|
|
expectedVer: "",
|
|
expectedDigest: "sha256:abc",
|
|
},
|
|
{
|
|
name: "OCI URL with version and digest",
|
|
chartURL: "oci://reg/chart:2.0@sha256:abc",
|
|
expectedBase: "oci://reg/chart",
|
|
expectedVer: "2.0",
|
|
expectedDigest: "sha256:abc",
|
|
},
|
|
{
|
|
name: "OCI URL with port, version, and digest",
|
|
chartURL: "oci://reg:5000/chart:1.0@sha256:a",
|
|
expectedBase: "oci://reg:5000/chart",
|
|
expectedVer: "1.0",
|
|
expectedDigest: "sha256:a",
|
|
},
|
|
{
|
|
name: "OCI URL with port only",
|
|
chartURL: "oci://reg:5000/chart",
|
|
expectedBase: "oci://reg:5000/chart",
|
|
expectedVer: "",
|
|
expectedDigest: "",
|
|
},
|
|
{
|
|
name: "OCI URL with port and digest, no version",
|
|
chartURL: "oci://reg:5000/chart@sha256:abc",
|
|
expectedBase: "oci://reg:5000/chart",
|
|
expectedVer: "",
|
|
expectedDigest: "sha256:abc",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
base, ver, digest := parseOCIChartRef(tt.chartURL)
|
|
assert.Equal(t, tt.expectedBase, base, "baseURL mismatch")
|
|
assert.Equal(t, tt.expectedVer, ver, "version mismatch")
|
|
assert.Equal(t, tt.expectedDigest, digest, "digest mismatch")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestParseVersionDigest(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
version string
|
|
expectedVer string
|
|
expectedDigest string
|
|
}{
|
|
{
|
|
name: "version only",
|
|
version: "2.0.0",
|
|
expectedVer: "2.0.0",
|
|
expectedDigest: "",
|
|
},
|
|
{
|
|
name: "version with digest",
|
|
version: "2.0.0@sha256:abc",
|
|
expectedVer: "2.0.0",
|
|
expectedDigest: "sha256:abc",
|
|
},
|
|
{
|
|
name: "digest only",
|
|
version: "@sha256:abc",
|
|
expectedVer: "",
|
|
expectedDigest: "sha256:abc",
|
|
},
|
|
{
|
|
name: "empty string",
|
|
version: "",
|
|
expectedVer: "",
|
|
expectedDigest: "",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ver, digest := parseVersionDigest(tt.version)
|
|
assert.Equal(t, tt.expectedVer, ver, "version mismatch")
|
|
assert.Equal(t, tt.expectedDigest, digest, "digest mismatch")
|
|
})
|
|
}
|
|
}
|