github.com/khulnasoft-lab/tunnel-db@v0.0.0-20231117205118-74e1113bd007/pkg/vulnsrctest/mock_updater.go (about)

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package vulnsrctest
     4  
     5  import mock "github.com/stretchr/testify/mock"
     6  
     7  // MockUpdater is an autogenerated mock type for the Updater type
     8  type MockUpdater struct {
     9  	mock.Mock
    10  }
    11  
    12  type UpdaterUpdateArgs struct {
    13  	Dir         string
    14  	DirAnything bool
    15  }
    16  
    17  type UpdaterUpdateReturns struct {
    18  	Err error
    19  }
    20  
    21  type UpdaterUpdateExpectation struct {
    22  	Args    UpdaterUpdateArgs
    23  	Returns UpdaterUpdateReturns
    24  }
    25  
    26  func (_m *MockUpdater) ApplyUpdateExpectation(e UpdaterUpdateExpectation) {
    27  	var args []interface{}
    28  	if e.Args.DirAnything {
    29  		args = append(args, mock.Anything)
    30  	} else {
    31  		args = append(args, e.Args.Dir)
    32  	}
    33  	_m.On("Update", args...).Return(e.Returns.Err)
    34  }
    35  
    36  func (_m *MockUpdater) ApplyUpdateExpectations(expectations []UpdaterUpdateExpectation) {
    37  	for _, e := range expectations {
    38  		_m.ApplyUpdateExpectation(e)
    39  	}
    40  }
    41  
    42  // Update provides a mock function with given fields: dir
    43  func (_m *MockUpdater) Update(dir string) error {
    44  	ret := _m.Called(dir)
    45  
    46  	var r0 error
    47  	if rf, ok := ret.Get(0).(func(string) error); ok {
    48  		r0 = rf(dir)
    49  	} else {
    50  		r0 = ret.Error(0)
    51  	}
    52  
    53  	return r0
    54  }