Update mocks

Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
This commit is contained in:
Gabriel Adrian Samfira 2025-04-27 20:34:04 +00:00
parent 884be62a4d
commit 55b4e74066
6 changed files with 404 additions and 119 deletions

View file

@ -1,4 +1,4 @@
// Code generated by mockery v2.42.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -117,7 +117,7 @@ func (_m *GithubClient) DeleteEntityHook(ctx context.Context, id int64) (*github
return r0, r1
}
// GetEntity provides a mock function with given fields:
// GetEntity provides a mock function with no fields
func (_m *GithubClient) GetEntity() params.GithubEntity {
ret := _m.Called()
@ -243,7 +243,7 @@ func (_m *GithubClient) GetWorkflowJobByID(ctx context.Context, owner string, re
return r0, r1, r2
}
// GithubBaseURL provides a mock function with given fields:
// GithubBaseURL provides a mock function with no fields
func (_m *GithubClient) GithubBaseURL() *url.URL {
ret := _m.Called()
@ -411,33 +411,21 @@ func (_m *GithubClient) PingEntityHook(ctx context.Context, id int64) (*github.R
}
// RemoveEntityRunner provides a mock function with given fields: ctx, runnerID
func (_m *GithubClient) RemoveEntityRunner(ctx context.Context, runnerID int64) (*github.Response, error) {
func (_m *GithubClient) RemoveEntityRunner(ctx context.Context, runnerID int64) error {
ret := _m.Called(ctx, runnerID)
if len(ret) == 0 {
panic("no return value specified for RemoveEntityRunner")
}
var r0 *github.Response
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, int64) (*github.Response, error)); ok {
return rf(ctx, runnerID)
}
if rf, ok := ret.Get(0).(func(context.Context, int64) *github.Response); ok {
var r0 error
if rf, ok := ret.Get(0).(func(context.Context, int64) error); ok {
r0 = rf(ctx, runnerID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*github.Response)
}
r0 = ret.Error(0)
}
if rf, ok := ret.Get(1).(func(context.Context, int64) error); ok {
r1 = rf(ctx, runnerID)
} else {
r1 = ret.Error(1)
}
return r0, r1
return r0
}
// NewGithubClient creates a new instance of GithubClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.

View file

@ -1,4 +1,4 @@
// Code generated by mockery v2.42.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -117,7 +117,7 @@ func (_m *GithubEntityOperations) DeleteEntityHook(ctx context.Context, id int64
return r0, r1
}
// GetEntity provides a mock function with given fields:
// GetEntity provides a mock function with no fields
func (_m *GithubEntityOperations) GetEntity() params.GithubEntity {
ret := _m.Called()
@ -204,7 +204,7 @@ func (_m *GithubEntityOperations) GetEntityJITConfig(ctx context.Context, instan
return r0, r1, r2
}
// GithubBaseURL provides a mock function with given fields:
// GithubBaseURL provides a mock function with no fields
func (_m *GithubEntityOperations) GithubBaseURL() *url.URL {
ret := _m.Called()
@ -372,33 +372,21 @@ func (_m *GithubEntityOperations) PingEntityHook(ctx context.Context, id int64)
}
// RemoveEntityRunner provides a mock function with given fields: ctx, runnerID
func (_m *GithubEntityOperations) RemoveEntityRunner(ctx context.Context, runnerID int64) (*github.Response, error) {
func (_m *GithubEntityOperations) RemoveEntityRunner(ctx context.Context, runnerID int64) error {
ret := _m.Called(ctx, runnerID)
if len(ret) == 0 {
panic("no return value specified for RemoveEntityRunner")
}
var r0 *github.Response
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, int64) (*github.Response, error)); ok {
return rf(ctx, runnerID)
}
if rf, ok := ret.Get(0).(func(context.Context, int64) *github.Response); ok {
var r0 error
if rf, ok := ret.Get(0).(func(context.Context, int64) error); ok {
r0 = rf(ctx, runnerID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*github.Response)
}
r0 = ret.Error(0)
}
if rf, ok := ret.Get(1).(func(context.Context, int64) error); ok {
r1 = rf(ctx, runnerID)
} else {
r1 = ret.Error(1)
}
return r0, r1
return r0
}
// NewGithubEntityOperations creates a new instance of GithubEntityOperations. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.

