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

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package vulnsrc
     4  
     5  import (
     6  	types "github.com/khulnasoft-lab/tunnel-db/pkg/types"
     7  	mock "github.com/stretchr/testify/mock"
     8  )
     9  
    10  // MockVulnSrc is an autogenerated mock type for the VulnSrc type
    11  type MockVulnSrc struct {
    12  	mock.Mock
    13  }
    14  
    15  type VulnSrcNameReturns struct {
    16  	_a0 types.SourceID
    17  }
    18  
    19  type VulnSrcNameExpectation struct {
    20  	Returns VulnSrcNameReturns
    21  }
    22  
    23  func (_m *MockVulnSrc) ApplyNameExpectation(e VulnSrcNameExpectation) {
    24  	var args []interface{}
    25  	_m.On("Name", args...).Return(e.Returns._a0)
    26  }
    27  
    28  func (_m *MockVulnSrc) ApplyNameExpectations(expectations []VulnSrcNameExpectation) {
    29  	for _, e := range expectations {
    30  		_m.ApplyNameExpectation(e)
    31  	}
    32  }
    33  
    34  // Name provides a mock function with given fields:
    35  func (_m *MockVulnSrc) Name() types.SourceID {
    36  	ret := _m.Called()
    37  
    38  	var r0 types.SourceID
    39  	if rf, ok := ret.Get(0).(func() types.SourceID); ok {
    40  		r0 = rf()
    41  	} else {
    42  		r0 = ret.Get(0).(types.SourceID)
    43  	}
    44  
    45  	return r0
    46  }
    47  
    48  type VulnSrcUpdateArgs struct {
    49  	Dir         string
    50  	DirAnything bool
    51  }
    52  
    53  type VulnSrcUpdateReturns struct {
    54  	Err error
    55  }
    56  
    57  type VulnSrcUpdateExpectation struct {
    58  	Args    VulnSrcUpdateArgs
    59  	Returns VulnSrcUpdateReturns
    60  }
    61  
    62  func (_m *MockVulnSrc) ApplyUpdateExpectation(e VulnSrcUpdateExpectation) {
    63  	var args []interface{}
    64  	if e.Args.DirAnything {
    65  		args = append(args, mock.Anything)
    66  	} else {
    67  		args = append(args, e.Args.Dir)
    68  	}
    69  	_m.On("Update", args...).Return(e.Returns.Err)
    70  }
    71  
    72  func (_m *MockVulnSrc) ApplyUpdateExpectations(expectations []VulnSrcUpdateExpectation) {
    73  	for _, e := range expectations {
    74  		_m.ApplyUpdateExpectation(e)
    75  	}
    76  }
    77  
    78  // Update provides a mock function with given fields: dir
    79  func (_m *MockVulnSrc) Update(dir string) error {
    80  	ret := _m.Called(dir)
    81  
    82  	var r0 error
    83  	if rf, ok := ret.Get(0).(func(string) error); ok {
    84  		r0 = rf(dir)
    85  	} else {
    86  		r0 = ret.Error(0)
    87  	}
    88  
    89  	return r0
    90  }