github.com/koko1123/flow-go-1@v0.29.6/engine/verification/verifier/engine_test.go

package verifier_test

import (
	"crypto/rand"
	"errors"
	"testing"

	"github.com/rs/zerolog"
	"github.com/stretchr/testify/mock"
	testifymock "github.com/stretchr/testify/mock"
	"github.com/stretchr/testify/require"
	"github.com/stretchr/testify/suite"

	"github.com/onflow/flow-go/crypto"
	"github.com/onflow/flow-go/crypto/hash"
	"github.com/koko1123/flow-go-1/engine/testutil/mocklocal"
	"github.com/koko1123/flow-go-1/engine/verification/utils"
	"github.com/koko1123/flow-go-1/engine/verification/verifier"
	chmodel "github.com/koko1123/flow-go-1/model/chunks"
	"github.com/koko1123/flow-go-1/model/flow"
	"github.com/koko1123/flow-go-1/model/verification"
	realModule "github.com/koko1123/flow-go-1/module"
	mockmodule "github.com/koko1123/flow-go-1/module/mock"
	"github.com/koko1123/flow-go-1/module/trace"
	"github.com/koko1123/flow-go-1/network/channels"
	"github.com/koko1123/flow-go-1/network/mocknetwork"
	protocol "github.com/koko1123/flow-go-1/state/protocol/mock"
	mockstorage "github.com/koko1123/flow-go-1/storage/mock"
	"github.com/koko1123/flow-go-1/utils/unittest"
)

type VerifierEngineTestSuite struct {
	suite.Suite
	net       *mocknetwork.Network
	tracer    realModule.Tracer
	state     *protocol.State
	ss        *protocol.Snapshot
	me        *mocklocal.MockLocal
	sk        crypto.PrivateKey
	hasher    hash.Hasher
	chain     flow.Chain
	pushCon   *mocknetwork.Conduit // mocks con for submitting result approvals
	pullCon   *mocknetwork.Conduit
	metrics   *mockmodule.VerificationMetrics // mocks performance monitoring metrics
	approvals *mockstorage.ResultApprovals
}

func TestVerifierEngine(t *testing.T) {
	suite.Run(t, new(VerifierEngineTestSuite))
}

func (suite *VerifierEngineTestSuite) SetupTest() {

	suite.state = &protocol.State{}
	suite.net = &mocknetwork.Network{}
	suite.tracer = trace.NewNoopTracer()
	suite.ss = &protocol.Snapshot{}
	suite.pushCon = &mocknetwork.Conduit{}
	suite.pullCon = &mocknetwork.Conduit{}
	suite.metrics = &mockmodule.VerificationMetrics{}
	suite.chain = flow.Testnet.Chain()
	suite.approvals = &mockstorage.ResultApprovals{}

	suite.approvals.On("Store", mock.Anything).Return(nil)
	suite.approvals.On("Index", mock.Anything, mock.Anything, mock.Anything).Return(nil)

	suite.net.On("Register", channels.PushApprovals, testifymock.Anything).
		Return(suite.pushCon, nil).
		Once()

	suite.net.On("Register", channels.ProvideApprovalsByChunk, testifymock.Anything).
		Return(suite.pullCon, nil).
		Once()

	suite.state.On("Final").Return(suite.ss)

	// Mocks the signature oracle of the engine
	//
	// generates signing and verification keys
	seed := make([]byte, crypto.KeyGenSeedMinLenBLSBLS12381)
	n, err := rand.Read(seed)
	require.Equal(suite.T(), n, crypto.KeyGenSeedMinLenBLSBLS12381)
	require.NoError(suite.T(), err)

	// creates private key of verification node
	sk, err := crypto.GeneratePrivateKey(crypto.BLSBLS12381, seed)
	require.NoError(suite.T(), err)
	suite.sk = sk

	// tag of hasher should be the same as the tag of engine's hasher
	suite.hasher = utils.NewResultApprovalHasher()

	// defines the identity of verification node and attaches its key.
	verIdentity := unittest.IdentityFixture(unittest.WithRole(flow.RoleVerification))
	verIdentity.StakingPubKey = sk.PublicKey()
	suite.me = mocklocal.NewMockLocal(sk, verIdentity.NodeID, suite.T())
}

func (suite *VerifierEngineTestSuite) TestNewEngine() *verifier.Engine {
	e, err := verifier.New(
		zerolog.Logger{},
		suite.metrics,
		suite.tracer,
		suite.net,
		suite.state,
		suite.me,
		ChunkVerifierMock{},
		suite.approvals)
	require.Nil(suite.T(), err)

	suite.net.AssertExpectations(suite.T())
	return e

}

func (suite *VerifierEngineTestSuite) TestIncorrectResult() {
	// TODO when ERs are verified
}

