2023-11-05 07:37:25 +01:00
// Code generated by mockery v2.35.4. DO NOT EDIT.
2020-03-11 15:25:40 +01:00
package quota
import (
context "context"
models "github.com/goharbor/harbor/src/pkg/quota/models"
mock "github.com/stretchr/testify/mock"
2020-04-12 18:14:12 +02:00
q "github.com/goharbor/harbor/src/lib/q"
2020-03-24 13:45:45 +01:00
quota "github.com/goharbor/harbor/src/controller/quota"
2020-03-16 03:20:17 +01:00
2020-07-14 16:28:53 +02:00
types "github.com/goharbor/harbor/src/pkg/quota/types"
2020-03-11 15:25:40 +01:00
)
// Controller is an autogenerated mock type for the Controller type
type Controller struct {
mock . Mock
}
2020-04-12 18:14:12 +02:00
// Count provides a mock function with given fields: ctx, query
func ( _m * Controller ) Count ( ctx context . Context , query * q . Query ) ( int64 , error ) {
ret := _m . Called ( ctx , query )
var r0 int64
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) ( int64 , error ) ) ; ok {
return rf ( ctx , query )
}
2020-04-12 18:14:12 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query ) int64 ) ; ok {
r0 = rf ( ctx , query )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * q . Query ) error ) ; ok {
r1 = rf ( ctx , query )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2020-03-11 15:25:40 +01:00
// Create provides a mock function with given fields: ctx, reference, referenceID, hardLimits, used
func ( _m * Controller ) Create ( ctx context . Context , reference string , referenceID string , hardLimits types . ResourceList , used ... types . ResourceList ) ( int64 , error ) {
_va := make ( [ ] interface { } , len ( used ) )
for _i := range used {
_va [ _i ] = used [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , reference , referenceID , hardLimits )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 int64
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string , types . ResourceList , ... types . ResourceList ) ( int64 , error ) ) ; ok {
return rf ( ctx , reference , referenceID , hardLimits , used ... )
}
2020-03-11 15:25:40 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string , types . ResourceList , ... types . ResourceList ) int64 ) ; ok {
r0 = rf ( ctx , reference , referenceID , hardLimits , used ... )
} else {
r0 = ret . Get ( 0 ) . ( int64 )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string , string , types . ResourceList , ... types . ResourceList ) error ) ; ok {
r1 = rf ( ctx , reference , referenceID , hardLimits , used ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Delete provides a mock function with given fields: ctx, id
func ( _m * Controller ) Delete ( ctx context . Context , id int64 ) error {
ret := _m . Called ( ctx , id )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) error ) ; ok {
r0 = rf ( ctx , id )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2020-07-22 05:18:05 +02:00
// Get provides a mock function with given fields: ctx, id, options
func ( _m * Controller ) Get ( ctx context . Context , id int64 , options ... quota . Option ) ( * models . Quota , error ) {
_va := make ( [ ] interface { } , len ( options ) )
for _i := range options {
_va [ _i ] = options [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , id )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
2020-03-11 15:25:40 +01:00
var r0 * models . Quota
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , ... quota . Option ) ( * models . Quota , error ) ) ; ok {
return rf ( ctx , id , options ... )
}
2020-07-22 05:18:05 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 , ... quota . Option ) * models . Quota ) ; ok {
r0 = rf ( ctx , id , options ... )
2020-03-11 15:25:40 +01:00
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * models . Quota )
}
}
2020-07-22 05:18:05 +02:00
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 , ... quota . Option ) error ) ; ok {
r1 = rf ( ctx , id , options ... )
2020-03-11 15:25:40 +01:00
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2020-07-22 05:18:05 +02:00
// GetByRef provides a mock function with given fields: ctx, reference, referenceID, options
func ( _m * Controller ) GetByRef ( ctx context . Context , reference string , referenceID string , options ... quota . Option ) ( * models . Quota , error ) {
_va := make ( [ ] interface { } , len ( options ) )
for _i := range options {
_va [ _i ] = options [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , reference , referenceID )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
2020-03-16 03:20:17 +01:00
var r0 * models . Quota
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string , ... quota . Option ) ( * models . Quota , error ) ) ; ok {
return rf ( ctx , reference , referenceID , options ... )
}
2020-07-22 05:18:05 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string , ... quota . Option ) * models . Quota ) ; ok {
r0 = rf ( ctx , reference , referenceID , options ... )
2020-03-16 03:20:17 +01:00
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * models . Quota )
}
}
2020-07-22 05:18:05 +02:00
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string , string , ... quota . Option ) error ) ; ok {
r1 = rf ( ctx , reference , referenceID , options ... )
2020-03-16 03:20:17 +01:00
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2020-03-11 15:25:40 +01:00
// IsEnabled provides a mock function with given fields: ctx, reference, referenceID
func ( _m * Controller ) IsEnabled ( ctx context . Context , reference string , referenceID string ) ( bool , error ) {
ret := _m . Called ( ctx , reference , referenceID )
var r0 bool
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string ) ( bool , error ) ) ; ok {
return rf ( ctx , reference , referenceID )
}
2020-03-11 15:25:40 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string ) bool ) ; ok {
r0 = rf ( ctx , reference , referenceID )
} else {
r0 = ret . Get ( 0 ) . ( bool )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string , string ) error ) ; ok {
r1 = rf ( ctx , reference , referenceID )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2020-07-22 05:18:05 +02:00
// List provides a mock function with given fields: ctx, query, options
func ( _m * Controller ) List ( ctx context . Context , query * q . Query , options ... quota . Option ) ( [ ] * models . Quota , error ) {
_va := make ( [ ] interface { } , len ( options ) )
for _i := range options {
_va [ _i ] = options [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , query )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
2020-04-12 18:14:12 +02:00
var r0 [ ] * models . Quota
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query , ... quota . Option ) ( [ ] * models . Quota , error ) ) ; ok {
return rf ( ctx , query , options ... )
}
2020-07-22 05:18:05 +02:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * q . Query , ... quota . Option ) [ ] * models . Quota ) ; ok {
r0 = rf ( ctx , query , options ... )
2020-04-12 18:14:12 +02:00
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( [ ] * models . Quota )
}
}
2020-07-22 05:18:05 +02:00
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * q . Query , ... quota . Option ) error ) ; ok {
r1 = rf ( ctx , query , options ... )
2020-04-12 18:14:12 +02:00
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2020-03-16 03:20:17 +01:00
// Refresh provides a mock function with given fields: ctx, reference, referenceID, options
func ( _m * Controller ) Refresh ( ctx context . Context , reference string , referenceID string , options ... quota . Option ) error {
_va := make ( [ ] interface { } , len ( options ) )
for _i := range options {
_va [ _i ] = options [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , reference , referenceID )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
2020-03-11 15:25:40 +01:00
var r0 error
2020-03-16 03:20:17 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string , ... quota . Option ) error ) ; ok {
r0 = rf ( ctx , reference , referenceID , options ... )
2020-03-11 15:25:40 +01:00
} else {
r0 = ret . Error ( 0 )
}
return r0
}
// Request provides a mock function with given fields: ctx, reference, referenceID, resources, f
func ( _m * Controller ) Request ( ctx context . Context , reference string , referenceID string , resources types . ResourceList , f func ( ) error ) error {
ret := _m . Called ( ctx , reference , referenceID , resources , f )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string , string , types . ResourceList , func ( ) error ) error ) ; ok {
r0 = rf ( ctx , reference , referenceID , resources , f )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2020-04-12 18:14:12 +02:00
// Update provides a mock function with given fields: ctx, _a1
func ( _m * Controller ) Update ( ctx context . Context , _a1 * models . Quota ) error {
ret := _m . Called ( ctx , _a1 )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * models . Quota ) error ) ; ok {
r0 = rf ( ctx , _a1 )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2022-06-07 10:04:33 +02:00
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 {
2022-06-07 10:04:33 +02:00
mock . TestingT
Cleanup ( func ( ) )
2023-11-05 07:37:25 +01:00
} ) * Controller {
2022-06-07 10:04:33 +02:00
mock := & Controller { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}