attestation_test.go

package iotago_test

import (
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/iotaledger/hive.go/lo"
	iotago "github.com/iotaledger/iota.go/v4"
	"github.com/iotaledger/iota.go/v4/builder"
	"github.com/iotaledger/iota.go/v4/tpkg"
)

func TestAttestation(t *testing.T) {
	block, err := builder.NewValidationBlockBuilder(tpkg.ZeroCostTestAPI).
		StrongParents(tpkg.SortedRandBlockIDs(2)).
		Sign(tpkg.RandAccountID(), tpkg.RandEd25519PrivateKey()).
		Build()

	require.NoError(t, err)
	require.Equal(t, iotago.BlockBodyTypeValidation, block.Body.Type())

	attestation := iotago.NewAttestation(tpkg.ZeroCostTestAPI, block)

	// Compare fields of block and attestation.
	{
		require.Equal(t, block.Header, attestation.Header)
		require.Equal(t, lo.PanicOnErr(block.Body.Hash()), attestation.BodyHash)
		require.Equal(t, block.Signature, attestation.Signature)
	}

	// Compare block ID and attestation block ID.
	{
		blockID, err := block.ID()
		require.NoError(t, err)

		blockIDFromAttestation, err := attestation.BlockID()
		require.NoError(t, err)

		require.Equal(t, blockID, blockIDFromAttestation)
	}

	// Check validity of signature.
	{
		valid, err := attestation.VerifySignature()
		require.NoError(t, err)
		require.True(t, valid)
	}
}