github.com/koko1123/flow-go-1@v0.29.6/module/executiondatasync/tracker/mock/storage.go (about)

     1  // Code generated by mockery v2.13.1. DO NOT EDIT.
     2  
     3  package mocktracker
     4  
     5  import (
     6  	tracker "github.com/koko1123/flow-go-1/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  	if rf, ok := ret.Get(0).(func() uint64); ok {
    21  		r0 = rf()
    22  	} else {
    23  		r0 = ret.Get(0).(uint64)
    24  	}
    25  
    26  	var r1 error
    27  	if rf, ok := ret.Get(1).(func() error); ok {
    28  		r1 = rf()
    29  	} else {
    30  		r1 = ret.Error(1)
    31  	}
    32  
    33  	return r0, r1
    34  }
    35  
    36  // GetPrunedHeight provides a mock function with given fields:
    37  func (_m *Storage) GetPrunedHeight() (uint64, error) {
    38  	ret := _m.Called()
    39  
    40  	var r0 uint64
    41  	if rf, ok := ret.Get(0).(func() uint64); ok {
    42  		r0 = rf()
    43  	} else {
    44  		r0 = ret.Get(0).(uint64)
    45  	}
    46  
    47  	var r1 error
    48  	if rf, ok := ret.Get(1).(func() error); ok {
    49  		r1 = rf()
    50  	} else {
    51  		r1 = ret.Error(1)
    52  	}
    53  
    54  	return r0, r1
    55  }
    56  
    57  // PruneUpToHeight provides a mock function with given fields: height
    58  func (_m *Storage) PruneUpToHeight(height uint64) error {
    59  	ret := _m.Called(height)
    60  
    61  	var r0 error
    62  	if rf, ok := ret.Get(0).(func(uint64) error); ok {
    63  		r0 = rf(height)
    64  	} else {
    65  		r0 = ret.Error(0)
    66  	}
    67  
    68  	return r0
    69  }
    70  
    71  // SetFulfilledHeight provides a mock function with given fields: height
    72  func (_m *Storage) SetFulfilledHeight(height uint64) error {
    73  	ret := _m.Called(height)
    74  
    75  	var r0 error
    76  	if rf, ok := ret.Get(0).(func(uint64) error); ok {
    77  		r0 = rf(height)
    78  	} else {
    79  		r0 = ret.Error(0)
    80  	}
    81  
    82  	return r0
    83  }
    84  
    85  // Update provides a mock function with given fields: _a0
    86  func (_m *Storage) Update(_a0 tracker.UpdateFn) error {
    87  	ret := _m.Called(_a0)
    88  
    89  	var r0 error
    90  	if rf, ok := ret.Get(0).(func(tracker.UpdateFn) error); ok {
    91  		r0 = rf(_a0)
    92  	} else {
    93  		r0 = ret.Error(0)
    94  	}
    95  
    96  	return r0
    97  }
    98  
    99  type mockConstructorTestingTNewStorage interface {
   100  	mock.TestingT
   101  	Cleanup(func())
   102  }
   103  
   104  // 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.
   105  func NewStorage(t mockConstructorTestingTNewStorage) *Storage {
   106  	mock := &Storage{}
   107  	mock.Mock.Test(t)
   108  
   109  	t.Cleanup(func() { mock.AssertExpectations(t) })
   110  
   111  	return mock
   112  }