github.com/jiasir/docker@v1.3.3-0.20170609024000-252e610103e7/image/store_test.go (about)

     1  package image
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/docker/docker/layer"
     7  	"github.com/docker/docker/pkg/testutil"
     8  	"github.com/opencontainers/go-digest"
     9  	"github.com/stretchr/testify/assert"
    10  )
    11  
    12  func TestRestore(t *testing.T) {
    13  	fs, cleanup := defaultFSStoreBackend(t)
    14  	defer cleanup()
    15  
    16  	id1, err := fs.Set([]byte(`{"comment": "abc", "rootfs": {"type": "layers"}}`))
    17  	assert.NoError(t, err)
    18  
    19  	_, err = fs.Set([]byte(`invalid`))
    20  	assert.NoError(t, err)
    21  
    22  	id2, err := fs.Set([]byte(`{"comment": "def", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
    23  	assert.NoError(t, err)
    24  
    25  	err = fs.SetMetadata(id2, "parent", []byte(id1))
    26  	assert.NoError(t, err)
    27  
    28  	is, err := NewImageStore(fs, &mockLayerGetReleaser{})
    29  	assert.NoError(t, err)
    30  
    31  	assert.Len(t, is.Map(), 2)
    32  
    33  	img1, err := is.Get(ID(id1))
    34  	assert.NoError(t, err)
    35  	assert.Equal(t, ID(id1), img1.computedID)
    36  	assert.Equal(t, string(id1), img1.computedID.String())
    37  
    38  	img2, err := is.Get(ID(id2))
    39  	assert.NoError(t, err)
    40  	assert.Equal(t, "abc", img1.Comment)
    41  	assert.Equal(t, "def", img2.Comment)
    42  
    43  	p, err := is.GetParent(ID(id1))
    44  	testutil.ErrorContains(t, err, "failed to read metadata")
    45  
    46  	p, err = is.GetParent(ID(id2))
    47  	assert.NoError(t, err)
    48  	assert.Equal(t, ID(id1), p)
    49  
    50  	children := is.Children(ID(id1))
    51  	assert.Len(t, children, 1)
    52  	assert.Equal(t, ID(id2), children[0])
    53  	assert.Len(t, is.Heads(), 1)
    54  
    55  	sid1, err := is.Search(string(id1)[:10])
    56  	assert.NoError(t, err)
    57  	assert.Equal(t, ID(id1), sid1)
    58  
    59  	sid1, err = is.Search(digest.Digest(id1).Hex()[:6])
    60  	assert.NoError(t, err)
    61  	assert.Equal(t, ID(id1), sid1)
    62  
    63  	invalidPattern := digest.Digest(id1).Hex()[1:6]
    64  	_, err = is.Search(invalidPattern)
    65  	testutil.ErrorContains(t, err, "No such image")
    66  }
    67  
    68  func TestAddDelete(t *testing.T) {
    69  	is, cleanup := defaultImageStore(t)
    70  	defer cleanup()
    71  
    72  	id1, err := is.Create([]byte(`{"comment": "abc", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
    73  	assert.NoError(t, err)
    74  	assert.Equal(t, ID("sha256:8d25a9c45df515f9d0fe8e4a6b1c64dd3b965a84790ddbcc7954bb9bc89eb993"), id1)
    75  
    76  	img, err := is.Get(id1)
    77  	assert.NoError(t, err)
    78  	assert.Equal(t, "abc", img.Comment)
    79  
    80  	id2, err := is.Create([]byte(`{"comment": "def", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
    81  	assert.NoError(t, err)
    82  
    83  	err = is.SetParent(id2, id1)
    84  	assert.NoError(t, err)
    85  
    86  	pid1, err := is.GetParent(id2)
    87  	assert.NoError(t, err)
    88  	assert.Equal(t, pid1, id1)
    89  
    90  	_, err = is.Delete(id1)
    91  	assert.NoError(t, err)
    92  
    93  	_, err = is.Get(id1)
    94  	testutil.ErrorContains(t, err, "failed to get digest")
    95  
    96  	_, err = is.Get(id2)
    97  	assert.NoError(t, err)
    98  
    99  	_, err = is.GetParent(id2)
   100  	testutil.ErrorContains(t, err, "failed to read metadata")
   101  }
   102  
   103  func TestSearchAfterDelete(t *testing.T) {
   104  	is, cleanup := defaultImageStore(t)
   105  	defer cleanup()
   106  
   107  	id, err := is.Create([]byte(`{"comment": "abc", "rootfs": {"type": "layers"}}`))
   108  	assert.NoError(t, err)
   109  
   110  	id1, err := is.Search(string(id)[:15])
   111  	assert.NoError(t, err)
   112  	assert.Equal(t, id1, id)
   113  
   114  	_, err = is.Delete(id)
   115  	assert.NoError(t, err)
   116  
   117  	_, err = is.Search(string(id)[:15])
   118  	testutil.ErrorContains(t, err, "No such image")
   119  }
   120  
   121  func TestParentReset(t *testing.T) {
   122  	is, cleanup := defaultImageStore(t)
   123  	defer cleanup()
   124  
   125  	id, err := is.Create([]byte(`{"comment": "abc1", "rootfs": {"type": "layers"}}`))
   126  	assert.NoError(t, err)
   127  
   128  	id2, err := is.Create([]byte(`{"comment": "abc2", "rootfs": {"type": "layers"}}`))
   129  	assert.NoError(t, err)
   130  
   131  	id3, err := is.Create([]byte(`{"comment": "abc3", "rootfs": {"type": "layers"}}`))
   132  	assert.NoError(t, err)
   133  
   134  	assert.NoError(t, is.SetParent(id, id2))
   135  	assert.Len(t, is.Children(id2), 1)
   136  
   137  	assert.NoError(t, is.SetParent(id, id3))
   138  	assert.Len(t, is.Children(id2), 0)
   139  	assert.Len(t, is.Children(id3), 1)
   140  }
   141  
   142  func defaultImageStore(t *testing.T) (Store, func()) {
   143  	fsBackend, cleanup := defaultFSStoreBackend(t)
   144  
   145  	store, err := NewImageStore(fsBackend, &mockLayerGetReleaser{})
   146  	assert.NoError(t, err)
   147  
   148  	return store, cleanup
   149  }
   150  
   151  type mockLayerGetReleaser struct{}
   152  
   153  func (ls *mockLayerGetReleaser) Get(layer.ChainID) (layer.Layer, error) {
   154  	return nil, nil
   155  }
   156  
   157  func (ls *mockLayerGetReleaser) Release(layer.Layer) ([]layer.Metadata, error) {
   158  	return nil, nil
   159  }