View file

@ -1,4 +1,4 @@
// Code generated by mockery v2.42.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -14,24 +14,6 @@ type PoolManager struct {
mock.Mock
}
// DeleteRunner provides a mock function with given fields: runner, forceRemove, bypassGHUnauthorizedError
func (_m *PoolManager) DeleteRunner(runner params.Instance, forceRemove bool, bypassGHUnauthorizedError bool) error {
ret := _m.Called(runner, forceRemove, bypassGHUnauthorizedError)
if len(ret) == 0 {
panic("no return value specified for DeleteRunner")
}
var r0 error
if rf, ok := ret.Get(0).(func(params.Instance, bool, bool) error); ok {
r0 = rf(runner, forceRemove, bypassGHUnauthorizedError)
} else {
r0 = ret.Error(0)
}
return r0
}
// GetWebhookInfo provides a mock function with given fields: ctx
func (_m *PoolManager) GetWebhookInfo(ctx context.Context) (params.HookInfo, error) {
ret := _m.Called(ctx)
@ -60,7 +42,7 @@ func (_m *PoolManager) GetWebhookInfo(ctx context.Context) (params.HookInfo, err
return r0, r1
}
// GithubRunnerRegistrationToken provides a mock function with given fields:
// GithubRunnerRegistrationToken provides a mock function with no fields
func (_m *PoolManager) GithubRunnerRegistrationToken() (string, error) {
ret := _m.Called()
@ -106,7 +88,7 @@ func (_m *PoolManager) HandleWorkflowJob(job params.WorkflowJob) error {
return r0
}
// ID provides a mock function with given fields:
// ID provides a mock function with no fields
func (_m *PoolManager) ID() string {
ret := _m.Called()
@ -152,7 +134,7 @@ func (_m *PoolManager) InstallWebhook(ctx context.Context, param params.InstallW
return r0, r1
}
// RootCABundle provides a mock function with given fields:
// RootCABundle provides a mock function with no fields
func (_m *PoolManager) RootCABundle() (params.CertificateBundle, error) {
ret := _m.Called()
@ -180,7 +162,12 @@ func (_m *PoolManager) RootCABundle() (params.CertificateBundle, error) {
return r0, r1
}
// Start provides a mock function with given fields:
// SetPoolRunningState provides a mock function with given fields: isRunning, failureReason
func (_m *PoolManager) SetPoolRunningState(isRunning bool, failureReason string) {
_m.Called(isRunning, failureReason)
}
// Start provides a mock function with no fields
func (_m *PoolManager) Start() error {
ret := _m.Called()
@ -198,7 +185,7 @@ func (_m *PoolManager) Start() error {
return r0
}
// Status provides a mock function with given fields:
// Status provides a mock function with no fields
func (_m *PoolManager) Status() params.PoolManagerStatus {
ret := _m.Called()
@ -216,7 +203,7 @@ func (_m *PoolManager) Status() params.PoolManagerStatus {
return r0
}
// Stop provides a mock function with given fields:
// Stop provides a mock function with no fields
func (_m *PoolManager) Stop() error {
ret := _m.Called()
@ -252,7 +239,7 @@ func (_m *PoolManager) UninstallWebhook(ctx context.Context) error {
return r0
}
// Wait provides a mock function with given fields:
// Wait provides a mock function with no fields
func (_m *PoolManager) Wait() error {
ret := _m.Called()
@ -270,7 +257,7 @@ func (_m *PoolManager) Wait() error {
return r0
}
// WebhookSecret provides a mock function with given fields:
// WebhookSecret provides a mock function with no fields
func (_m *PoolManager) WebhookSecret() string {
ret := _m.Called()

View file

@ -1,15 +1,17 @@
// Code generated by mockery v2.42.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
import (
context "context"
common "github.com/cloudbase/garm/runner/common"
garm_provider_commonparams "github.com/cloudbase/garm-provider-common/params"
mock "github.com/stretchr/testify/mock"
params "github.com/cloudbase/garm/params"
"github.com/cloudbase/garm/runner/common"
)
// Provider is an autogenerated mock type for the Provider type
@ -17,7 +19,7 @@ type Provider struct {
mock.Mock
}
// AsParams provides a mock function with given fields:
// AsParams provides a mock function with no fields
func (_m *Provider) AsParams() params.Provider {
ret := _m.Called()
@ -35,9 +37,9 @@ func (_m *Provider) AsParams() params.Provider {
return r0
}
// CreateInstance provides a mock function with given fields: ctx, bootstrapParams
// CreateInstance provides a mock function with given fields: ctx, bootstrapParams, createInstanceParams
func (_m *Provider) CreateInstance(ctx context.Context, bootstrapParams garm_provider_commonparams.BootstrapInstance, createInstanceParams common.CreateInstanceParams) (garm_provider_commonparams.ProviderInstance, error) {
ret := _m.Called(ctx, bootstrapParams)
ret := _m.Called(ctx, bootstrapParams, createInstanceParams)
if len(ret) == 0 {
panic("no return value specified for CreateInstance")
@ -45,17 +47,17 @@ func (_m *Provider) CreateInstance(ctx context.Context, bootstrapParams garm_pro
var r0 garm_provider_commonparams.ProviderInstance
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, garm_provider_commonparams.BootstrapInstance) (garm_provider_commonparams.ProviderInstance, error)); ok {
return rf(ctx, bootstrapParams)
if rf, ok := ret.Get(0).(func(context.Context, garm_provider_commonparams.BootstrapInstance, common.CreateInstanceParams) (garm_provider_commonparams.ProviderInstance, error)); ok {
return rf(ctx, bootstrapParams, createInstanceParams)
}
if rf, ok := ret.Get(0).(func(context.Context, garm_provider_commonparams.BootstrapInstance) garm_provider_commonparams.ProviderInstance); ok {
r0 = rf(ctx, bootstrapParams)
if rf, ok := ret.Get(0).(func(context.Context, garm_provider_commonparams.BootstrapInstance, common.CreateInstanceParams) garm_provider_commonparams.ProviderInstance); ok {
r0 = rf(ctx, bootstrapParams, createInstanceParams)
} else {
r0 = ret.Get(0).(garm_provider_commonparams.ProviderInstance)
}
if rf, ok := ret.Get(1).(func(context.Context, garm_provider_commonparams.BootstrapInstance) error); ok {
r1 = rf(ctx, bootstrapParams)
if rf, ok := ret.Get(1).(func(context.Context, garm_provider_commonparams.BootstrapInstance, common.CreateInstanceParams) error); ok {
r1 = rf(ctx, bootstrapParams, createInstanceParams)
} else {
r1 = ret.Error(1)
}
@ -63,17 +65,17 @@ func (_m *Provider) CreateInstance(ctx context.Context, bootstrapParams garm_pro
return r0, r1
}
// DeleteInstance provides a mock function with given fields: ctx, instance
// DeleteInstance provides a mock function with given fields: ctx, instance, deleteInstanceParams
func (_m *Provider) DeleteInstance(ctx context.Context, instance string, deleteInstanceParams common.DeleteInstanceParams) error {
ret := _m.Called(ctx, instance)
ret := _m.Called(ctx, instance, deleteInstanceParams)
if len(ret) == 0 {
panic("no return value specified for DeleteInstance")
}
var r0 error
if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
r0 = rf(ctx, instance)
if rf, ok := ret.Get(0).(func(context.Context, string, common.DeleteInstanceParams) error); ok {
r0 = rf(ctx, instance, deleteInstanceParams)
} else {
r0 = ret.Error(0)
}
@ -81,7 +83,7 @@ func (_m *Provider) DeleteInstance(ctx context.Context, instance string, deleteI
return r0
}
// DisableJITConfig provides a mock function with given fields:
// DisableJITConfig provides a mock function with no fields
func (_m *Provider) DisableJITConfig() bool {
ret := _m.Called()
@ -99,9 +101,9 @@ func (_m *Provider) DisableJITConfig() bool {
return r0
}
// GetInstance provides a mock function with given fields: ctx, instance
// GetInstance provides a mock function with given fields: ctx, instance, getInstanceParams
func (_m *Provider) GetInstance(ctx context.Context, instance string, getInstanceParams common.GetInstanceParams) (garm_provider_commonparams.ProviderInstance, error) {
ret := _m.Called(ctx, instance)
ret := _m.Called(ctx, instance, getInstanceParams)
if len(ret) == 0 {
panic("no return value specified for GetInstance")
@ -109,17 +111,17 @@ func (_m *Provider) GetInstance(ctx context.Context, instance string, getInstanc
var r0 garm_provider_commonparams.ProviderInstance
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, string) (garm_provider_commonparams.ProviderInstance, error)); ok {
return rf(ctx, instance)
if rf, ok := ret.Get(0).(func(context.Context, string, common.GetInstanceParams) (garm_provider_commonparams.ProviderInstance, error)); ok {
return rf(ctx, instance, getInstanceParams)
}
if rf, ok := ret.Get(0).(func(context.Context, string) garm_provider_commonparams.ProviderInstance); ok {
r0 = rf(ctx, instance)
if rf, ok := ret.Get(0).(func(context.Context, string, common.GetInstanceParams) garm_provider_commonparams.ProviderInstance); ok {
r0 = rf(ctx, instance, getInstanceParams)
} else {
r0 = ret.Get(0).(garm_provider_commonparams.ProviderInstance)
}
if rf, ok := ret.Get(1).(func(context.Context, string) error); ok {
r1 = rf(ctx, instance)
if rf, ok := ret.Get(1).(func(context.Context, string, common.GetInstanceParams) error); ok {
r1 = rf(ctx, instance, getInstanceParams)
} else {
r1 = ret.Error(1)
}
@ -127,9 +129,9 @@ func (_m *Provider) GetInstance(ctx context.Context, instance string, getInstanc
return r0, r1
}
// ListInstances provides a mock function with given fields: ctx, poolID
// ListInstances provides a mock function with given fields: ctx, poolID, listInstancesParams
func (_m *Provider) ListInstances(ctx context.Context, poolID string, listInstancesParams common.ListInstancesParams) ([]garm_provider_commonparams.ProviderInstance, error) {
ret := _m.Called(ctx, poolID)
ret := _m.Called(ctx, poolID, listInstancesParams)
if len(ret) == 0 {
panic("no return value specified for ListInstances")
@ -137,19 +139,19 @@ func (_m *Provider) ListInstances(ctx context.Context, poolID string, listInstan
var r0 []garm_provider_commonparams.ProviderInstance
var r1 error
if rf, ok := ret.Get(0).(func(context.Context, string) ([]garm_provider_commonparams.ProviderInstance, error)); ok {
return rf(ctx, poolID)
if rf, ok := ret.Get(0).(func(context.Context, string, common.ListInstancesParams) ([]garm_provider_commonparams.ProviderInstance, error)); ok {
return rf(ctx, poolID, listInstancesParams)
}
if rf, ok := ret.Get(0).(func(context.Context, string) []garm_provider_commonparams.ProviderInstance); ok {
r0 = rf(ctx, poolID)
if rf, ok := ret.Get(0).(func(context.Context, string, common.ListInstancesParams) []garm_provider_commonparams.ProviderInstance); ok {
r0 = rf(ctx, poolID, listInstancesParams)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]garm_provider_commonparams.ProviderInstance)
}
}
if rf, ok := ret.Get(1).(func(context.Context, string) error); ok {
r1 = rf(ctx, poolID)
if rf, ok := ret.Get(1).(func(context.Context, string, common.ListInstancesParams) error); ok {
r1 = rf(ctx, poolID, listInstancesParams)
} else {
r1 = ret.Error(1)
}
@ -157,17 +159,17 @@ func (_m *Provider) ListInstances(ctx context.Context, poolID string, listInstan
return r0, r1
}
// RemoveAllInstances provides a mock function with given fields: ctx
func (_m *Provider) RemoveAllInstances(ctx context.Context, removeAllInstances common.RemoveAllInstancesParams) error {
ret := _m.Called(ctx)
// RemoveAllInstances provides a mock function with given fields: ctx, removeAllInstancesParams
func (_m *Provider) RemoveAllInstances(ctx context.Context, removeAllInstancesParams common.RemoveAllInstancesParams) error {
ret := _m.Called(ctx, removeAllInstancesParams)
if len(ret) == 0 {
panic("no return value specified for RemoveAllInstances")
}
var r0 error
if rf, ok := ret.Get(0).(func(context.Context) error); ok {
r0 = rf(ctx)
if rf, ok := ret.Get(0).(func(context.Context, common.RemoveAllInstancesParams) error); ok {
r0 = rf(ctx, removeAllInstancesParams)
} else {
r0 = ret.Error(0)
}
@ -175,17 +177,17 @@ func (_m *Provider) RemoveAllInstances(ctx context.Context, removeAllInstances c
return r0
}
// Start provides a mock function with given fields: ctx, instance
// Start provides a mock function with given fields: ctx, instance, startParams
func (_m *Provider) Start(ctx context.Context, instance string, startParams common.StartParams) error {
ret := _m.Called(ctx, instance)
ret := _m.Called(ctx, instance, startParams)
if len(ret) == 0 {
panic("no return value specified for Start")
}
var r0 error
if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
r0 = rf(ctx, instance)
if rf, ok := ret.Get(0).(func(context.Context, string, common.StartParams) error); ok {
r0 = rf(ctx, instance, startParams)
} else {
r0 = ret.Error(0)
}
@ -193,17 +195,17 @@ func (_m *Provider) Start(ctx context.Context, instance string, startParams comm
return r0
}
// Stop provides a mock function with given fields: ctx, instance
// Stop provides a mock function with given fields: ctx, instance, stopParams
func (_m *Provider) Stop(ctx context.Context, instance string, stopParams common.StopParams) error {
ret := _m.Called(ctx, instance)
ret := _m.Called(ctx, instance, stopParams)
if len(ret) == 0 {
panic("no return value specified for Stop")
}
var r0 error
if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
r0 = rf(ctx, instance)
if rf, ok := ret.Get(0).(func(context.Context, string, common.StopParams) error); ok {
r0 = rf(ctx, instance, stopParams)
} else {
r0 = ret.Error(0)
}

View file

@ -1,4 +1,4 @@
// Code generated by mockery v2.42.0. DO NOT EDIT.
// Code generated by mockery v2.53.3. DO NOT EDIT.
package mocks
@ -193,7 +193,7 @@ func (_m *PoolManagerController) GetEnterprisePoolManager(enterprise params.Ente
return r0, r1
}
// GetEnterprisePoolManagers provides a mock function with given fields:
// GetEnterprisePoolManagers provides a mock function with no fields
func (_m *PoolManagerController) GetEnterprisePoolManagers() (map[string]common.PoolManager, error) {
ret := _m.Called()
@ -253,7 +253,7 @@ func (_m *PoolManagerController) GetOrgPoolManager(org params.Organization) (com
return r0, r1
}
// GetOrgPoolManagers provides a mock function with given fields:
// GetOrgPoolManagers provides a mock function with no fields
func (_m *PoolManagerController) GetOrgPoolManagers() (map[string]common.PoolManager, error) {
ret := _m.Called()
@ -313,7 +313,7 @@ func (_m *PoolManagerController) GetRepoPoolManager(repo params.Repository) (com
return r0, r1
}
// GetRepoPoolManagers provides a mock function with given fields:
// GetRepoPoolManagers provides a mock function with no fields
func (_m *PoolManagerController) GetRepoPoolManagers() (map[string]common.PoolManager, error) {
ret := _m.Called()