package image

import (
	"testing"

	"github.com/docker/docker/layer"
	"github.com/docker/docker/pkg/testutil/assert"
	"github.com/opencontainers/go-digest"
)

func TestRestore(t *testing.T) {
	fs, cleanup := defaultFSStoreBackend(t)
	defer cleanup()

	id1, err := fs.Set([]byte(`{"comment": "abc", "rootfs": {"type": "layers"}}`))
	assert.NilError(t, err)

	_, err = fs.Set([]byte(`invalid`))
	assert.NilError(t, err)

	id2, err := fs.Set([]byte(`{"comment": "def", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
	assert.NilError(t, err)

	err = fs.SetMetadata(id2, "parent", []byte(id1))
	assert.NilError(t, err)

	is, err := NewImageStore(fs, &mockLayerGetReleaser{})
	assert.NilError(t, err)

	assert.Equal(t, len(is.Map()), 2)

	img1, err := is.Get(ID(id1))
	assert.NilError(t, err)
	assert.Equal(t, img1.computedID, ID(id1))
	assert.Equal(t, img1.computedID.String(), string(id1))

	img2, err := is.Get(ID(id2))
	assert.NilError(t, err)
	assert.Equal(t, img1.Comment, "abc")
	assert.Equal(t, img2.Comment, "def")

	p, err := is.GetParent(ID(id1))
	assert.Error(t, err, "failed to read metadata")

	p, err = is.GetParent(ID(id2))
	assert.NilError(t, err)
	assert.Equal(t, p, ID(id1))

	children := is.Children(ID(id1))
	assert.Equal(t, len(children), 1)
	assert.Equal(t, children[0], ID(id2))
	assert.Equal(t, len(is.Heads()), 1)

	sid1, err := is.Search(string(id1)[:10])
	assert.NilError(t, err)
	assert.Equal(t, sid1, ID(id1))

	sid1, err = is.Search(digest.Digest(id1).Hex()[:6])
	assert.NilError(t, err)
	assert.Equal(t, sid1, ID(id1))

	invalidPattern := digest.Digest(id1).Hex()[1:6]
	_, err = is.Search(invalidPattern)
	assert.Error(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.NilError(t, err)
	assert.Equal(t, id1, ID("sha256:8d25a9c45df515f9d0fe8e4a6b1c64dd3b965a84790ddbcc7954bb9bc89eb993"))

	img, err := is.Get(id1)
	assert.NilError(t, err)
	assert.Equal(t, img.Comment, "abc")

	id2, err := is.Create([]byte(`{"comment": "def", "rootfs": {"type": "layers", "diff_ids": ["2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae"]}}`))
	assert.NilError(t, err)

	err = is.SetParent(id2, id1)
	assert.NilError(t, err)

	pid1, err := is.GetParent(id2)
	assert.NilError(t, err)
	assert.Equal(t, pid1, id1)

	_, err = is.Delete(id1)
	assert.NilError(t, err)

	_, err = is.Get(id1)
	assert.Error(t, err, "failed to get digest")

	_, err = is.Get(id2)
	assert.NilError(t, err)

	_, err = is.GetParent(id2)
	assert.Error(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.NilError(t, err)

	id1, err := is.Search(string(id)[:15])
	assert.NilError(t, err)
	assert.Equal(t, id1, id)

	_, err = is.Delete(id)
	assert.NilError(t, err)

	_, err = is.Search(string(id)[:15])
	assert.Error(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.NilError(t, err)

	id2, err := is.Create([]byte(`{"comment": "abc2", "rootfs": {"type": "layers"}}`))
	assert.NilError(t, err)

	id3, err := is.Create([]byte(`{"comment": "abc3", "rootfs": {"type": "layers"}}`))
	assert.NilError(t, err)

	assert.NilError(t, is.SetParent(id, id2))
	assert.Equal(t, len(is.Children(id2)), 1)

	assert.NilError(t, is.SetParent(id, id3))
	assert.Equal(t, len(is.Children(id2)), 0)
	assert.Equal(t, len(is.Children(id3)), 1)
}

func defaultImageStore(t *testing.T) (Store, func()) {
	fsBackend, cleanup := defaultFSStoreBackend(t)

	store, err := NewImageStore(fsBackend, &mockLayerGetReleaser{})
	assert.NilError(t, err)

	return store, cleanup
}

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
}