store_test.go 4.7 KB

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