2024-04-11 05:37:59 +02:00
// Code generated by mockery v2.42.2. DO NOT EDIT.
2020-03-11 15:25:40 +01:00
package driver
import (
context "context"
driver "github.com/goharbor/harbor/src/pkg/quota/driver"
mock "github.com/stretchr/testify/mock"
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
)
// Driver is an autogenerated mock type for the Driver type
type Driver struct {
mock . Mock
}
// CalculateUsage provides a mock function with given fields: ctx, key
func ( _m * Driver ) CalculateUsage ( ctx context . Context , key string ) ( types . ResourceList , error ) {
ret := _m . Called ( ctx , key )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for CalculateUsage" )
}
2020-03-11 15:25:40 +01:00
var r0 types . ResourceList
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) ( types . ResourceList , error ) ) ; ok {
return rf ( ctx , key )
}
2020-03-11 15:25:40 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) types . ResourceList ) ; ok {
r0 = rf ( ctx , key )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( types . ResourceList )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string ) error ) ; ok {
r1 = rf ( ctx , key )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Enabled provides a mock function with given fields: ctx, key
func ( _m * Driver ) Enabled ( ctx context . Context , key string ) ( bool , error ) {
ret := _m . Called ( ctx , key )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Enabled" )
}
2020-03-11 15:25:40 +01:00
var r0 bool
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) ( bool , error ) ) ; ok {
return rf ( ctx , key )
}
2020-03-11 15:25:40 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) bool ) ; ok {
r0 = rf ( ctx , key )
} else {
r0 = ret . Get ( 0 ) . ( bool )
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string ) error ) ; ok {
r1 = rf ( ctx , key )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// HardLimits provides a mock function with given fields: ctx
func ( _m * Driver ) HardLimits ( ctx context . Context ) types . ResourceList {
ret := _m . Called ( ctx )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for HardLimits" )
}
2020-03-11 15:25:40 +01:00
var r0 types . ResourceList
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context ) types . ResourceList ) ; ok {
r0 = rf ( ctx )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( types . ResourceList )
}
}
return r0
}
// Load provides a mock function with given fields: ctx, key
func ( _m * Driver ) Load ( ctx context . Context , key string ) ( driver . RefObject , error ) {
ret := _m . Called ( ctx , key )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Load" )
}
2020-03-11 15:25:40 +01:00
var r0 driver . RefObject
2023-03-11 01:16:11 +01:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) ( driver . RefObject , error ) ) ; ok {
return rf ( ctx , key )
}
2020-03-11 15:25:40 +01:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , string ) driver . RefObject ) ; ok {
r0 = rf ( ctx , key )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( driver . RefObject )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , string ) error ) ; ok {
r1 = rf ( ctx , key )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// Validate provides a mock function with given fields: hardLimits
func ( _m * Driver ) Validate ( hardLimits types . ResourceList ) error {
ret := _m . Called ( hardLimits )
2024-04-11 05:37:59 +02:00
if len ( ret ) == 0 {
panic ( "no return value specified for Validate" )
}
2020-03-11 15:25:40 +01:00
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( types . ResourceList ) error ) ; ok {
r0 = rf ( hardLimits )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
2022-06-07 10:04:33 +02:00
2023-11-05 07:37:25 +01:00
// NewDriver creates a new instance of Driver. 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 NewDriver ( t interface {
2022-06-07 10:04:33 +02:00
mock . TestingT
Cleanup ( func ( ) )
2023-11-05 07:37:25 +01:00
} ) * Driver {
2022-06-07 10:04:33 +02:00
mock := & Driver { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}