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

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