mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
179 lines
4.7 KiB
Go
179 lines
4.7 KiB
Go
![]() |
package image
|
||
|
|
||
|
import (
|
||
|
"runtime"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/docker/docker/internal/testutil"
|
||
|
"github.com/docker/docker/layer"
|
||
|
"github.com/opencontainers/go-digest"
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
)
|
||
|
|
||
|
func TestRestore(t *testing.T) {
|
||
|
fs, cleanup := defaultFSStoreBackend(t)
|
||
|
defer cleanup()
|
||
|
|
||
|
id1, err := fs.Set([]byte(`{"comment": "abc", "rootfs": {"type": "layers"}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
_, err = fs.Set([]byte(`invalid`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
id2, err := fs.Set([]byte(`{"comment": "def", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
err = fs.SetMetadata(id2, "parent", []byte(id1))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
is, err := NewImageStore(fs, runtime.GOOS, &mockLayerGetReleaser{})
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
assert.Len(t, is.Map(), 2)
|
||
|
|
||
|
img1, err := is.Get(ID(id1))
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, ID(id1), img1.computedID)
|
||
|
assert.Equal(t, string(id1), img1.computedID.String())
|
||
|
|
||
|
img2, err := is.Get(ID(id2))
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, "abc", img1.Comment)
|
||
|
assert.Equal(t, "def", img2.Comment)
|
||
|
|
||
|
_, err = is.GetParent(ID(id1))
|
||
|
testutil.ErrorContains(t, err, "failed to read metadata")
|
||
|
|
||
|
p, err := is.GetParent(ID(id2))
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, ID(id1), p)
|
||
|
|
||
|
children := is.Children(ID(id1))
|
||
|
assert.Len(t, children, 1)
|
||
|
assert.Equal(t, ID(id2), children[0])
|
||
|
assert.Len(t, is.Heads(), 1)
|
||
|
|
||
|
sid1, err := is.Search(string(id1)[:10])
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, ID(id1), sid1)
|
||
|
|
||
|
sid1, err = is.Search(digest.Digest(id1).Hex()[:6])
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, ID(id1), sid1)
|
||
|
|
||
|
invalidPattern := digest.Digest(id1).Hex()[1:6]
|
||
|
_, err = is.Search(invalidPattern)
|
||
|
testutil.ErrorContains(t, err, "No such image")
|
||
|
}
|
||
|
|
||
|
func TestAddDelete(t *testing.T) {
|
||
|
is, cleanup := defaultImageStore(t)
|
||
|
defer cleanup()
|
||
|
|
||
|
id1, err := is.Create([]byte(`{"comment": "abc", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, ID("sha256:8d25a9c45df515f9d0fe8e4a6b1c64dd3b965a84790ddbcc7954bb9bc89eb993"), id1)
|
||
|
|
||
|
img, err := is.Get(id1)
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, "abc", img.Comment)
|
||
|
|
||
|
id2, err := is.Create([]byte(`{"comment": "def", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
err = is.SetParent(id2, id1)
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
pid1, err := is.GetParent(id2)
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, pid1, id1)
|
||
|
|
||
|
_, err = is.Delete(id1)
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
_, err = is.Get(id1)
|
||
|
testutil.ErrorContains(t, err, "failed to get digest")
|
||
|
|
||
|
_, err = is.Get(id2)
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
_, err = is.GetParent(id2)
|
||
|
testutil.ErrorContains(t, err, "failed to read metadata")
|
||
|
}
|
||
|
|
||
|
func TestSearchAfterDelete(t *testing.T) {
|
||
|
is, cleanup := defaultImageStore(t)
|
||
|
defer cleanup()
|
||
|
|
||
|
id, err := is.Create([]byte(`{"comment": "abc", "rootfs": {"type": "layers"}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
id1, err := is.Search(string(id)[:15])
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, id1, id)
|
||
|
|
||
|
_, err = is.Delete(id)
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
_, err = is.Search(string(id)[:15])
|
||
|
testutil.ErrorContains(t, err, "No such image")
|
||
|
}
|
||
|
|
||
|
func TestParentReset(t *testing.T) {
|
||
|
is, cleanup := defaultImageStore(t)
|
||
|
defer cleanup()
|
||
|
|
||
|
id, err := is.Create([]byte(`{"comment": "abc1", "rootfs": {"type": "layers"}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
id2, err := is.Create([]byte(`{"comment": "abc2", "rootfs": {"type": "layers"}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
id3, err := is.Create([]byte(`{"comment": "abc3", "rootfs": {"type": "layers"}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
assert.NoError(t, is.SetParent(id, id2))
|
||
|
assert.Len(t, is.Children(id2), 1)
|
||
|
|
||
|
assert.NoError(t, is.SetParent(id, id3))
|
||
|
assert.Len(t, is.Children(id2), 0)
|
||
|
assert.Len(t, is.Children(id3), 1)
|
||
|
}
|
||
|
|
||
|
func defaultImageStore(t *testing.T) (Store, func()) {
|
||
|
fsBackend, cleanup := defaultFSStoreBackend(t)
|
||
|
|
||
|
store, err := NewImageStore(fsBackend, runtime.GOOS, &mockLayerGetReleaser{})
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
return store, cleanup
|
||
|
}
|
||
|
|
||
|
func TestGetAndSetLastUpdated(t *testing.T) {
|
||
|
store, cleanup := defaultImageStore(t)
|
||
|
defer cleanup()
|
||
|
|
||
|
id, err := store.Create([]byte(`{"comment": "abc1", "rootfs": {"type": "layers"}}`))
|
||
|
assert.NoError(t, err)
|
||
|
|
||
|
updated, err := store.GetLastUpdated(id)
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, updated.IsZero(), true)
|
||
|
|
||
|
assert.NoError(t, store.SetLastUpdated(id))
|
||
|
|
||
|
updated, err = store.GetLastUpdated(id)
|
||
|
assert.NoError(t, err)
|
||
|
assert.Equal(t, updated.IsZero(), false)
|
||
|
}
|
||
|
|
||
|
type mockLayerGetReleaser struct{}
|
||
|
|
||
|
func (ls *mockLayerGetReleaser) Get(layer.ChainID) (layer.Layer, error) {
|
||
|
return nil, nil
|
||
|
}
|
||
|
|
||
|
func (ls *mockLayerGetReleaser) Release(layer.Layer) ([]layer.Metadata, error) {
|
||
|
return nil, nil
|
||
|
}
|