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

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package vulnsrctest
     4  
     5  import (
     6  	types "github.com/khulnasoft-lab/tunnel-db/pkg/types"
     7  	mock "github.com/stretchr/testify/mock"
     8  )
     9  
    10  // MockGetter is an autogenerated mock type for the Getter type
    11  type MockGetter struct {
    12  	mock.Mock
    13  }
    14  
    15  type GetterGetArgs struct {
    16  	_a0         string
    17  	_a0Anything bool
    18  	_a1         string
    19  	_a1Anything bool
    20  }
    21  
    22  type GetterGetReturns struct {
    23  	_a0 []types.Advisory
    24  	_a1 error
    25  }
    26  
    27  type GetterGetExpectation struct {
    28  	Args    GetterGetArgs
    29  	Returns GetterGetReturns
    30  }
    31  
    32  func (_m *MockGetter) ApplyGetExpectation(e GetterGetExpectation) {
    33  	var args []interface{}
    34  	if e.Args._a0Anything {
    35  		args = append(args, mock.Anything)
    36  	} else {
    37  		args = append(args, e.Args._a0)
    38  	}
    39  	if e.Args._a1Anything {
    40  		args = append(args, mock.Anything)
    41  	} else {
    42  		args = append(args, e.Args._a1)
    43  	}
    44  	_m.On("Get", args...).Return(e.Returns._a0, e.Returns._a1)
    45  }
    46  
    47  func (_m *MockGetter) ApplyGetExpectations(expectations []GetterGetExpectation) {
    48  	for _, e := range expectations {
    49  		_m.ApplyGetExpectation(e)
    50  	}
    51  }
    52  
    53  // Get provides a mock function with given fields: _a0, _a1
    54  func (_m *MockGetter) Get(_a0 string, _a1 string) ([]types.Advisory, error) {
    55  	ret := _m.Called(_a0, _a1)
    56  
    57  	var r0 []types.Advisory
    58  	if rf, ok := ret.Get(0).(func(string, string) []types.Advisory); ok {
    59  		r0 = rf(_a0, _a1)
    60  	} else {
    61  		if ret.Get(0) != nil {
    62  			r0 = ret.Get(0).([]types.Advisory)
    63  		}
    64  	}
    65  
    66  	var r1 error
    67  	if rf, ok := ret.Get(1).(func(string, string) error); ok {
    68  		r1 = rf(_a0, _a1)
    69  	} else {
    70  		r1 = ret.Error(1)
    71  	}
    72  
    73  	return r0, r1
    74  }