// TestVerifyHappyPath tests the verification path for a single verifiable chunk, which is
// assigned to the verifier node, and is passed by the ingest engine
// The tests evaluates that a result approval is emitted to all consensus nodes
// about the input execution receipt
func (suite *VerifierEngineTestSuite) TestVerifyHappyPath() {

	eng := suite.TestNewEngine()
	myID := unittest.IdentifierFixture()
	consensusNodes := unittest.IdentityListFixture(1, unittest.WithRole(flow.RoleConsensus))
	// creates a verifiable chunk
	vChunk := unittest.VerifiableChunkDataFixture(uint64(0))

	// mocking node ID using the LocalMock
	suite.me.MockNodeID(myID)
	suite.ss.On("Identities", testifymock.Anything).Return(consensusNodes, nil)

	// mocks metrics
	// reception of verifiable chunk
	suite.metrics.On("OnVerifiableChunkReceivedAtVerifierEngine").Return()
	// emission of result approval
	suite.metrics.On("OnResultApprovalDispatchedInNetworkByVerifier").Return()

	suite.pushCon.
		On("Publish", testifymock.Anything, testifymock.Anything).
		Return(nil).
		Run(func(args testifymock.Arguments) {
			// check that the approval matches the input execution result
			ra, ok := args[0].(*flow.ResultApproval)
			suite.Assert().True(ok)
			suite.Assert().Equal(vChunk.Result.ID(), ra.Body.ExecutionResultID)

			// verifies the signatures
			atstID := ra.Body.Attestation.ID()
			suite.Assert().True(suite.sk.PublicKey().Verify(ra.Body.AttestationSignature, atstID[:], suite.hasher))
			bodyID := ra.Body.ID()
			suite.Assert().True(suite.sk.PublicKey().Verify(ra.VerifierSignature, bodyID[:], suite.hasher))

			// spock should be non-nil
			suite.Assert().NotNil(ra.Body.Spock)
		}).
		Once()

	err := eng.ProcessLocal(vChunk)
	suite.Assert().NoError(err)
	suite.ss.AssertExpectations(suite.T())
	suite.pushCon.AssertExpectations(suite.T())

}

func (suite *VerifierEngineTestSuite) TestVerifyUnhappyPaths() {
	eng := suite.TestNewEngine()
	myID := unittest.IdentifierFixture()
	consensusNodes := unittest.IdentityListFixture(1, unittest.WithRole(flow.RoleConsensus))

	// mocking node ID using the LocalMock
	suite.me.MockNodeID(myID)
	suite.ss.On("Identities", testifymock.Anything).Return(consensusNodes, nil)

	// mocks metrics
	// reception of verifiable chunk
	suite.metrics.On("OnVerifiableChunkReceivedAtVerifierEngine").Return()

	// we shouldn't receive any result approval
	suite.pushCon.
		On("Publish", testifymock.Anything, testifymock.Anything).
		Return(nil).
		Run(func(args testifymock.Arguments) {
			// TODO change this to check challeneges
			_, ok := args[0].(*flow.ResultApproval)
			// TODO change this to false when missing register is rolled back
			suite.Assert().True(ok)
		})

	// emission of result approval
	suite.metrics.On("OnResultApprovalDispatchedInNetworkByVerifier").Return()

	var tests = []struct {
		vc          *verification.VerifiableChunkData
		expectedErr error
	}{
		{unittest.VerifiableChunkDataFixture(uint64(1)), nil},
		{unittest.VerifiableChunkDataFixture(uint64(2)), nil},
		{unittest.VerifiableChunkDataFixture(uint64(3)), nil},
	}
	for _, test := range tests {
		err := eng.ProcessLocal(test.vc)
		suite.Assert().NoError(err)
	}
}

type ChunkVerifierMock struct {
}

func (v ChunkVerifierMock) Verify(vc *verification.VerifiableChunkData) ([]byte, chmodel.ChunkFault, error) {
	if vc.IsSystemChunk {
		return nil, nil, nil
	}

	switch vc.Chunk.Index {
	case 0:
		return []byte{}, nil, nil
	// return error
	case 1:
		return nil, chmodel.NewCFMissingRegisterTouch(
			[]string{"test missing register touch"},
			vc.Chunk.Index,
			vc.Result.ID(),
			unittest.TransactionFixture().ID()), nil

	case 2:
		return nil, chmodel.NewCFInvalidVerifiableChunk(
			"test",
			errors.New("test invalid verifiable chunk"),
			vc.Chunk.Index,
			vc.Result.ID()), nil

	case 3:
		return nil, chmodel.NewCFNonMatchingFinalState(
			unittest.StateCommitmentFixture(),
			unittest.StateCommitmentFixture(),
			vc.Chunk.Index,
			vc.Result.ID()), nil

	// TODO add cases for challenges
	// return successful by default
	default:
		return nil, nil, nil
	}

}