metadata_test.go

package metadata

import (
	"bytes"
	"math/rand"
	"testing"

	"github.com/ipld/go-ipld-prime/codec/dagcbor"
	"github.com/ipld/go-ipld-prime/fluent"
	cidlink "github.com/ipld/go-ipld-prime/linking/cid"
	basicnode "github.com/ipld/go-ipld-prime/node/basic"
	"github.com/stretchr/testify/require"

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

func TestDecodeEncodeMetadata(t *testing.T) {
	cids := testutil.GenerateCids(10)
	initialMetadata := make(Metadata, 0, 10)
	nd := fluent.MustBuildList(basicnode.Prototype.List, 10, func(fla fluent.ListAssembler) {
		for _, k := range cids {
			blockPresent := rand.Int31()%2 == 0
			initialMetadata = append(initialMetadata, Item{k, blockPresent})
			fla.AssembleValue().CreateMap(2, func(fma fluent.MapAssembler) {
				fma.AssembleEntry("link").AssignLink(cidlink.Link{Cid: k})
				fma.AssembleEntry("blockPresent").AssignBool(blockPresent)
			})
		}
	})

	// verify metadata matches
	encoded, err := EncodeMetadata(initialMetadata)
	require.NoError(t, err, "encode errored")

	decodedMetadata, err := DecodeMetadata(encoded)
	require.NoError(t, err, "decode errored")
	require.Equal(t, initialMetadata, decodedMetadata, "metadata changed during encoding and decoding")

	// verify metadata is equivalent of IPLD node encoding
	encodedNode := new(bytes.Buffer)
	err = dagcbor.Encoder(nd, encodedNode)
	require.NoError(t, err)
	decodedMetadataFromNode, err := DecodeMetadata(encodedNode.Bytes())
	require.NoError(t, err)
	require.Equal(t, decodedMetadata, decodedMetadataFromNode, "metadata not equal to IPLD encoding")

	nb := basicnode.Prototype.List.NewBuilder()
	err = dagcbor.Decoder(nb, encodedNode)
	require.NoError(t, err)
	decodedNode := nb.Build()
	require.Equal(t, nd, decodedNode)
	nb = basicnode.Prototype.List.NewBuilder()
	err = dagcbor.Decoder(nb, bytes.NewReader(encoded))
	require.NoError(t, err)
	decodedNodeFromMetadata := nb.Build()
	require.Equal(t, decodedNode, decodedNodeFromMetadata, "deserialzed metadata does not match deserialized node")
}