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

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