github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/storage/mock/beacon_private_keys.go (about)

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