sa.go

Documentation: github.com/letsencrypt/boulder/test/inmem/sa

     1  package sa
     2  
     3  import (
     4  	"context"
     5  	"io"
     6  
     7  	corepb "github.com/letsencrypt/boulder/core/proto"
     8  	"github.com/letsencrypt/boulder/sa"
     9  	sapb "github.com/letsencrypt/boulder/sa/proto"
    10  
    11  	"google.golang.org/grpc"
    12  	"google.golang.org/protobuf/types/known/emptypb"
    13  )
    14  
    15  // SA meets the `sapb.StorageAuthorityClient` interface and acts as a
    16  // wrapper for an inner `sa.SQLStorageAuthority` (which in turn meets
    17  // the `sapb.StorageAuthorityServer` interface). Only methods used by
    18  // unit tests need to be implemented.
    19  type SA struct {
    20  	sapb.StorageAuthorityClient
    21  	Impl *sa.SQLStorageAuthority
    22  }
    23  
    24  func (sa SA) NewRegistration(ctx context.Context, req *corepb.Registration, _ ...grpc.CallOption) (*corepb.Registration, error) {
    25  	return sa.Impl.NewRegistration(ctx, req)
    26  }
    27  
    28  func (sa SA) GetRegistration(ctx context.Context, req *sapb.RegistrationID, _ ...grpc.CallOption) (*corepb.Registration, error) {
    29  	return sa.Impl.GetRegistration(ctx, req)
    30  }
    31  
    32  func (sa SA) CountRegistrationsByIP(ctx context.Context, req *sapb.CountRegistrationsByIPRequest, _ ...grpc.CallOption) (*sapb.Count, error) {
    33  	return sa.Impl.CountRegistrationsByIP(ctx, req)
    34  }
    35  
    36  func (sa SA) CountRegistrationsByIPRange(ctx context.Context, req *sapb.CountRegistrationsByIPRequest, _ ...grpc.CallOption) (*sapb.Count, error) {
    37  	return sa.Impl.CountRegistrationsByIPRange(ctx, req)
    38  }
    39  
    40  func (sa SA) DeactivateRegistration(ctx context.Context, req *sapb.RegistrationID, _ ...grpc.CallOption) (*emptypb.Empty, error) {
    41  	return sa.Impl.DeactivateRegistration(ctx, req)
    42  }
    43  
    44  func (sa SA) GetAuthorization2(ctx context.Context, req *sapb.AuthorizationID2, _ ...grpc.CallOption) (*corepb.Authorization, error) {
    45  	return sa.Impl.GetAuthorization2(ctx, req)
    46  }
    47  
    48  func (sa SA) GetAuthorizations2(ctx context.Context, req *sapb.GetAuthorizationsRequest, _ ...grpc.CallOption) (*sapb.Authorizations, error) {
    49  	return sa.Impl.GetAuthorizations2(ctx, req)
    50  }
    51  
    52  func (sa SA) GetPendingAuthorization2(ctx context.Context, req *sapb.GetPendingAuthorizationRequest, _ ...grpc.CallOption) (*corepb.Authorization, error) {
    53  	return sa.Impl.GetPendingAuthorization2(ctx, req)
    54  }
    55  
    56  func (sa SA) GetValidAuthorizations2(ctx context.Context, req *sapb.GetValidAuthorizationsRequest, _ ...grpc.CallOption) (*sapb.Authorizations, error) {
    57  	return sa.Impl.GetValidAuthorizations2(ctx, req)
    58  }
    59  
    60  func (sa SA) GetValidOrderAuthorizations2(ctx context.Context, req *sapb.GetValidOrderAuthorizationsRequest, _ ...grpc.CallOption) (*sapb.Authorizations, error) {
    61  	return sa.Impl.GetValidOrderAuthorizations2(ctx, req)
    62  }
    63  
    64  func (sa SA) CountPendingAuthorizations2(ctx context.Context, req *sapb.RegistrationID, _ ...grpc.CallOption) (*sapb.Count, error) {
    65  	return sa.Impl.CountPendingAuthorizations2(ctx, req)
    66  }
    67  
    68  func (sa SA) DeactivateAuthorization2(ctx context.Context, req *sapb.AuthorizationID2, _ ...grpc.CallOption) (*emptypb.Empty, error) {
    69  	return sa.Impl.DeactivateAuthorization2(ctx, req)
    70  }
    71  
    72  func (sa SA) FinalizeAuthorization2(ctx context.Context, req *sapb.FinalizeAuthorizationRequest, _ ...grpc.CallOption) (*emptypb.Empty, error) {
    73  	return sa.Impl.FinalizeAuthorization2(ctx, req)
    74  }
    75  
    76  func (sa SA) NewOrderAndAuthzs(ctx context.Context, req *sapb.NewOrderAndAuthzsRequest, _ ...grpc.CallOption) (*corepb.Order, error) {
    77  	return sa.Impl.NewOrderAndAuthzs(ctx, req)
    78  }
    79  
    80  func (sa SA) GetOrder(ctx context.Context, req *sapb.OrderRequest, _ ...grpc.CallOption) (*corepb.Order, error) {
    81  	return sa.Impl.GetOrder(ctx, req)
    82  }
    83  
    84  func (sa SA) GetOrderForNames(ctx context.Context, req *sapb.GetOrderForNamesRequest, _ ...grpc.CallOption) (*corepb.Order, error) {
    85  	return sa.Impl.GetOrderForNames(ctx, req)
    86  }
    87  
    88  func (sa SA) CountOrders(ctx context.Context, req *sapb.CountOrdersRequest, _ ...grpc.CallOption) (*sapb.Count, error) {
    89  	return sa.Impl.CountOrders(ctx, req)
    90  }
    91  
    92  func (sa SA) SetOrderError(ctx context.Context, req *sapb.SetOrderErrorRequest, _ ...grpc.CallOption) (*emptypb.Empty, error) {
    93  	return sa.Impl.SetOrderError(ctx, req)
    94  }
    95  
    96  func (sa SA) SetOrderProcessing(ctx context.Context, req *sapb.OrderRequest, _ ...grpc.CallOption) (*emptypb.Empty, error) {
    97  	return sa.Impl.SetOrderProcessing(ctx, req)
    98  }
    99  
   100  func (sa SA) FinalizeOrder(ctx context.Context, req *sapb.FinalizeOrderRequest, _ ...grpc.CallOption) (*emptypb.Empty, error) {
   101  	return sa.Impl.FinalizeOrder(ctx, req)
   102  }
   103  
   104  func (sa SA) AddPrecertificate(ctx context.Context, req *sapb.AddCertificateRequest, _ ...grpc.CallOption) (*emptypb.Empty, error) {
   105  	return sa.Impl.AddPrecertificate(ctx, req)
   106  }
   107  
   108  func (sa SA) CountCertificatesByNames(ctx context.Context, req *sapb.CountCertificatesByNamesRequest, _ ...grpc.CallOption) (*sapb.CountByNames, error) {
   109  	return sa.Impl.CountCertificatesByNames(ctx, req)
   110  }
   111  
   112  func (sa SA) RevokeCertificate(ctx context.Context, req *sapb.RevokeCertificateRequest, _ ...grpc.CallOption) (*emptypb.Empty, error) {
   113  	return sa.Impl.RevokeCertificate(ctx, req)
   114  }
   115  
   116  func (sa SA) GetCertificateStatus(ctx context.Context, req *sapb.Serial, _ ...grpc.CallOption) (*corepb.CertificateStatus, error) {
   117  	return sa.Impl.GetCertificateStatus(ctx, req)
   118  }
   119  
   120  func (sa SA) AddBlockedKey(ctx context.Context, req *sapb.AddBlockedKeyRequest, _ ...grpc.CallOption) (*emptypb.Empty, error) {
   121  	return sa.Impl.AddBlockedKey(ctx, req)
   122  }
   123  
   124  func (sa SA) FQDNSetExists(ctx context.Context, req *sapb.FQDNSetExistsRequest, _ ...grpc.CallOption) (*sapb.Exists, error) {
   125  	return sa.Impl.FQDNSetExists(ctx, req)
   126  }
   127  
   128  type mockSerialsForIncidentStream_Result struct {
   129  	serial *sapb.IncidentSerial
   130  	err    error
   131  }
   132  
   133  type mockSerialsForIncidentStream_Client struct {
   134  	grpc.ClientStream
   135  	stream <-chan mockSerialsForIncidentStream_Result
   136  }
   137  
   138  func (c mockSerialsForIncidentStream_Client) Recv() (*sapb.IncidentSerial, error) {
   139  	sfiData := <-c.stream
   140  	return sfiData.serial, sfiData.err
   141  }
   142  
   143  type mockSerialsForIncidentStream_Server struct {
   144  	grpc.ServerStream
   145  	context context.Context
   146  	stream  chan<- mockSerialsForIncidentStream_Result
   147  }
   148  
   149  func (s mockSerialsForIncidentStream_Server) Send(serial *sapb.IncidentSerial) error {
   150  	s.stream <- mockSerialsForIncidentStream_Result{serial, nil}
   151  	return nil
   152  }
   153  
   154  func (s mockSerialsForIncidentStream_Server) Context() context.Context {
   155  	return s.context
   156  }
   157  
   158  func (sa SA) SerialsForIncident(ctx context.Context, req *sapb.SerialsForIncidentRequest, _ ...grpc.CallOption) (sapb.StorageAuthority_SerialsForIncidentClient, error) {
   159  	streamChan := make(chan mockSerialsForIncidentStream_Result)
   160  	client := mockSerialsForIncidentStream_Client{stream: streamChan}
   161  	server := mockSerialsForIncidentStream_Server{context: ctx, stream: streamChan}
   162  	go func() {
   163  		err := sa.Impl.SerialsForIncident(req, server)
   164  		if err != nil {
   165  			streamChan <- mockSerialsForIncidentStream_Result{nil, err}
   166  		}
   167  		streamChan <- mockSerialsForIncidentStream_Result{nil, io.EOF}
   168  		close(streamChan)
   169  	}()
   170  	return client, nil
   171  }
   172
View as plain text