github.com/devseccon/trivy@v0.47.1-0.20231123133102-bd902a0bd996/pkg/db/mock_operation.go (about)

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package db
     4  
     5  import (
     6  	context "context"
     7  
     8  	mock "github.com/stretchr/testify/mock"
     9  
    10  	"github.com/devseccon/trivy/pkg/fanal/types"
    11  )
    12  
    13  // MockOperation is an autogenerated mock type for the Operation type
    14  type MockOperation struct {
    15  	mock.Mock
    16  }
    17  
    18  type OperationDownloadArgs struct {
    19  	Ctx         context.Context
    20  	CtxAnything bool
    21  	Dst         string
    22  	DstAnything bool
    23  }
    24  
    25  type OperationDownloadReturns struct {
    26  	Err error
    27  }
    28  
    29  type OperationDownloadExpectation struct {
    30  	Args    OperationDownloadArgs
    31  	Returns OperationDownloadReturns
    32  }
    33  
    34  func (_m *MockOperation) ApplyDownloadExpectation(e OperationDownloadExpectation) {
    35  	var args []interface{}
    36  	if e.Args.CtxAnything {
    37  		args = append(args, mock.Anything)
    38  	} else {
    39  		args = append(args, e.Args.Ctx)
    40  	}
    41  	if e.Args.DstAnything {
    42  		args = append(args, mock.Anything)
    43  	} else {
    44  		args = append(args, e.Args.Dst)
    45  	}
    46  	_m.On("Download", args...).Return(e.Returns.Err)
    47  }
    48  
    49  func (_m *MockOperation) ApplyDownloadExpectations(expectations []OperationDownloadExpectation) {
    50  	for _, e := range expectations {
    51  		_m.ApplyDownloadExpectation(e)
    52  	}
    53  }
    54  
    55  // Download provides a mock function with given fields: ctx, dst
    56  func (_m *MockOperation) Download(ctx context.Context, dst string, opt types.RegistryOptions) error {
    57  	ret := _m.Called(ctx, dst, opt)
    58  
    59  	var r0 error
    60  	if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
    61  		r0 = rf(ctx, dst)
    62  	} else {
    63  		r0 = ret.Error(0)
    64  	}
    65  
    66  	return r0
    67  }
    68  
    69  type OperationNeedsUpdateArgs struct {
    70  	CliVersion         string
    71  	CliVersionAnything bool
    72  	Skip               bool
    73  	SkipAnything       bool
    74  }
    75  
    76  type OperationNeedsUpdateReturns struct {
    77  	Need bool
    78  	Err  error
    79  }
    80  
    81  type OperationNeedsUpdateExpectation struct {
    82  	Args    OperationNeedsUpdateArgs
    83  	Returns OperationNeedsUpdateReturns
    84  }
    85  
    86  func (_m *MockOperation) ApplyNeedsUpdateExpectation(e OperationNeedsUpdateExpectation) {
    87  	var args []interface{}
    88  	if e.Args.CliVersionAnything {
    89  		args = append(args, mock.Anything)
    90  	} else {
    91  		args = append(args, e.Args.CliVersion)
    92  	}
    93  	if e.Args.SkipAnything {
    94  		args = append(args, mock.Anything)
    95  	} else {
    96  		args = append(args, e.Args.Skip)
    97  	}
    98  	_m.On("NeedsUpdate", args...).Return(e.Returns.Need, e.Returns.Err)
    99  }
   100  
   101  func (_m *MockOperation) ApplyNeedsUpdateExpectations(expectations []OperationNeedsUpdateExpectation) {
   102  	for _, e := range expectations {
   103  		_m.ApplyNeedsUpdateExpectation(e)
   104  	}
   105  }
   106  
   107  // NeedsUpdate provides a mock function with given fields: cliVersion, skip
   108  func (_m *MockOperation) NeedsUpdate(cliVersion string, skip bool) (bool, error) {
   109  	ret := _m.Called(cliVersion, skip)
   110  
   111  	var r0 bool
   112  	if rf, ok := ret.Get(0).(func(string, bool) bool); ok {
   113  		r0 = rf(cliVersion, skip)
   114  	} else {
   115  		r0 = ret.Get(0).(bool)
   116  	}
   117  
   118  	var r1 error
   119  	if rf, ok := ret.Get(1).(func(string, bool) error); ok {
   120  		r1 = rf(cliVersion, skip)
   121  	} else {
   122  		r1 = ret.Error(1)
   123  	}
   124  
   125  	return r0, r1
   126  }