github.com/koko1123/flow-go-1@v0.29.6/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  	encodable "github.com/koko1123/flow-go-1/model/encodable"
     7  	mock "github.com/stretchr/testify/mock"
     8  )
     9  
    10  // BeaconPrivateKeys is an autogenerated mock type for the BeaconPrivateKeys type
    11  type BeaconPrivateKeys struct {
    12  	mock.Mock
    13  }
    14  
    15  // InsertMyBeaconPrivateKey provides a mock function with given fields: epochCounter, key
    16  func (_m *BeaconPrivateKeys) InsertMyBeaconPrivateKey(epochCounter uint64, key *encodable.RandomBeaconPrivKey) error {
    17  	ret := _m.Called(epochCounter, key)
    18  
    19  	var r0 error
    20  	if rf, ok := ret.Get(0).(func(uint64, *encodable.RandomBeaconPrivKey) error); ok {
    21  		r0 = rf(epochCounter, key)
    22  	} else {
    23  		r0 = ret.Error(0)
    24  	}
    25  
    26  	return r0
    27  }
    28  
    29  // RetrieveMyBeaconPrivateKey provides a mock function with given fields: epochCounter
    30  func (_m *BeaconPrivateKeys) RetrieveMyBeaconPrivateKey(epochCounter uint64) (*encodable.RandomBeaconPrivKey, error) {
    31  	ret := _m.Called(epochCounter)
    32  
    33  	var r0 *encodable.RandomBeaconPrivKey
    34  	if rf, ok := ret.Get(0).(func(uint64) *encodable.RandomBeaconPrivKey); ok {
    35  		r0 = rf(epochCounter)
    36  	} else {
    37  		if ret.Get(0) != nil {
    38  			r0 = ret.Get(0).(*encodable.RandomBeaconPrivKey)
    39  		}
    40  	}
    41  
    42  	var r1 error
    43  	if rf, ok := ret.Get(1).(func(uint64) error); ok {
    44  		r1 = rf(epochCounter)
    45  	} else {
    46  		r1 = ret.Error(1)
    47  	}
    48  
    49  	return r0, r1
    50  }