Mockserver can handle requests to the leases API

This should allow clients using the leases API to perform UTs.

Fixes #14426

Signed-off-by: Babis Kaidos <babis_k@outlook.com>
dependabot/go_modules/go.uber.org/atomic-1.10.0
Babis Kaidos 2022-09-06 00:30:56 +03:00 committed by Babis Kaidos
parent 3dc5348d94
commit 0eb642c4d9
1 changed files with 27 additions and 0 deletions

View File

@ -132,6 +132,7 @@ func (ms *MockServers) StartAt(idx int) (err error) {
svr := grpc.NewServer()
pb.RegisterKVServer(svr, &mockKVServer{})
pb.RegisterLeaseServer(svr, &mockLeaseServer{})
ms.Servers[idx].GrpcServer = svr
ms.wg.Add(1)
@ -185,3 +186,29 @@ func (m *mockKVServer) Txn(context.Context, *pb.TxnRequest) (*pb.TxnResponse, er
func (m *mockKVServer) Compact(context.Context, *pb.CompactionRequest) (*pb.CompactionResponse, error) {
return &pb.CompactionResponse{}, nil
}
func (m *mockKVServer) Lease(context.Context, *pb.LeaseGrantRequest) (*pb.LeaseGrantResponse, error) {
return &pb.LeaseGrantResponse{}, nil
}
type mockLeaseServer struct{}
func (s mockLeaseServer) LeaseGrant(context.Context, *pb.LeaseGrantRequest) (*pb.LeaseGrantResponse, error) {
return &pb.LeaseGrantResponse{}, nil
}
func (s *mockLeaseServer) LeaseRevoke(context.Context, *pb.LeaseRevokeRequest) (*pb.LeaseRevokeResponse, error) {
return &pb.LeaseRevokeResponse{}, nil
}
func (s *mockLeaseServer) LeaseKeepAlive(pb.Lease_LeaseKeepAliveServer) error {
return nil
}
func (s *mockLeaseServer) LeaseTimeToLive(context.Context, *pb.LeaseTimeToLiveRequest) (*pb.LeaseTimeToLiveResponse, error) {
return &pb.LeaseTimeToLiveResponse{}, nil
}
func (s *mockLeaseServer) LeaseLeases(context.Context, *pb.LeaseLeasesRequest) (*pb.LeaseLeasesResponse, error) {
return &pb.LeaseLeasesResponse{}, nil
}