github.com/koko1123/flow-go-1@v0.29.6/engine/execution/ingestion/mock/ingest_rpc.go (about)

     1  // Code generated by mockery v2.13.1. DO NOT EDIT.
     2  
     3  package mock
     4  
     5  import (
     6  	context "context"
     7  
     8  	flow "github.com/koko1123/flow-go-1/model/flow"
     9  
    10  	mock "github.com/stretchr/testify/mock"
    11  )
    12  
    13  // IngestRPC is an autogenerated mock type for the IngestRPC type
    14  type IngestRPC struct {
    15  	mock.Mock
    16  }
    17  
    18  // ExecuteScriptAtBlockID provides a mock function with given fields: ctx, script, arguments, blockID
    19  func (_m *IngestRPC) ExecuteScriptAtBlockID(ctx context.Context, script []byte, arguments [][]byte, blockID flow.Identifier) ([]byte, error) {
    20  	ret := _m.Called(ctx, script, arguments, blockID)
    21  
    22  	var r0 []byte
    23  	if rf, ok := ret.Get(0).(func(context.Context, []byte, [][]byte, flow.Identifier) []byte); ok {
    24  		r0 = rf(ctx, script, arguments, blockID)
    25  	} else {
    26  		if ret.Get(0) != nil {
    27  			r0 = ret.Get(0).([]byte)
    28  		}
    29  	}
    30  
    31  	var r1 error
    32  	if rf, ok := ret.Get(1).(func(context.Context, []byte, [][]byte, flow.Identifier) error); ok {
    33  		r1 = rf(ctx, script, arguments, blockID)
    34  	} else {
    35  		r1 = ret.Error(1)
    36  	}
    37  
    38  	return r0, r1
    39  }
    40  
    41  // GetAccount provides a mock function with given fields: ctx, address, blockID
    42  func (_m *IngestRPC) GetAccount(ctx context.Context, address flow.Address, blockID flow.Identifier) (*flow.Account, error) {
    43  	ret := _m.Called(ctx, address, blockID)
    44  
    45  	var r0 *flow.Account
    46  	if rf, ok := ret.Get(0).(func(context.Context, flow.Address, flow.Identifier) *flow.Account); ok {
    47  		r0 = rf(ctx, address, blockID)
    48  	} else {
    49  		if ret.Get(0) != nil {
    50  			r0 = ret.Get(0).(*flow.Account)
    51  		}
    52  	}
    53  
    54  	var r1 error
    55  	if rf, ok := ret.Get(1).(func(context.Context, flow.Address, flow.Identifier) error); ok {
    56  		r1 = rf(ctx, address, blockID)
    57  	} else {
    58  		r1 = ret.Error(1)
    59  	}
    60  
    61  	return r0, r1
    62  }
    63  
    64  // GetRegisterAtBlockID provides a mock function with given fields: ctx, owner, key, blockID
    65  func (_m *IngestRPC) GetRegisterAtBlockID(ctx context.Context, owner []byte, key []byte, blockID flow.Identifier) ([]byte, error) {
    66  	ret := _m.Called(ctx, owner, key, blockID)
    67  
    68  	var r0 []byte
    69  	if rf, ok := ret.Get(0).(func(context.Context, []byte, []byte, flow.Identifier) []byte); ok {
    70  		r0 = rf(ctx, owner, key, blockID)
    71  	} else {
    72  		if ret.Get(0) != nil {
    73  			r0 = ret.Get(0).([]byte)
    74  		}
    75  	}
    76  
    77  	var r1 error
    78  	if rf, ok := ret.Get(1).(func(context.Context, []byte, []byte, flow.Identifier) error); ok {
    79  		r1 = rf(ctx, owner, key, blockID)
    80  	} else {
    81  		r1 = ret.Error(1)
    82  	}
    83  
    84  	return r0, r1
    85  }
    86  
    87  type mockConstructorTestingTNewIngestRPC interface {
    88  	mock.TestingT
    89  	Cleanup(func())
    90  }
    91  
    92  // NewIngestRPC creates a new instance of IngestRPC. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
    93  func NewIngestRPC(t mockConstructorTestingTNewIngestRPC) *IngestRPC {
    94  	mock := &IngestRPC{}
    95  	mock.Mock.Test(t)
    96  
    97  	t.Cleanup(func() { mock.AssertExpectations(t) })
    98  
    99  	return mock
   100  }