unverifiedblockstore_test.go

package unverifiedblockstore

import (
	"bytes"
	"io"
	"testing"

	"github.com/ipld/go-ipld-prime"
	cidlink "github.com/ipld/go-ipld-prime/linking/cid"
	"github.com/stretchr/testify/require"
	"go.opentelemetry.io/otel/trace"

	"github.com/ipfs/go-graphsync/testutil"
)

func TestVerifyBlockPresent(t *testing.T) {
	blocksWritten := make(map[ipld.Link][]byte)
	lsys := testutil.NewTestStore(blocksWritten)
	unverifiedBlockStore := New(lsys.StorageWriteOpener)
	block := testutil.GenerateBlocksOfSize(1, 100)[0]
	reader, err := lsys.StorageReadOpener(ipld.LinkContext{}, cidlink.Link{Cid: block.Cid()})
	require.Nil(t, reader)
	require.Error(t, err, "block should not be loadable till it's verified and stored")

	data, err := unverifiedBlockStore.VerifyBlock(cidlink.Link{Cid: block.Cid()}, ipld.LinkContext{})
	require.Nil(t, data)
	require.Error(t, err, "block should not be verifiable till it's added as an unverifiable block")

	unverifiedBlockStore.AddUnverifiedBlock(trace.Link{}, cidlink.Link{Cid: block.Cid()}, block.RawData())
	reader, err = lsys.StorageReadOpener(ipld.LinkContext{}, cidlink.Link{Cid: block.Cid()})
	require.Nil(t, reader)
	require.Error(t, err, "block should not be loadable till it's verified")

	data, err = unverifiedBlockStore.VerifyBlock(cidlink.Link{Cid: block.Cid()}, ipld.LinkContext{})
	require.NoError(t, err)
	require.Equal(t, block.RawData(), data, "block should be returned on verification if added")

	reader, err = lsys.StorageReadOpener(ipld.LinkContext{}, cidlink.Link{Cid: block.Cid()})
	require.NoError(t, err)
	var buffer bytes.Buffer
	_, err = io.Copy(&buffer, reader)
	require.NoError(t, err)
	require.Equal(t, block.RawData(), buffer.Bytes(), "block should be stored and loadable after verification")
	data, err = unverifiedBlockStore.VerifyBlock(cidlink.Link{Cid: block.Cid()}, ipld.LinkContext{})
	require.Nil(t, data)
	require.Error(t, err, "block cannot be verified twice")
}