2023-11-05 07:37:25 +01:00
// Code generated by mockery v2.35.4. DO NOT EDIT.
2023-07-12 13:18:08 +02:00
package securityhub
import (
context "context"
model "github.com/goharbor/harbor/src/pkg/securityhub/model"
mock "github.com/stretchr/testify/mock"
2023-07-19 04:17:14 +02:00
q "github.com/goharbor/harbor/src/lib/q"
2023-07-12 13:18:08 +02:00
securityhub "github.com/goharbor/harbor/src/controller/securityhub"
)
// Controller is an autogenerated mock type for the Controller type
type Controller struct {
mock . Mock
}
2023-07-19 04:17:14 +02:00
// CountVuls provides a mock function with given fields: ctx, scannerUUID, projectID, tuneCount, query
func ( _m * Controller ) CountVuls ( ctx context . Context , scannerUUID string , projectID int64 , tuneCount bool , query * q . Query ) ( int64 , error ) {
ret := _m . Called ( ctx , scannerUUID , projectID , tuneCount , query )
var r0 int64
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , int64 , bool , * q . Query ) ( int64 , error ) ) ; ok {
return rf ( ctx , scannerUUID , projectID , tuneCount , query )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , int64 , bool , * q . Query ) int64 ) ; ok {
r0 = rf ( ctx , scannerUUID , projectID , tuneCount , query )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string , int64 , bool , * q . Query ) error ) ; ok {
r1 = rf ( ctx , scannerUUID , projectID , tuneCount , query )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// ListVuls provides a mock function with given fields: ctx, scannerUUID, projectID, withTag, query
func ( _m * Controller ) ListVuls ( ctx context . Context , scannerUUID string , projectID int64 , withTag bool , query * q . Query ) ( [ ] * model . VulnerabilityItem , error ) {
ret := _m . Called ( ctx , scannerUUID , projectID , withTag , query )
var r0 [ ] * model . VulnerabilityItem
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , int64 , bool , * q . Query ) ( [ ] * model . VulnerabilityItem , error ) ) ; ok {
return rf ( ctx , scannerUUID , projectID , withTag , query )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , int64 , bool , * q . Query ) [ ] * model . VulnerabilityItem ) ; ok {
r0 = rf ( ctx , scannerUUID , projectID , withTag , query )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * model . VulnerabilityItem )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string , int64 , bool , * q . Query ) error ) ; ok {
r1 = rf ( ctx , scannerUUID , projectID , withTag , query )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2023-07-12 13:18:08 +02:00
// SecuritySummary provides a mock function with given fields: ctx, projectID, options
func ( _m * Controller ) SecuritySummary ( ctx context . Context , projectID int64 , options ... securityhub . Option ) ( * model . Summary , error ) {
_va := make ( [ ] interface { } , len ( options ) )
for _i := range options {
_va [ _i ] = options [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , projectID )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * model . Summary
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , ... securityhub . Option ) ( * model . Summary , error ) ) ; ok {
return rf ( ctx , projectID , options ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , ... securityhub . Option ) * model . Summary ) ; ok {
r0 = rf ( ctx , projectID , options ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * model . Summary )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 , ... securityhub . Option ) error ) ; ok {
r1 = rf ( ctx , projectID , options ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2023-11-05 07:37:25 +01:00
// NewController creates a new instance of Controller. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
// The first argument is typically a *testing.T value.
func NewController ( t interface {
2023-07-12 13:18:08 +02:00
mock . TestingT
Cleanup ( func ( ) )
2023-11-05 07:37:25 +01:00
} ) * Controller {
2023-07-12 13:18:08 +02:00
mock := & Controller { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}