package grpc_test import ( "context" "testing" "time" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "github.com/tendermint/tendermint/crypto" "github.com/tendermint/tendermint/crypto/tmhash" "github.com/tendermint/tendermint/libs/log" tmrand "github.com/tendermint/tendermint/libs/rand" tmgrpc "github.com/tendermint/tendermint/privval/grpc" privvalproto "github.com/tendermint/tendermint/proto/tendermint/privval" tmproto "github.com/tendermint/tendermint/proto/tendermint/types" "github.com/tendermint/tendermint/types" ) const ChainID = "123" func TestGetPubKey(t *testing.T) { testCases := []struct { name string pv types.PrivValidator err bool }{ {name: "valid", pv: types.NewMockPV(), err: false}, {name: "error on pubkey", pv: types.NewErroringMockPV(), err: true}, } for _, tc := range testCases { tc := tc t.Run(tc.name, func(t *testing.T) { s := tmgrpc.NewSignerServer(ChainID, tc.pv, log.TestingLogger()) req := &privvalproto.PubKeyRequest{ChainId: ChainID} resp, err := s.GetPubKey(context.Background(), req) if tc.err { require.Error(t, err) } else { pk, err := tc.pv.GetPubKey(context.Background()) require.NoError(t, err) assert.Equal(t, resp.PubKey.GetEd25519(), pk.Bytes()) } }) } } func TestSignVote(t *testing.T) { ts := time.Now() hash := tmrand.Bytes(tmhash.Size) valAddr := tmrand.Bytes(crypto.AddressSize) testCases := []struct { name string pv types.PrivValidator have, want *types.Vote err bool }{ {name: "valid", pv: types.NewMockPV(), have: &types.Vote{ Type: tmproto.PrecommitType, Height: 1, Round: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, ValidatorAddress: valAddr, ValidatorIndex: 1, }, want: &types.Vote{ Type: tmproto.PrecommitType, Height: 1, Round: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, ValidatorAddress: valAddr, ValidatorIndex: 1, }, err: false}, {name: "invalid vote", pv: types.NewErroringMockPV(), have: &types.Vote{ Type: tmproto.PrecommitType, Height: 1, Round: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, ValidatorAddress: valAddr, ValidatorIndex: 1, Signature: []byte("signed"), }, want: &types.Vote{ Type: tmproto.PrecommitType, Height: 1, Round: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, ValidatorAddress: valAddr, ValidatorIndex: 1, Signature: []byte("signed"), }, err: true}, } for _, tc := range testCases { tc := tc t.Run(tc.name, func(t *testing.T) { s := tmgrpc.NewSignerServer(ChainID, tc.pv, log.TestingLogger()) req := &privvalproto.SignVoteRequest{ChainId: ChainID, Vote: tc.have.ToProto()} resp, err := s.SignVote(context.Background(), req) if tc.err { require.Error(t, err) } else { pbVote := tc.want.ToProto() require.NoError(t, tc.pv.SignVote(context.Background(), ChainID, pbVote)) assert.Equal(t, pbVote.Signature, resp.Vote.Signature) } }) } } func TestSignProposal(t *testing.T) { ts := time.Now() hash := tmrand.Bytes(tmhash.Size) testCases := []struct { name string pv types.PrivValidator have, want *types.Proposal err bool }{ {name: "valid", pv: types.NewMockPV(), have: &types.Proposal{ Type: tmproto.ProposalType, Height: 1, Round: 2, POLRound: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, }, want: &types.Proposal{ Type: tmproto.ProposalType, Height: 1, Round: 2, POLRound: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, }, err: false}, {name: "invalid proposal", pv: types.NewErroringMockPV(), have: &types.Proposal{ Type: tmproto.ProposalType, Height: 1, Round: 2, POLRound: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, Signature: []byte("signed"), }, want: &types.Proposal{ Type: tmproto.ProposalType, Height: 1, Round: 2, POLRound: 2, BlockID: types.BlockID{Hash: hash, PartSetHeader: types.PartSetHeader{Hash: hash, Total: 2}}, Timestamp: ts, Signature: []byte("signed"), }, err: true}, } for _, tc := range testCases { tc := tc t.Run(tc.name, func(t *testing.T) { s := tmgrpc.NewSignerServer(ChainID, tc.pv, log.TestingLogger()) req := &privvalproto.SignProposalRequest{ChainId: ChainID, Proposal: tc.have.ToProto()} resp, err := s.SignProposal(context.Background(), req) if tc.err { require.Error(t, err) } else { pbProposal := tc.want.ToProto() require.NoError(t, tc.pv.SignProposal(context.Background(), ChainID, pbProposal)) assert.Equal(t, pbProposal.Signature, resp.Proposal.Signature) } }) } }