Update Erroring service to implement SignerClient

wealdtech · Sep 13, 2024 · 588ac49 · 588ac49
1 parent ff055b7
commit 588ac49
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 5 deletions.
diff --git a/go.sum b/go.sum
@@ -63,8 +63,8 @@ github.com/wealdtech/eth2-signer-api v1.7.2 h1:9wmwWEstUwukyZmh0OhQfSHm9KrqFHF7o
 github.com/wealdtech/eth2-signer-api v1.7.2/go.mod h1:HOdnGSKi9z6OkV/UgpKpbsF3HcOAJkIjjjSWTXisnWI=
 github.com/wealdtech/go-eth2-types/v2 v2.8.2 h1:b5aXlNBLKgjAg/Fft9VvGlqAUCQMP5LzYhlHRrr4yPg=
 github.com/wealdtech/go-eth2-types/v2 v2.8.2/go.mod h1:IAz9Lz1NVTaHabQa+4zjk2QDKMv8LVYo0n46M9o/TXw=
-github.com/wealdtech/go-eth2-wallet-types/v2 v2.11.0 h1:yX9+FfUXvPDvZ8Q5bhF+64AWrQwh4a3/HpfTx99DnZc=
-github.com/wealdtech/go-eth2-wallet-types/v2 v2.11.0/go.mod h1:UVP9YFcnPiIzHqbmCMW3qrQ3TK5FOqr1fmKqNT9JGr8=
+github.com/wealdtech/go-eth2-wallet-types/v2 v2.12.0 h1:w0OrVImtQfjN7XCqALfgVtcmH8FChaUZjwNGKGzqp0E=
+github.com/wealdtech/go-eth2-wallet-types/v2 v2.12.0/go.mod h1:m8xsnPZLq1vt7bnMveTc4xxVJUv8mBL21iPU4kALYGE=
 go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.51.0 h1:A3SayB3rNyt+1S6qpI9mHPkeHTZbD7XILEqWnYZb2l0=
 go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.51.0/go.mod h1:27iA5uvhuRNmalO+iEUdVn5ZMj2qy10Mm+XRIpRmyuU=
 go.opentelemetry.io/otel v1.26.0 h1:LQwgL5s/1W7YiiRwxf03QGnWLb2HW4pLiAhaA5cZXBs=

diff --git a/mock/erroring.go b/mock/erroring.go
@@ -18,20 +18,35 @@ import (
 	"errors"
 
 	pb "github.com/wealdtech/eth2-signer-api/pb/v1"
+	"google.golang.org/grpc"
 )
 
 // ErroringService is a mock service that returns errors.
 type ErroringService struct{}
 
 // Sign returns an error.
-func (s *ErroringService) Sign(_ context.Context, _ *pb.SignRequest) (*pb.SignResponse, error) {
+func (s *ErroringService) Sign(_ context.Context, _ *pb.SignRequest, _ ...grpc.CallOption) (*pb.SignResponse, error) {
 	return &pb.SignResponse{
 		State: pb.ResponseState_UNKNOWN,
 	}, errors.New("mock error")
 }
 
+// Multisign returns an error.
+func (s *ErroringService) Multisign(_ context.Context, in *pb.MultisignRequest, _ ...grpc.CallOption) (*pb.MultisignResponse, error) {
+	responses := make([]*pb.SignResponse, len(in.GetRequests()))
+	for i := range responses {
+		responses[i] = &pb.SignResponse{
+			State: pb.ResponseState_UNKNOWN,
+		}
+	}
+
+	return &pb.MultisignResponse{
+		Responses: responses,
+	}, errors.New("mock error")
+}
+
 // SignBeaconAttestation returns an error.
-func (s *ErroringService) SignBeaconAttestation(_ context.Context, _ *pb.SignBeaconAttestationRequest) (*pb.SignResponse, error) {
+func (s *ErroringService) SignBeaconAttestation(_ context.Context, _ *pb.SignBeaconAttestationRequest, _ ...grpc.CallOption) (*pb.SignResponse, error) {
 	return &pb.SignResponse{
 		State: pb.ResponseState_UNKNOWN,
 	}, errors.New("mock error")
@@ -40,6 +55,7 @@ func (s *ErroringService) SignBeaconAttestation(_ context.Context, _ *pb.SignBea
 // SignBeaconAttestations returns an error.
 func (s *ErroringService) SignBeaconAttestations(_ context.Context,
 	in *pb.SignBeaconAttestationsRequest,
+	_ ...grpc.CallOption,
 ) (
 	*pb.MultisignResponse,
 	error,
@@ -57,7 +73,7 @@ func (s *ErroringService) SignBeaconAttestations(_ context.Context,
 }
 
 // SignBeaconProposal returns an error.
-func (s *ErroringService) SignBeaconProposal(_ context.Context, _ *pb.SignBeaconProposalRequest) (*pb.SignResponse, error) {
+func (s *ErroringService) SignBeaconProposal(_ context.Context, _ *pb.SignBeaconProposalRequest, _ ...grpc.CallOption) (*pb.SignResponse, error) {
 	return &pb.SignResponse{
 		State: pb.ResponseState_UNKNOWN,
 	}, errors.New("mock error")