github.com/koko1123/flow-go-1@v0.29.6/module/mempool/mock/chunk_statuses.go (about)

     1  // Code generated by mockery v2.13.1. DO NOT EDIT.
     2  
     3  package mempool
     4  
     5  import (
     6  	flow "github.com/koko1123/flow-go-1/model/flow"
     7  
     8  	mock "github.com/stretchr/testify/mock"
     9  
    10  	verification "github.com/koko1123/flow-go-1/model/verification"
    11  )
    12  
    13  // ChunkStatuses is an autogenerated mock type for the ChunkStatuses type
    14  type ChunkStatuses struct {
    15  	mock.Mock
    16  }
    17  
    18  // Add provides a mock function with given fields: status
    19  func (_m *ChunkStatuses) Add(status *verification.ChunkStatus) bool {
    20  	ret := _m.Called(status)
    21  
    22  	var r0 bool
    23  	if rf, ok := ret.Get(0).(func(*verification.ChunkStatus) bool); ok {
    24  		r0 = rf(status)
    25  	} else {
    26  		r0 = ret.Get(0).(bool)
    27  	}
    28  
    29  	return r0
    30  }
    31  
    32  // All provides a mock function with given fields:
    33  func (_m *ChunkStatuses) All() []*verification.ChunkStatus {
    34  	ret := _m.Called()
    35  
    36  	var r0 []*verification.ChunkStatus
    37  	if rf, ok := ret.Get(0).(func() []*verification.ChunkStatus); ok {
    38  		r0 = rf()
    39  	} else {
    40  		if ret.Get(0) != nil {
    41  			r0 = ret.Get(0).([]*verification.ChunkStatus)
    42  		}
    43  	}
    44  
    45  	return r0
    46  }
    47  
    48  // Get provides a mock function with given fields: chunkIndex, resultID
    49  func (_m *ChunkStatuses) Get(chunkIndex uint64, resultID flow.Identifier) (*verification.ChunkStatus, bool) {
    50  	ret := _m.Called(chunkIndex, resultID)
    51  
    52  	var r0 *verification.ChunkStatus
    53  	if rf, ok := ret.Get(0).(func(uint64, flow.Identifier) *verification.ChunkStatus); ok {
    54  		r0 = rf(chunkIndex, resultID)
    55  	} else {
    56  		if ret.Get(0) != nil {
    57  			r0 = ret.Get(0).(*verification.ChunkStatus)
    58  		}
    59  	}
    60  
    61  	var r1 bool
    62  	if rf, ok := ret.Get(1).(func(uint64, flow.Identifier) bool); ok {
    63  		r1 = rf(chunkIndex, resultID)
    64  	} else {
    65  		r1 = ret.Get(1).(bool)
    66  	}
    67  
    68  	return r0, r1
    69  }
    70  
    71  // Remove provides a mock function with given fields: chunkIndex, resultID
    72  func (_m *ChunkStatuses) Remove(chunkIndex uint64, resultID flow.Identifier) bool {
    73  	ret := _m.Called(chunkIndex, resultID)
    74  
    75  	var r0 bool
    76  	if rf, ok := ret.Get(0).(func(uint64, flow.Identifier) bool); ok {
    77  		r0 = rf(chunkIndex, resultID)
    78  	} else {
    79  		r0 = ret.Get(0).(bool)
    80  	}
    81  
    82  	return r0
    83  }
    84  
    85  // Size provides a mock function with given fields:
    86  func (_m *ChunkStatuses) Size() uint {
    87  	ret := _m.Called()
    88  
    89  	var r0 uint
    90  	if rf, ok := ret.Get(0).(func() uint); ok {
    91  		r0 = rf()
    92  	} else {
    93  		r0 = ret.Get(0).(uint)
    94  	}
    95  
    96  	return r0
    97  }
    98  
    99  type mockConstructorTestingTNewChunkStatuses interface {
   100  	mock.TestingT
   101  	Cleanup(func())
   102  }
   103  
   104  // NewChunkStatuses creates a new instance of ChunkStatuses. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   105  func NewChunkStatuses(t mockConstructorTestingTNewChunkStatuses) *ChunkStatuses {
   106  	mock := &ChunkStatuses{}
   107  	mock.Mock.Test(t)
   108  
   109  	t.Cleanup(func() { mock.AssertExpectations(t) })
   110  
   111  	return mock
   112  }