package grpc_test import ( "context" "net" "testing" "time" grpc "google.golang.org/grpc" "google.golang.org/grpc/test/bufconn" "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 = "chain-id" func dialer(pv types.PrivValidator, logger log.Logger) (*grpc.Server, func(context.Context, string) (net.Conn, error)) { listener := bufconn.Listen(1024 * 1024) server := grpc.NewServer() s := tmgrpc.NewSignerServer(chainID, pv, logger) privvalproto.RegisterPrivValidatorAPIServer(server, s) go func() { if err := server.Serve(listener); err != nil { panic(err) } }() return server, func(context.Context, string) (net.Conn, error) { return listener.Dial() } } func TestSignerClient_GetPubKey(t *testing.T) { ctx := context.Background() mockPV := types.NewMockPV() logger := log.TestingLogger() srv, dialer := dialer(mockPV, logger) defer srv.Stop() conn, err := grpc.DialContext(ctx, "", grpc.WithInsecure(), grpc.WithContextDialer(dialer)) if err != nil { panic(err) } defer conn.Close() client, err := tmgrpc.NewSignerClient(conn, chainID, logger) require.NoError(t, err) pk, err := client.GetPubKey() require.NoError(t, err) assert.Equal(t, mockPV.PrivKey.PubKey(), pk) } func TestSignerClient_SignVote(t *testing.T) { ctx := context.Background() mockPV := types.NewMockPV() logger := log.TestingLogger() srv, dialer := dialer(mockPV, logger) defer srv.Stop() conn, err := grpc.DialContext(ctx, "", grpc.WithInsecure(), grpc.WithContextDialer(dialer)) if err != nil { panic(err) } defer conn.Close() client, err := tmgrpc.NewSignerClient(conn, chainID, logger) require.NoError(t, err) ts := time.Now() hash := tmrand.Bytes(tmhash.Size) valAddr := tmrand.Bytes(crypto.AddressSize) 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, } 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, } pbHave := have.ToProto() err = client.SignVote(chainID, pbHave) require.NoError(t, err) pbWant := want.ToProto() require.NoError(t, mockPV.SignVote(chainID, pbWant)) assert.Equal(t, pbWant.Signature, pbHave.Signature) } func TestSignerClient_SignProposal(t *testing.T) { ctx := context.Background() mockPV := types.NewMockPV() logger := log.TestingLogger() srv, dialer := dialer(mockPV, logger) defer srv.Stop() conn, err := grpc.DialContext(ctx, "", grpc.WithInsecure(), grpc.WithContextDialer(dialer)) if err != nil { panic(err) } defer conn.Close() client, err := tmgrpc.NewSignerClient(conn, chainID, logger) require.NoError(t, err) ts := time.Now() hash := tmrand.Bytes(tmhash.Size) 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, } pbHave := have.ToProto() err = client.SignProposal(chainID, pbHave) require.NoError(t, err) pbWant := want.ToProto() require.NoError(t, mockPV.SignProposal(chainID, pbWant)) assert.Equal(t, pbWant.Signature, pbHave.Signature) }