github.com/pachyderm/pachyderm@v1.13.4/src/server/pkg/transactionenv/txncontext/testing.go (about)

     1  package txncontext
     2  
     3  import (
     4  	"github.com/pachyderm/pachyderm/src/client/auth"
     5  	"github.com/pachyderm/pachyderm/src/client/pfs"
     6  	"github.com/pachyderm/pachyderm/src/client/pkg/errors"
     7  	"github.com/pachyderm/pachyderm/src/client/pps"
     8  	col "github.com/pachyderm/pachyderm/src/server/pkg/collection"
     9  )
    10  
    11  func unimplementedError(name string) error {
    12  	return errors.Errorf("%s is not implemented in this mock", name)
    13  }
    14  
    15  // MockPfsPropagater is a simple mock that can be used to satisfy the
    16  // PfsPropagater interface
    17  type MockPfsPropagater struct{}
    18  
    19  // NewMockPfsPropagater instantiates a MockPfsPropagater
    20  func NewMockPfsPropagater() *MockPfsPropagater {
    21  	return &MockPfsPropagater{}
    22  }
    23  
    24  // PropagateCommit always errors
    25  func (mpp *MockPfsPropagater) PropagateCommit(branch *pfs.Branch, isNewCommit bool) error {
    26  	return unimplementedError("PfsPropagater.PropagateCommit")
    27  }
    28  
    29  // Run always errors
    30  func (mpp *MockPfsPropagater) Run() error {
    31  	return unimplementedError("PfsPropagater.Run")
    32  }
    33  
    34  // MockAuthTransactionServer is a simple mock that can be used to satisfy the
    35  // AuthTransactionServer interface
    36  type MockAuthTransactionServer struct{}
    37  
    38  // NewMockAuthTransactionServer instantiates a MockAuthTransactionServer
    39  func NewMockAuthTransactionServer() *MockAuthTransactionServer {
    40  	return &MockAuthTransactionServer{}
    41  }
    42  
    43  // AuthorizeInTransaction always errors
    44  func (mats *MockAuthTransactionServer) AuthorizeInTransaction(*TransactionContext, *auth.AuthorizeRequest) (*auth.AuthorizeResponse, error) {
    45  	return nil, unimplementedError("AuthTransactionServer.AuthorizeInTransaction")
    46  }
    47  
    48  // GetScopeInTransaction always errors
    49  func (mats *MockAuthTransactionServer) GetScopeInTransaction(*TransactionContext, *auth.GetScopeRequest) (*auth.GetScopeResponse, error) {
    50  	return nil, unimplementedError("AuthTransactionServer.GetScopeInTransaction")
    51  }
    52  
    53  // SetScopeInTransaction always errors
    54  func (mats *MockAuthTransactionServer) SetScopeInTransaction(*TransactionContext, *auth.SetScopeRequest) (*auth.SetScopeResponse, error) {
    55  	return nil, unimplementedError("AuthTransactionServer.SetScopeInTransaction")
    56  }
    57  
    58  // GetACLInTransaction always errors
    59  func (mats *MockAuthTransactionServer) GetACLInTransaction(*TransactionContext, *auth.GetACLRequest) (*auth.GetACLResponse, error) {
    60  	return nil, unimplementedError("AuthTransactionServer.GetACLInTransaction")
    61  }
    62  
    63  // SetACLInTransaction always errors
    64  func (mats *MockAuthTransactionServer) SetACLInTransaction(*TransactionContext, *auth.SetACLRequest) (*auth.SetACLResponse, error) {
    65  	return nil, unimplementedError("AuthTransactionServer.SetACLInTransaction")
    66  }
    67  
    68  // MockPfsTransactionServer is a simple mock that can be used to satisfy the
    69  // PfsTransactionServer interface
    70  type MockPfsTransactionServer struct{}
    71  
    72  // NewMockPfsTransactionServer instantiates a MockPfsTransactionServer
    73  func NewMockPfsTransactionServer() *MockPfsTransactionServer {
    74  	return &MockPfsTransactionServer{}
    75  }
    76  
    77  // NewPropagater returns a MockPfsPropagater
    78  func (mpts *MockPfsTransactionServer) NewPropagater(col.STM) PfsPropagater {
    79  	return NewMockPfsPropagater()
    80  }
    81  
    82  // CreateRepoInTransaction always errors
    83  func (mpts *MockPfsTransactionServer) CreateRepoInTransaction(*TransactionContext, *pfs.CreateRepoRequest) error {
    84  	return unimplementedError("PfsTransactionServer.CreateRepoInTransaction")
    85  }
    86  
    87  // InspectRepoInTransaction always errors
    88  func (mpts *MockPfsTransactionServer) InspectRepoInTransaction(*TransactionContext, *pfs.InspectRepoRequest) (*pfs.RepoInfo, error) {
    89  	return nil, unimplementedError("PfsTransactionServer.InspectRepoInTransaction")
    90  }
    91  
    92  // DeleteRepoInTransaction always errors
    93  func (mpts *MockPfsTransactionServer) DeleteRepoInTransaction(*TransactionContext, *pfs.DeleteRepoRequest) error {
    94  	return unimplementedError("PfsTransactionServer.DeleteRepoInTransaction")
    95  }
    96  
    97  // StartCommitInTransaction always errors
    98  func (mpts *MockPfsTransactionServer) StartCommitInTransaction(*TransactionContext, *pfs.StartCommitRequest, *pfs.Commit) (*pfs.Commit, error) {
    99  	return nil, unimplementedError("PfsTransactionServer.StartCommitInTransaction")
   100  }
   101  
   102  // FinishCommitInTransaction always errors
   103  func (mpts *MockPfsTransactionServer) FinishCommitInTransaction(*TransactionContext, *pfs.FinishCommitRequest) error {
   104  	return unimplementedError("PfsTransactionServer.FinishCommitInTransaction")
   105  }
   106  
   107  // DeleteCommitInTransaction always errors
   108  func (mpts *MockPfsTransactionServer) DeleteCommitInTransaction(*TransactionContext, *pfs.DeleteCommitRequest) error {
   109  	return unimplementedError("PfsTransactionServer.DeleteCommitInTransaction")
   110  }
   111  
   112  // CreateBranchInTransaction always errors
   113  func (mpts *MockPfsTransactionServer) CreateBranchInTransaction(*TransactionContext, *pfs.CreateBranchRequest) error {
   114  	return unimplementedError("PfsTransactionServer.CreateBranchInTransaction")
   115  }
   116  
   117  // DeleteBranchInTransaction always errors
   118  func (mpts *MockPfsTransactionServer) DeleteBranchInTransaction(*TransactionContext, *pfs.DeleteBranchRequest) error {
   119  	return unimplementedError("PfsTransactionServer.DeleteBranchInTransaction")
   120  }
   121  
   122  // MockPpsTransactionServer is a simple mock that can be used to satisfy the
   123  // PpsTransactionServer interface
   124  type MockPpsTransactionServer struct{}
   125  
   126  // NewMockPpsTransactionServer instantiates a MockPpsTransactionServer
   127  func NewMockPpsTransactionServer() *MockPpsTransactionServer {
   128  	return &MockPpsTransactionServer{}
   129  }
   130  
   131  // UpdateJobStateInTransaction always errors
   132  func (mpts *MockPpsTransactionServer) UpdateJobStateInTransaction(*TransactionContext, *pps.UpdateJobStateRequest) error {
   133  	return unimplementedError("PpsTransactionServer.UpdateJobStateInTransaction")
   134  }
   135  
   136  // CreatePipelineInTransaction always errors
   137  func (mpts *MockPpsTransactionServer) CreatePipelineInTransaction(*TransactionContext, *pps.CreatePipelineRequest, **pfs.Commit) error {
   138  	return unimplementedError("PpsTransactionServer.UpdateJobStateInTransaction")
   139  }