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

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