github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/module/mock/block_requester.go (about) 1 // Code generated by mockery v2.21.4. DO NOT EDIT. 2 3 package mock 4 5 import ( 6 flow "github.com/onflow/flow-go/model/flow" 7 mock "github.com/stretchr/testify/mock" 8 ) 9 10 // BlockRequester is an autogenerated mock type for the BlockRequester type 11 type BlockRequester struct { 12 mock.Mock 13 } 14 15 // Prune provides a mock function with given fields: final 16 func (_m *BlockRequester) Prune(final *flow.Header) { 17 _m.Called(final) 18 } 19 20 // RequestBlock provides a mock function with given fields: blockID, height 21 func (_m *BlockRequester) RequestBlock(blockID flow.Identifier, height uint64) { 22 _m.Called(blockID, height) 23 } 24 25 // RequestHeight provides a mock function with given fields: height 26 func (_m *BlockRequester) RequestHeight(height uint64) { 27 _m.Called(height) 28 } 29 30 type mockConstructorTestingTNewBlockRequester interface { 31 mock.TestingT 32 Cleanup(func()) 33 } 34 35 // NewBlockRequester creates a new instance of BlockRequester. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 36 func NewBlockRequester(t mockConstructorTestingTNewBlockRequester) *BlockRequester { 37 mock := &BlockRequester{} 38 mock.Mock.Test(t) 39 40 t.Cleanup(func() { mock.AssertExpectations(t) }) 41 42 return mock 43 }