adc23d5f96
git-subtree-dir: x git-subtree-mainline: 7d05a6ee8f44b314fa697a427439e5fa4d78c3d7 git-subtree-split: a10a11b9d371f36b7c3510da32a1d70b74e27bd1
81 lines
2.6 KiB
Go
81 lines
2.6 KiB
Go
package blob
|
|
|
|
import "testing"
|
|
|
|
// test refs
|
|
const (
|
|
refTooLong = "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
|
|
)
|
|
|
|
var testRefs = map[string]Ref{
|
|
"mistral:latest": {name: "mistral", tag: "latest"},
|
|
"mistral": {name: "mistral"},
|
|
"mistral:30B": {name: "mistral", tag: "30B"},
|
|
"mistral:7b": {name: "mistral", tag: "7b"},
|
|
"mistral:7b+Q4_0": {name: "mistral", tag: "7b", build: "Q4_0"},
|
|
"mistral+KQED": {name: "mistral", build: "KQED"},
|
|
"mistral.x-3:7b+Q4_0": {name: "mistral.x-3", tag: "7b", build: "Q4_0"},
|
|
"mistral:7b+q4_0": {name: "mistral", tag: "7b", build: "Q4_0"},
|
|
"llama2": {name: "llama2"},
|
|
|
|
// invalid
|
|
"mistral:7b+Q4_0:latest": {},
|
|
"mi tral": {},
|
|
}
|
|
|
|
func TestRefParts(t *testing.T) {
|
|
const wantNumParts = 5
|
|
var ref Ref
|
|
if len(ref.Parts()) != wantNumParts {
|
|
t.Errorf("Parts() = %d; want %d", len(ref.Parts()), wantNumParts)
|
|
}
|
|
}
|
|
|
|
func TestParseRef(t *testing.T) {
|
|
for s, want := range testRefs {
|
|
t.Run(s, func(t *testing.T) {
|
|
got := ParseRef(s)
|
|
if got != want {
|
|
t.Errorf("ParseRef(%q) = %q; want %q", s, got, want)
|
|
}
|
|
|
|
// test round-trip
|
|
if ParseRef(got.String()) != got {
|
|
t.Errorf("String() = %q; want %q", got.String(), s)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRefFull(t *testing.T) {
|
|
const empty = "!(MISSING DOMAIN)/!(MISSING NAMESPACE)/!(MISSING NAME):!(MISSING TAG)+!(MISSING BUILD)"
|
|
|
|
cases := []struct {
|
|
in string
|
|
wantFull string
|
|
}{
|
|
{"", empty},
|
|
{"example.com/mistral:7b+x", "!(MISSING DOMAIN)/example.com/mistral:7b+X"},
|
|
{"example.com/mistral:7b+Q4_0", "!(MISSING DOMAIN)/example.com/mistral:7b+Q4_0"},
|
|
{"example.com/x/mistral:latest", "example.com/x/mistral:latest+!(MISSING BUILD)"},
|
|
{"example.com/x/mistral:latest+Q4_0", "example.com/x/mistral:latest+Q4_0"},
|
|
|
|
{"mistral:7b+x", "!(MISSING DOMAIN)/!(MISSING NAMESPACE)/mistral:7b+X"},
|
|
{"mistral:7b+q4_0", "!(MISSING DOMAIN)/!(MISSING NAMESPACE)/mistral:7b+Q4_0"},
|
|
{"mistral:7b+Q4_0", "!(MISSING DOMAIN)/!(MISSING NAMESPACE)/mistral:7b+Q4_0"},
|
|
{"mistral:latest", "!(MISSING DOMAIN)/!(MISSING NAMESPACE)/mistral:latest+!(MISSING BUILD)"},
|
|
{"mistral", "!(MISSING DOMAIN)/!(MISSING NAMESPACE)/mistral:!(MISSING TAG)+!(MISSING BUILD)"},
|
|
{"mistral:30b", "!(MISSING DOMAIN)/!(MISSING NAMESPACE)/mistral:30b+!(MISSING BUILD)"},
|
|
}
|
|
|
|
for _, tt := range cases {
|
|
t.Run(tt.in, func(t *testing.T) {
|
|
ref := ParseRef(tt.in)
|
|
t.Logf("ParseRef(%q) = %#v", tt.in, ref)
|
|
if g := ref.Full(); g != tt.wantFull {
|
|
t.Errorf("Full(%q) = %q; want %q", tt.in, g, tt.wantFull)
|
|
}
|
|
})
|
|
}
|
|
}
|