Initial commit
This commit is contained in:
Ria Bhatia
356
vendor/github.com/hyperhq/hypercli/reference/store_test.go
generated
vendored
Normal file
356
vendor/github.com/hyperhq/hypercli/reference/store_test.go
generated
vendored
Normal file
@@ -0,0 +1,356 @@
|
||||
package reference
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/hyperhq/hypercli/image"
|
||||
)
|
||||
|
||||
var (
|
||||
saveLoadTestCases = map[string]image.ID{
|
||||
"registry:5000/foobar:HEAD": "sha256:470022b8af682154f57a2163d030eb369549549cba00edc69e1b99b46bb924d6",
|
||||
"registry:5000/foobar:alternate": "sha256:ae300ebc4a4f00693702cfb0a5e0b7bc527b353828dc86ad09fb95c8a681b793",
|
||||
"registry:5000/foobar:latest": "sha256:6153498b9ac00968d71b66cca4eac37e990b5f9eb50c26877eb8799c8847451b",
|
||||
"registry:5000/foobar:master": "sha256:6c9917af4c4e05001b346421959d7ea81b6dc9d25718466a37a6add865dfd7fc",
|
||||
"jess/hollywood:latest": "sha256:ae7a5519a0a55a2d4ef20ddcbd5d0ca0888a1f7ab806acc8e2a27baf46f529fe",
|
||||
"registry@sha256:367eb40fd0330a7e464777121e39d2f5b3e8e23a1e159342e53ab05c9e4d94e6": "sha256:24126a56805beb9711be5f4590cc2eb55ab8d4a85ebd618eed72bb19fc50631c",
|
||||
"busybox:latest": "sha256:91e54dfb11794fad694460162bf0cb0a4fa710cfa3f60979c177d920813e267c",
|
||||
}
|
||||
|
||||
marshalledSaveLoadTestCases = []byte(`{"Repositories":{"busybox":{"busybox:latest":"sha256:91e54dfb11794fad694460162bf0cb0a4fa710cfa3f60979c177d920813e267c"},"jess/hollywood":{"jess/hollywood:latest":"sha256:ae7a5519a0a55a2d4ef20ddcbd5d0ca0888a1f7ab806acc8e2a27baf46f529fe"},"registry":{"registry@sha256:367eb40fd0330a7e464777121e39d2f5b3e8e23a1e159342e53ab05c9e4d94e6":"sha256:24126a56805beb9711be5f4590cc2eb55ab8d4a85ebd618eed72bb19fc50631c"},"registry:5000/foobar":{"registry:5000/foobar:HEAD":"sha256:470022b8af682154f57a2163d030eb369549549cba00edc69e1b99b46bb924d6","registry:5000/foobar:alternate":"sha256:ae300ebc4a4f00693702cfb0a5e0b7bc527b353828dc86ad09fb95c8a681b793","registry:5000/foobar:latest":"sha256:6153498b9ac00968d71b66cca4eac37e990b5f9eb50c26877eb8799c8847451b","registry:5000/foobar:master":"sha256:6c9917af4c4e05001b346421959d7ea81b6dc9d25718466a37a6add865dfd7fc"}}}`)
|
||||
)
|
||||
|
||||
func TestLoad(t *testing.T) {
|
||||
jsonFile, err := ioutil.TempFile("", "tag-store-test")
|
||||
if err != nil {
|
||||
t.Fatalf("error creating temp file: %v", err)
|
||||
}
|
||||
defer os.RemoveAll(jsonFile.Name())
|
||||
|
||||
// Write canned json to the temp file
|
||||
_, err = jsonFile.Write(marshalledSaveLoadTestCases)
|
||||
if err != nil {
|
||||
t.Fatalf("error writing to temp file: %v", err)
|
||||
}
|
||||
jsonFile.Close()
|
||||
|
||||
store, err := NewReferenceStore(jsonFile.Name())
|
||||
if err != nil {
|
||||
t.Fatalf("error creating tag store: %v", err)
|
||||
}
|
||||
|
||||
for refStr, expectedID := range saveLoadTestCases {
|
||||
ref, err := ParseNamed(refStr)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to parse reference: %v", err)
|
||||
}
|
||||
id, err := store.Get(ref)
|
||||
if err != nil {
|
||||
t.Fatalf("could not find reference %s: %v", refStr, err)
|
||||
}
|
||||
if id != expectedID {
|
||||
t.Fatalf("expected %s - got %s", expectedID, id)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestSave(t *testing.T) {
|
||||
jsonFile, err := ioutil.TempFile("", "tag-store-test")
|
||||
if err != nil {
|
||||
t.Fatalf("error creating temp file: %v", err)
|
||||
}
|
||||
_, err = jsonFile.Write([]byte(`{}`))
|
||||
jsonFile.Close()
|
||||
defer os.RemoveAll(jsonFile.Name())
|
||||
|
||||
store, err := NewReferenceStore(jsonFile.Name())
|
||||
if err != nil {
|
||||
t.Fatalf("error creating tag store: %v", err)
|
||||
}
|
||||
|
||||
for refStr, id := range saveLoadTestCases {
|
||||
ref, err := ParseNamed(refStr)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to parse reference: %v", err)
|
||||
}
|
||||
if canonical, ok := ref.(Canonical); ok {
|
||||
err = store.AddDigest(canonical, id, false)
|
||||
if err != nil {
|
||||
t.Fatalf("could not add digest reference %s: %v", refStr, err)
|
||||
}
|
||||
} else {
|
||||
err = store.AddTag(ref, id, false)
|
||||
if err != nil {
|
||||
t.Fatalf("could not add reference %s: %v", refStr, err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
jsonBytes, err := ioutil.ReadFile(jsonFile.Name())
|
||||
if err != nil {
|
||||
t.Fatalf("could not read json file: %v", err)
|
||||
}
|
||||
|
||||
if !bytes.Equal(jsonBytes, marshalledSaveLoadTestCases) {
|
||||
t.Fatalf("save output did not match expectations\nexpected:\n%s\ngot:\n%s", marshalledSaveLoadTestCases, jsonBytes)
|
||||
}
|
||||
}
|
||||
|
||||
func TestAddDeleteGet(t *testing.T) {
|
||||
jsonFile, err := ioutil.TempFile("", "tag-store-test")
|
||||
if err != nil {
|
||||
t.Fatalf("error creating temp file: %v", err)
|
||||
}
|
||||
_, err = jsonFile.Write([]byte(`{}`))
|
||||
jsonFile.Close()
|
||||
defer os.RemoveAll(jsonFile.Name())
|
||||
|
||||
store, err := NewReferenceStore(jsonFile.Name())
|
||||
if err != nil {
|
||||
t.Fatalf("error creating tag store: %v", err)
|
||||
}
|
||||
|
||||
testImageID1 := image.ID("sha256:9655aef5fd742a1b4e1b7b163aa9f1c76c186304bf39102283d80927c916ca9c")
|
||||
testImageID2 := image.ID("sha256:9655aef5fd742a1b4e1b7b163aa9f1c76c186304bf39102283d80927c916ca9d")
|
||||
testImageID3 := image.ID("sha256:9655aef5fd742a1b4e1b7b163aa9f1c76c186304bf39102283d80927c916ca9e")
|
||||
|
||||
// Try adding a reference with no tag or digest
|
||||
nameOnly, err := WithName("username/repo")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if err = store.AddTag(nameOnly, testImageID1, false); err != nil {
|
||||
t.Fatalf("error adding to store: %v", err)
|
||||
}
|
||||
|
||||
// Add a few references
|
||||
ref1, err := ParseNamed("username/repo1:latest")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if err = store.AddTag(ref1, testImageID1, false); err != nil {
|
||||
t.Fatalf("error adding to store: %v", err)
|
||||
}
|
||||
|
||||
ref2, err := ParseNamed("username/repo1:old")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if err = store.AddTag(ref2, testImageID2, false); err != nil {
|
||||
t.Fatalf("error adding to store: %v", err)
|
||||
}
|
||||
|
||||
ref3, err := ParseNamed("username/repo1:alias")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if err = store.AddTag(ref3, testImageID1, false); err != nil {
|
||||
t.Fatalf("error adding to store: %v", err)
|
||||
}
|
||||
|
||||
ref4, err := ParseNamed("username/repo2:latest")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if err = store.AddTag(ref4, testImageID2, false); err != nil {
|
||||
t.Fatalf("error adding to store: %v", err)
|
||||
}
|
||||
|
||||
ref5, err := ParseNamed("username/repo3@sha256:58153dfb11794fad694460162bf0cb0a4fa710cfa3f60979c177d920813e267c")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if err = store.AddDigest(ref5.(Canonical), testImageID2, false); err != nil {
|
||||
t.Fatalf("error adding to store: %v", err)
|
||||
}
|
||||
|
||||
// Attempt to overwrite with force == false
|
||||
if err = store.AddTag(ref4, testImageID3, false); err == nil || !strings.HasPrefix(err.Error(), "Conflict:") {
|
||||
t.Fatalf("did not get expected error on overwrite attempt - got %v", err)
|
||||
}
|
||||
// Repeat to overwrite with force == true
|
||||
if err = store.AddTag(ref4, testImageID3, true); err != nil {
|
||||
t.Fatalf("failed to force tag overwrite: %v", err)
|
||||
}
|
||||
|
||||
// Check references so far
|
||||
id, err := store.Get(nameOnly)
|
||||
if err != nil {
|
||||
t.Fatalf("Get returned error: %v", err)
|
||||
}
|
||||
if id != testImageID1 {
|
||||
t.Fatalf("id mismatch: got %s instead of %s", id.String(), testImageID1.String())
|
||||
}
|
||||
|
||||
id, err = store.Get(ref1)
|
||||
if err != nil {
|
||||
t.Fatalf("Get returned error: %v", err)
|
||||
}
|
||||
if id != testImageID1 {
|
||||
t.Fatalf("id mismatch: got %s instead of %s", id.String(), testImageID1.String())
|
||||
}
|
||||
|
||||
id, err = store.Get(ref2)
|
||||
if err != nil {
|
||||
t.Fatalf("Get returned error: %v", err)
|
||||
}
|
||||
if id != testImageID2 {
|
||||
t.Fatalf("id mismatch: got %s instead of %s", id.String(), testImageID2.String())
|
||||
}
|
||||
|
||||
id, err = store.Get(ref3)
|
||||
if err != nil {
|
||||
t.Fatalf("Get returned error: %v", err)
|
||||
}
|
||||
if id != testImageID1 {
|
||||
t.Fatalf("id mismatch: got %s instead of %s", id.String(), testImageID1.String())
|
||||
}
|
||||
|
||||
id, err = store.Get(ref4)
|
||||
if err != nil {
|
||||
t.Fatalf("Get returned error: %v", err)
|
||||
}
|
||||
if id != testImageID3 {
|
||||
t.Fatalf("id mismatch: got %s instead of %s", id.String(), testImageID3.String())
|
||||
}
|
||||
|
||||
id, err = store.Get(ref5)
|
||||
if err != nil {
|
||||
t.Fatalf("Get returned error: %v", err)
|
||||
}
|
||||
if id != testImageID2 {
|
||||
t.Fatalf("id mismatch: got %s instead of %s", id.String(), testImageID3.String())
|
||||
}
|
||||
|
||||
// Get should return ErrDoesNotExist for a nonexistent repo
|
||||
nonExistRepo, err := ParseNamed("username/nonexistrepo:latest")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if _, err = store.Get(nonExistRepo); err != ErrDoesNotExist {
|
||||
t.Fatal("Expected ErrDoesNotExist from Get")
|
||||
}
|
||||
|
||||
// Get should return ErrDoesNotExist for a nonexistent tag
|
||||
nonExistTag, err := ParseNamed("username/repo1:nonexist")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
if _, err = store.Get(nonExistTag); err != ErrDoesNotExist {
|
||||
t.Fatal("Expected ErrDoesNotExist from Get")
|
||||
}
|
||||
|
||||
// Check References
|
||||
refs := store.References(testImageID1)
|
||||
if len(refs) != 3 {
|
||||
t.Fatal("unexpected number of references")
|
||||
}
|
||||
// Looking for the references in this order verifies that they are
|
||||
// returned lexically sorted.
|
||||
if refs[0].String() != ref3.String() {
|
||||
t.Fatalf("unexpected reference: %v", refs[0].String())
|
||||
}
|
||||
if refs[1].String() != ref1.String() {
|
||||
t.Fatalf("unexpected reference: %v", refs[1].String())
|
||||
}
|
||||
if refs[2].String() != nameOnly.String()+":latest" {
|
||||
t.Fatalf("unexpected reference: %v", refs[2].String())
|
||||
}
|
||||
|
||||
// Check ReferencesByName
|
||||
repoName, err := WithName("username/repo1")
|
||||
if err != nil {
|
||||
t.Fatalf("could not parse reference: %v", err)
|
||||
}
|
||||
associations := store.ReferencesByName(repoName)
|
||||
if len(associations) != 3 {
|
||||
t.Fatal("unexpected number of associations")
|
||||
}
|
||||
// Looking for the associations in this order verifies that they are
|
||||
// returned lexically sorted.
|
||||
if associations[0].Ref.String() != ref3.String() {
|
||||
t.Fatalf("unexpected reference: %v", associations[0].Ref.String())
|
||||
}
|
||||
if associations[0].ImageID != testImageID1 {
|
||||
t.Fatalf("unexpected reference: %v", associations[0].Ref.String())
|
||||
}
|
||||
if associations[1].Ref.String() != ref1.String() {
|
||||
t.Fatalf("unexpected reference: %v", associations[1].Ref.String())
|
||||
}
|
||||
if associations[1].ImageID != testImageID1 {
|
||||
t.Fatalf("unexpected reference: %v", associations[1].Ref.String())
|
||||
}
|
||||
if associations[2].Ref.String() != ref2.String() {
|
||||
t.Fatalf("unexpected reference: %v", associations[2].Ref.String())
|
||||
}
|
||||
if associations[2].ImageID != testImageID2 {
|
||||
t.Fatalf("unexpected reference: %v", associations[2].Ref.String())
|
||||
}
|
||||
|
||||
// Delete should return ErrDoesNotExist for a nonexistent repo
|
||||
if _, err = store.Delete(nonExistRepo); err != ErrDoesNotExist {
|
||||
t.Fatal("Expected ErrDoesNotExist from Delete")
|
||||
}
|
||||
|
||||
// Delete should return ErrDoesNotExist for a nonexistent tag
|
||||
if _, err = store.Delete(nonExistTag); err != ErrDoesNotExist {
|
||||
t.Fatal("Expected ErrDoesNotExist from Delete")
|
||||
}
|
||||
|
||||
// Delete a few references
|
||||
if deleted, err := store.Delete(ref1); err != nil || deleted != true {
|
||||
t.Fatal("Delete failed")
|
||||
}
|
||||
if _, err := store.Get(ref1); err != ErrDoesNotExist {
|
||||
t.Fatal("Expected ErrDoesNotExist from Get")
|
||||
}
|
||||
if deleted, err := store.Delete(ref5); err != nil || deleted != true {
|
||||
t.Fatal("Delete failed")
|
||||
}
|
||||
if _, err := store.Get(ref5); err != ErrDoesNotExist {
|
||||
t.Fatal("Expected ErrDoesNotExist from Get")
|
||||
}
|
||||
if deleted, err := store.Delete(nameOnly); err != nil || deleted != true {
|
||||
t.Fatal("Delete failed")
|
||||
}
|
||||
if _, err := store.Get(nameOnly); err != ErrDoesNotExist {
|
||||
t.Fatal("Expected ErrDoesNotExist from Get")
|
||||
}
|
||||
}
|
||||
|
||||
func TestInvalidTags(t *testing.T) {
|
||||
tmpDir, err := ioutil.TempDir("", "tag-store-test")
|
||||
defer os.RemoveAll(tmpDir)
|
||||
|
||||
store, err := NewReferenceStore(filepath.Join(tmpDir, "repositories.json"))
|
||||
if err != nil {
|
||||
t.Fatalf("error creating tag store: %v", err)
|
||||
}
|
||||
id := image.ID("sha256:470022b8af682154f57a2163d030eb369549549cba00edc69e1b99b46bb924d6")
|
||||
|
||||
// sha256 as repo name
|
||||
ref, err := ParseNamed("sha256:abc")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
err = store.AddTag(ref, id, true)
|
||||
if err == nil {
|
||||
t.Fatalf("expected setting tag %q to fail", ref)
|
||||
}
|
||||
|
||||
// setting digest as a tag
|
||||
ref, err = ParseNamed("registry@sha256:367eb40fd0330a7e464777121e39d2f5b3e8e23a1e159342e53ab05c9e4d94e6")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
err = store.AddTag(ref, id, true)
|
||||
if err == nil {
|
||||
t.Fatalf("expected setting digest %q to fail", ref)
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user