node_test.go

package restore

import (
	"testing"

	format "github.com/ipfs/go-ipld-format"
	"github.com/ipfs/go-merkledag"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestNodeRedundant(t *testing.T) {
	nd := NewNode(merkledag.NodeWithData([]byte("1234567890")))
	r := merkledag.NewRawNode([]byte("12345"))
	nd.AddRedundantNode(r)
	assert.Len(t, nd.Redundant(), 1)
	nd.RemoveRedundantNode(r.Cid())
	assert.Len(t, nd.Redundant(), 0)
}

func TestNodeMarshalUnmarshal(t *testing.T) {
	in := NewNode(merkledag.NodeWithData([]byte("1234567890")))
	in.AddRedundantNode(merkledag.NewRawNode([]byte("12345")))

	data, err := MarshalNode(in)
	require.NoError(t, err)

	out, err := UnmarshalNode(data)
	require.NoError(t, err)

	out.SetCidBuilder(in.CidBuilder())
	assert.True(t, in.Cid().Equals(out.Cid()))
}

func TestNodeDecode(t *testing.T) {
	in := NewNode(merkledag.NodeWithData([]byte("1234567890")))
	out, err := format.Decode(in)
	require.NoError(t, err)
	assert.True(t, in.Cid().Equals(out.Cid()))
}

func TestNodeCache(t *testing.T) {
	nd := NewNode(merkledag.NodeWithData([]byte("1234567890")))
	nd.AddRedundantNode(merkledag.NewRawNode([]byte("12345")))
	assert.Zero(t, nd.cid)
	assert.Nil(t, nd.cache)

	nd.Cid()
	assert.NotZero(t, nd.cid)
	assert.NotNil(t, nd.cache)

	nd.AddNodeLink("", merkledag.NewRawNode([]byte("12345")))
	assert.Nil(t, nd.cache)
}

func TestNode_Copy(t *testing.T) {
	nd := NewNode(merkledag.NodeWithData([]byte("1234567890")))
	r := merkledag.NewRawNode([]byte("12345"))
	nd.AddRedundantNode(r)

	cp := nd.Copy()
	nd.SetData([]byte{})
	nd.RemoveRedundantNode(r.Cid())

	assert.NotNil(t, cp.(*Node).Redundant())
	assert.NotNil(t, cp.(*Node).Data())
}