github.com/onflow/flow-go@v0.33.17/storage/mock/epoch_states.go (about)

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package mock
     4  
     5  import (
     6  	badger "github.com/dgraph-io/badger/v2"
     7  
     8  	flow "github.com/onflow/flow-go/model/flow"
     9  
    10  	mock "github.com/stretchr/testify/mock"
    11  )
    12  
    13  // EpochStates is an autogenerated mock type for the EpochStates type
    14  type EpochStates struct {
    15  	mock.Mock
    16  }
    17  
    18  // ByBlockID provides a mock function with given fields: _a0
    19  func (_m *EpochStates) ByBlockID(_a0 flow.Identifier) (*flow.EpochStatus, error) {
    20  	ret := _m.Called(_a0)
    21  
    22  	var r0 *flow.EpochStatus
    23  	if rf, ok := ret.Get(0).(func(flow.Identifier) *flow.EpochStatus); ok {
    24  		r0 = rf(_a0)
    25  	} else {
    26  		if ret.Get(0) != nil {
    27  			r0 = ret.Get(0).(*flow.EpochStatus)
    28  		}
    29  	}
    30  
    31  	var r1 error
    32  	if rf, ok := ret.Get(1).(func(flow.Identifier) error); ok {
    33  		r1 = rf(_a0)
    34  	} else {
    35  		r1 = ret.Error(1)
    36  	}
    37  
    38  	return r0, r1
    39  }
    40  
    41  // StoreTx provides a mock function with given fields: blockID, state
    42  func (_m *EpochStates) StoreTx(blockID flow.Identifier, state *flow.EpochStatus) func(*badger.Txn) error {
    43  	ret := _m.Called(blockID, state)
    44  
    45  	var r0 func(*badger.Txn) error
    46  	if rf, ok := ret.Get(0).(func(flow.Identifier, *flow.EpochStatus) func(*badger.Txn) error); ok {
    47  		r0 = rf(blockID, state)
    48  	} else {
    49  		if ret.Get(0) != nil {
    50  			r0 = ret.Get(0).(func(*badger.Txn) error)
    51  		}
    52  	}
    53  
    54  	return r0
    55  }