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

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package scanner
     4  
     5  import (
     6  	"context"
     7  
     8  	mock "github.com/stretchr/testify/mock"
     9  
    10  	fanaltypes "github.com/devseccon/trivy/pkg/fanal/types"
    11  
    12  	types "github.com/devseccon/trivy/pkg/types"
    13  )
    14  
    15  // MockDriver is an autogenerated mock type for the Driver type
    16  type MockDriver struct {
    17  	mock.Mock
    18  }
    19  
    20  type DriverScanArgs struct {
    21  	Ctx              context.Context
    22  	CtxAnything      bool
    23  	Target           string
    24  	TargetAnything   bool
    25  	ImageID          string
    26  	ImageIDAnything  bool
    27  	LayerIDs         []string
    28  	LayerIDsAnything bool
    29  	Options          types.ScanOptions
    30  	OptionsAnything  bool
    31  }
    32  
    33  type DriverScanReturns struct {
    34  	Results types.Results
    35  	OsFound fanaltypes.OS
    36  	Err     error
    37  }
    38  
    39  type DriverScanExpectation struct {
    40  	Args    DriverScanArgs
    41  	Returns DriverScanReturns
    42  }
    43  
    44  func (_m *MockDriver) ApplyScanExpectation(e DriverScanExpectation) {
    45  	var args []interface{}
    46  	if e.Args.CtxAnything {
    47  		args = append(args, mock.Anything)
    48  	} else {
    49  		args = append(args, e.Args.Ctx)
    50  	}
    51  	if e.Args.TargetAnything {
    52  		args = append(args, mock.Anything)
    53  	} else {
    54  		args = append(args, e.Args.Target)
    55  	}
    56  	if e.Args.ImageIDAnything {
    57  		args = append(args, mock.Anything)
    58  	} else {
    59  		args = append(args, e.Args.ImageID)
    60  	}
    61  	if e.Args.LayerIDsAnything {
    62  		args = append(args, mock.Anything)
    63  	} else {
    64  		args = append(args, e.Args.LayerIDs)
    65  	}
    66  	if e.Args.OptionsAnything {
    67  		args = append(args, mock.Anything)
    68  	} else {
    69  		args = append(args, e.Args.Options)
    70  	}
    71  	_m.On("Scan", args...).Return(e.Returns.Results, e.Returns.OsFound, e.Returns.Err)
    72  }
    73  
    74  func (_m *MockDriver) ApplyScanExpectations(expectations []DriverScanExpectation) {
    75  	for _, e := range expectations {
    76  		_m.ApplyScanExpectation(e)
    77  	}
    78  }
    79  
    80  // Scan provides a mock function with given fields: ctx, target, imageID, layerIDs, options
    81  func (_m *MockDriver) Scan(ctx context.Context, target string, artifactKey string, blobKeys []string, options types.ScanOptions) (types.Results, fanaltypes.OS, error) {
    82  	ret := _m.Called(ctx, target, artifactKey, blobKeys, options)
    83  
    84  	var r0 types.Results
    85  	if rf, ok := ret.Get(0).(func(context.Context, string, string, []string, types.ScanOptions) types.Results); ok {
    86  		r0 = rf(ctx, target, artifactKey, blobKeys, options)
    87  	} else {
    88  		if ret.Get(0) != nil {
    89  			r0 = ret.Get(0).(types.Results)
    90  		}
    91  	}
    92  
    93  	var r1 fanaltypes.OS
    94  	if rf, ok := ret.Get(1).(func(context.Context, string, string, []string, types.ScanOptions) fanaltypes.OS); ok {
    95  		r1 = rf(ctx, target, artifactKey, blobKeys, options)
    96  	} else {
    97  		if ret.Get(1) != nil {
    98  			r1 = ret.Get(1).(fanaltypes.OS)
    99  		}
   100  	}
   101  
   102  	var r2 error
   103  	if rf, ok := ret.Get(2).(func(context.Context, string, string, []string, types.ScanOptions) error); ok {
   104  		r2 = rf(ctx, target, artifactKey, blobKeys, options)
   105  	} else {
   106  		r2 = ret.Error(2)
   107  	}
   108  
   109  	return r0, r1, r2
   110  }