From 91030584ba9db777aceece1826b4344d5b3840bf Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Thu, 21 Dec 2023 02:44:28 +0000
Subject: [PATCH 1/2] Bump github.com/vektra/mockery/v2 from 2.36.0 to 2.39.1
 in /tools

Bumps [github.com/vektra/mockery/v2](https://github.com/vektra/mockery) from 2.36.0 to 2.39.1.
- [Release notes](https://github.com/vektra/mockery/releases)
- [Changelog](https://github.com/vektra/mockery/blob/master/docs/changelog.md)
- [Commits](https://github.com/vektra/mockery/compare/v2.36.0...v2.39.1)

---
updated-dependencies:
- dependency-name: github.com/vektra/mockery/v2
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <support@github.com>
---
 tools/go.mod | 4 ++--
 tools/go.sum | 8 ++++----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/tools/go.mod b/tools/go.mod
index 827f7ce9fd..5a8c1b4011 100644
--- a/tools/go.mod
+++ b/tools/go.mod
@@ -20,7 +20,7 @@ require (
 	github.com/quasilyte/go-consistent v0.6.0
 	github.com/reviewdog/reviewdog v0.15.0
 	github.com/vburenin/ifacemaker v1.2.1
-	github.com/vektra/mockery/v2 v2.36.0
+	github.com/vektra/mockery/v2 v2.39.1
 	golang.org/x/perf v0.0.0-20230717203022-1ba3a21238c9
 	golang.org/x/tools v0.16.0
 	google.golang.org/grpc/cmd/protoc-gen-go-grpc v1.3.0
@@ -54,7 +54,7 @@ require (
 	github.com/bradleyfalzon/ghinstallation/v2 v2.6.0 // indirect
 	github.com/bufbuild/protocompile v0.6.1-0.20231027200301-089712432bdc // indirect
 	github.com/bufbuild/protovalidate-go v0.4.0 // indirect
-	github.com/chigopher/pathlib v0.15.0 // indirect
+	github.com/chigopher/pathlib v0.19.1 // indirect
 	github.com/cilium/ebpf v0.7.0 // indirect
 	github.com/cloudflare/circl v1.3.3 // indirect
 	github.com/containerd/stargz-snapshotter/estargz v0.15.1 // indirect
diff --git a/tools/go.sum b/tools/go.sum
index 0b18292f72..dda3efe684 100644
--- a/tools/go.sum
+++ b/tools/go.sum
@@ -136,8 +136,8 @@ github.com/bwesterb/go-ristretto v1.2.3/go.mod h1:fUIoIZaG73pV5biE2Blr2xEzDoMj7N
 github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=
 github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=
 github.com/cespare/xxhash/v2 v2.1.1/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
-github.com/chigopher/pathlib v0.15.0 h1:1pg96WL3iC1/YyWV4UJSl3E0GBf4B+h5amBtsbAAieY=
-github.com/chigopher/pathlib v0.15.0/go.mod h1:3+YPPV21mU9vyw8Mjp+F33CyCfE6iOzinpiqBcccv7I=
+github.com/chigopher/pathlib v0.19.1 h1:RoLlUJc0CqBGwq239cilyhxPNLXTK+HXoASGyGznx5A=
+github.com/chigopher/pathlib v0.19.1/go.mod h1:tzC1dZLW8o33UQpWkNkhvPwL5n4yyFRFm/jL1YGWFvY=
 github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
 github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
 github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
@@ -788,8 +788,8 @@ github.com/vbatts/tar-split v0.11.5 h1:3bHCTIheBm1qFTcgh9oPu+nNBtX+XJIupG/vacinC
 github.com/vbatts/tar-split v0.11.5/go.mod h1:yZbwRsSeGjusneWgA781EKej9HF8vme8okylkAeNKLk=
 github.com/vburenin/ifacemaker v1.2.1 h1:3Vq8B/bfBgjWTkv+jDg4dVL1KHt3k1K4lO7XRxYA2sk=
 github.com/vburenin/ifacemaker v1.2.1/go.mod h1:5WqrzX2aD7/hi+okBjcaEQJMg4lDGrpuEX3B8L4Wgrs=
-github.com/vektra/mockery/v2 v2.36.0 h1:TO4tST9f7lm4DHybPI2MC4Vb74bL89PtWvRhpNPAVo4=
-github.com/vektra/mockery/v2 v2.36.0/go.mod h1:diB13hxXG6QrTR0ol2Rk8s2dRMftzvExSvPDKr+IYKk=
+github.com/vektra/mockery/v2 v2.39.1 h1:zgnW69s+351ZF/L+O5pO64MpVP96aDtw8jwOGvGXzwU=
+github.com/vektra/mockery/v2 v2.39.1/go.mod h1:dPzGtjT0/Uu4hqpF6QNHwz+GLago7lq1bxdj9wHbGKo=
 github.com/vvakame/sdlog v1.2.0 h1:gwZRXZ0EmhJQJN/Do/+PTQigcmFiSqZ07aDjxqGOLT8=
 github.com/vvakame/sdlog v1.2.0/go.mod h1:gFYv2g/iR3pJSxkJz0YnkNmhNbXT5R3PzWsfZKGQADY=
 github.com/xanzy/go-gitlab v0.91.1 h1:gnV57IPGYywWer32oXKBcdmc8dVxeKl3AauV8Bu17rw=

From 68f0cfaf9f5f92453caef2088ef07e80760bbda2 Mon Sep 17 00:00:00 2001
From: Artem Gavrilov <artem.gavrilov@percona.com>
Date: Thu, 21 Dec 2023 12:30:44 +0200
Subject: [PATCH 2/2] Regenerate

---
 .../pmm/server/docker/mock_functions_test.go  |  62 ++++++++-
 agent/agentlocal/mock_client_test.go          |  14 ++-
 agent/agentlocal/mock_supervisor_test.go      |  10 +-
 agent/client/mock_connection_checker_test.go  |   6 +-
 agent/client/mock_service_info_broker_test.go |   6 +-
 agent/client/mock_supervisor_test.go          |  18 ++-
 agent/versioner/mock_exec_functions_test.go   |  10 +-
 .../backup/mock_agent_service_test.go         |   6 +-
 .../backup/mock_compatibility_service_test.go |  10 +-
 .../services/backup/mock_jobs_service_test.go |  22 +++-
 .../backup/mock_pbm_pitr_service_test.go      |  10 +-
 .../backup/mock_removal_service_test.go       |  10 +-
 managed/services/backup/mock_storage_test.go  |  18 ++-
 .../services/backup/mock_versioner_test.go    |   6 +-
 .../checks/mock_agents_registry_test.go       |  38 +++++-
 .../client/mock_kube_client_connector.go      | 118 +++++++++++++++++-
 .../grafana/mock_aws_instance_checker_test.go |   6 +-
 .../inventory/mock_agent_service_test.go      |   6 +-
 .../inventory/mock_agents_registry_test.go    |   6 +-
 .../mock_agents_state_updater_test.go         |   2 +-
 .../inventory/mock_connection_checker_test.go |   6 +-
 .../inventory/mock_inventory_metrics_test.go  |  14 ++-
 .../inventory/mock_prometheus_service_test.go |   2 +-
 .../mock_service_info_broker_test.go          |   6 +-
 .../inventory/mock_version_cache_test.go      |   2 +-
 .../alerting/mock_grafana_client_test.go      |  14 ++-
 .../management/backup/mock_aws_s3_test.go     |  14 ++-
 .../backup/mock_backup_service_test.go        |  14 ++-
 .../backup/mock_pbm_pitr_service_test.go      |   6 +-
 .../backup/mock_removal_service_test.go       |   6 +-
 .../backup/mock_schedule_service_test.go      |  14 ++-
 .../dbaas/mock_components_service_test.go     |  26 +++-
 .../dbaas/mock_dbaas_client_test.go           |  38 +++++-
 .../dbaas/mock_grafana_client_test.go         |  14 ++-
 .../dbaas/mock_kube_storage_manager_test.go   |  10 +-
 .../dbaas/mock_kubernetes_client_test.go      | 106 +++++++++++++++-
 .../dbaas/mock_version_service_test.go        |  30 ++++-
 .../management/dump/mock_dump_service_test.go |  14 ++-
 .../management/ia/mock_alert_manager_test.go  |  18 ++-
 .../ia/mock_templates_service_test.go         |   6 +-
 .../management/ia/mock_vm_alert_test.go       |   2 +-
 .../management/mock_agents_registry_test.go   |   6 +-
 .../mock_agents_state_updater_test.go         |   2 +-
 .../management/mock_api_key_provider_test.go  |  10 +-
 .../management/mock_checks_service_test.go    |  38 +++++-
 .../mock_connection_checker_test.go           |   6 +-
 .../management/mock_grafana_client_test.go    |   6 +-
 .../management/mock_jobs_service_test.go      |   6 +-
 .../mock_prometheus_service_test.go           |   2 +-
 .../mock_service_info_broker_test.go          |   6 +-
 .../management/mock_version_cache_test.go     |   2 +-
 .../mock_victoria_metrics_client_test.go      |   6 +-
 .../qan/mock_qan_collector_client_test.go     |   6 +-
 .../scheduler/mock_backup_service_test.go     |   6 +-
 .../server/mock_agents_state_updater_test.go  |   6 +-
 .../server/mock_alertmanager_service_test.go  |   6 +-
 .../server/mock_checks_service_test.go        |   6 +-
 managed/services/server/mock_emailer_test.go  |   6 +-
 .../server/mock_grafana_client_test.go        |   6 +-
 .../server/mock_prometheus_service_test.go    |   6 +-
 .../server/mock_rules_service_test.go         |   6 +-
 .../server/mock_supervisord_service_test.go   |  30 ++++-
 .../server/mock_telemetry_service_test.go     |  10 +-
 .../server/mock_templates_service_test.go     |   2 +-
 .../mock_vm_alert_external_rules_test.go      |  18 ++-
 .../mock_data_source_locator_test.go          |   6 +-
 .../telemetry/mock_data_source_test.go        |  18 ++-
 .../mock_distribution_util_service_test.go    |   6 +-
 .../services/telemetry/mock_sender_test.go    |   6 +-
 .../versioncache/mock_versioner_test.go       |   6 +-
 70 files changed, 922 insertions(+), 70 deletions(-)

diff --git a/admin/commands/pmm/server/docker/mock_functions_test.go b/admin/commands/pmm/server/docker/mock_functions_test.go
index 021064c634..028e561f94 100644
--- a/admin/commands/pmm/server/docker/mock_functions_test.go
+++ b/admin/commands/pmm/server/docker/mock_functions_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package docker
 
@@ -25,6 +25,10 @@ type MockFunctions struct {
 func (_m *MockFunctions) ChangeServerPassword(ctx context.Context, containerID string, newPassword string) error {
 	ret := _m.Called(ctx, containerID, newPassword)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ChangeServerPassword")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) error); ok {
 		r0 = rf(ctx, containerID, newPassword)
@@ -39,6 +43,10 @@ func (_m *MockFunctions) ChangeServerPassword(ctx context.Context, containerID s
 func (_m *MockFunctions) ContainerInspect(ctx context.Context, containerID string) (types.ContainerJSON, error) {
 	ret := _m.Called(ctx, containerID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ContainerInspect")
+	}
+
 	var r0 types.ContainerJSON
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (types.ContainerJSON, error)); ok {
@@ -63,6 +71,10 @@ func (_m *MockFunctions) ContainerInspect(ctx context.Context, containerID strin
 func (_m *MockFunctions) ContainerStop(ctx context.Context, containerID string, timeout *int) error {
 	ret := _m.Called(ctx, containerID, timeout)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ContainerStop")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, *int) error); ok {
 		r0 = rf(ctx, containerID, timeout)
@@ -77,6 +89,10 @@ func (_m *MockFunctions) ContainerStop(ctx context.Context, containerID string,
 func (_m *MockFunctions) ContainerUpdate(ctx context.Context, containerID string, updateConfig container.UpdateConfig) (container.ContainerUpdateOKBody, error) {
 	ret := _m.Called(ctx, containerID, updateConfig)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ContainerUpdate")
+	}
+
 	var r0 container.ContainerUpdateOKBody
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, container.UpdateConfig) (container.ContainerUpdateOKBody, error)); ok {
@@ -101,6 +117,10 @@ func (_m *MockFunctions) ContainerUpdate(ctx context.Context, containerID string
 func (_m *MockFunctions) ContainerWait(ctx context.Context, containerID string, condition container.WaitCondition) (<-chan container.WaitResponse, <-chan error) {
 	ret := _m.Called(ctx, containerID, condition)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ContainerWait")
+	}
+
 	var r0 <-chan container.WaitResponse
 	var r1 <-chan error
 	if rf, ok := ret.Get(0).(func(context.Context, string, container.WaitCondition) (<-chan container.WaitResponse, <-chan error)); ok {
@@ -129,6 +149,10 @@ func (_m *MockFunctions) ContainerWait(ctx context.Context, containerID string,
 func (_m *MockFunctions) CreateVolume(ctx context.Context, volumeName string, labels map[string]string) (*volume.Volume, error) {
 	ret := _m.Called(ctx, volumeName, labels)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateVolume")
+	}
+
 	var r0 *volume.Volume
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, map[string]string) (*volume.Volume, error)); ok {
@@ -155,6 +179,10 @@ func (_m *MockFunctions) CreateVolume(ctx context.Context, volumeName string, la
 func (_m *MockFunctions) FindServerContainers(ctx context.Context) ([]types.Container, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FindServerContainers")
+	}
+
 	var r0 []types.Container
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) ([]types.Container, error)); ok {
@@ -181,6 +209,10 @@ func (_m *MockFunctions) FindServerContainers(ctx context.Context) ([]types.Cont
 func (_m *MockFunctions) GetDockerClient() *client.Client {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDockerClient")
+	}
+
 	var r0 *client.Client
 	if rf, ok := ret.Get(0).(func() *client.Client); ok {
 		r0 = rf()
@@ -197,6 +229,10 @@ func (_m *MockFunctions) GetDockerClient() *client.Client {
 func (_m *MockFunctions) HaveDockerAccess(ctx context.Context) bool {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for HaveDockerAccess")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(context.Context) bool); ok {
 		r0 = rf(ctx)
@@ -211,6 +247,10 @@ func (_m *MockFunctions) HaveDockerAccess(ctx context.Context) bool {
 func (_m *MockFunctions) InstallDocker(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InstallDocker")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
@@ -225,6 +265,10 @@ func (_m *MockFunctions) InstallDocker(ctx context.Context) error {
 func (_m *MockFunctions) IsDockerInstalled() (bool, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsDockerInstalled")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (bool, error)); ok {
@@ -249,6 +293,10 @@ func (_m *MockFunctions) IsDockerInstalled() (bool, error) {
 func (_m *MockFunctions) ParsePullImageProgress(r io.Reader, p *tea.Program) (<-chan struct{}, <-chan error) {
 	ret := _m.Called(r, p)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ParsePullImageProgress")
+	}
+
 	var r0 <-chan struct{}
 	var r1 <-chan error
 	if rf, ok := ret.Get(0).(func(io.Reader, *tea.Program) (<-chan struct{}, <-chan error)); ok {
@@ -277,6 +325,10 @@ func (_m *MockFunctions) ParsePullImageProgress(r io.Reader, p *tea.Program) (<-
 func (_m *MockFunctions) PullImage(ctx context.Context, dockerImage string, opts types.ImagePullOptions) (io.Reader, error) {
 	ret := _m.Called(ctx, dockerImage, opts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PullImage")
+	}
+
 	var r0 io.Reader
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, types.ImagePullOptions) (io.Reader, error)); ok {
@@ -303,6 +355,10 @@ func (_m *MockFunctions) PullImage(ctx context.Context, dockerImage string, opts
 func (_m *MockFunctions) RunContainer(ctx context.Context, config *container.Config, hostConfig *container.HostConfig, containerName string) (string, error) {
 	ret := _m.Called(ctx, config, hostConfig, containerName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RunContainer")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *container.Config, *container.HostConfig, string) (string, error)); ok {
@@ -327,6 +383,10 @@ func (_m *MockFunctions) RunContainer(ctx context.Context, config *container.Con
 func (_m *MockFunctions) WaitForHealthyContainer(ctx context.Context, containerID string) <-chan pkgdocker.WaitHealthyResponse {
 	ret := _m.Called(ctx, containerID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WaitForHealthyContainer")
+	}
+
 	var r0 <-chan pkgdocker.WaitHealthyResponse
 	if rf, ok := ret.Get(0).(func(context.Context, string) <-chan pkgdocker.WaitHealthyResponse); ok {
 		r0 = rf(ctx, containerID)
diff --git a/agent/agentlocal/mock_client_test.go b/agent/agentlocal/mock_client_test.go
index e4e0e47cad..dfa57c8e84 100644
--- a/agent/agentlocal/mock_client_test.go
+++ b/agent/agentlocal/mock_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package agentlocal
 
@@ -30,6 +30,10 @@ func (_m *mockClient) Describe(_a0 chan<- *prometheus.Desc) {
 func (_m *mockClient) GetConnectionUpTime() float32 {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetConnectionUpTime")
+	}
+
 	var r0 float32
 	if rf, ok := ret.Get(0).(func() float32); ok {
 		r0 = rf()
@@ -44,6 +48,10 @@ func (_m *mockClient) GetConnectionUpTime() float32 {
 func (_m *mockClient) GetNetworkInformation() (time.Duration, time.Duration, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNetworkInformation")
+	}
+
 	var r0 time.Duration
 	var r1 time.Duration
 	var r2 error
@@ -75,6 +83,10 @@ func (_m *mockClient) GetNetworkInformation() (time.Duration, time.Duration, err
 func (_m *mockClient) GetServerConnectMetadata() *agentpb.ServerConnectMetadata {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetServerConnectMetadata")
+	}
+
 	var r0 *agentpb.ServerConnectMetadata
 	if rf, ok := ret.Get(0).(func() *agentpb.ServerConnectMetadata); ok {
 		r0 = rf()
diff --git a/agent/agentlocal/mock_supervisor_test.go b/agent/agentlocal/mock_supervisor_test.go
index 6201748dbf..aea0b4a37b 100644
--- a/agent/agentlocal/mock_supervisor_test.go
+++ b/agent/agentlocal/mock_supervisor_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package agentlocal
 
@@ -17,6 +17,10 @@ type mockSupervisor struct {
 func (_m *mockSupervisor) AgentsList() []*agentlocalpb.AgentInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for AgentsList")
+	}
+
 	var r0 []*agentlocalpb.AgentInfo
 	if rf, ok := ret.Get(0).(func() []*agentlocalpb.AgentInfo); ok {
 		r0 = rf()
@@ -33,6 +37,10 @@ func (_m *mockSupervisor) AgentsList() []*agentlocalpb.AgentInfo {
 func (_m *mockSupervisor) AgentsLogs() map[string][]string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for AgentsLogs")
+	}
+
 	var r0 map[string][]string
 	if rf, ok := ret.Get(0).(func() map[string][]string); ok {
 		r0 = rf()
diff --git a/agent/client/mock_connection_checker_test.go b/agent/client/mock_connection_checker_test.go
index 610ab654c1..82fbf51c2c 100644
--- a/agent/client/mock_connection_checker_test.go
+++ b/agent/client/mock_connection_checker_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package client
 
@@ -19,6 +19,10 @@ type mockConnectionChecker struct {
 func (_m *mockConnectionChecker) Check(ctx context.Context, req *agentpb.CheckConnectionRequest, id uint32) *agentpb.CheckConnectionResponse {
 	ret := _m.Called(ctx, req, id)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Check")
+	}
+
 	var r0 *agentpb.CheckConnectionResponse
 	if rf, ok := ret.Get(0).(func(context.Context, *agentpb.CheckConnectionRequest, uint32) *agentpb.CheckConnectionResponse); ok {
 		r0 = rf(ctx, req, id)
diff --git a/agent/client/mock_service_info_broker_test.go b/agent/client/mock_service_info_broker_test.go
index 9123c456fc..34c16a522b 100644
--- a/agent/client/mock_service_info_broker_test.go
+++ b/agent/client/mock_service_info_broker_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package client
 
@@ -19,6 +19,10 @@ type mockServiceInfoBroker struct {
 func (_m *mockServiceInfoBroker) GetInfoFromService(ctx context.Context, req *agentpb.ServiceInfoRequest, id uint32) *agentpb.ServiceInfoResponse {
 	ret := _m.Called(ctx, req, id)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetInfoFromService")
+	}
+
 	var r0 *agentpb.ServiceInfoResponse
 	if rf, ok := ret.Get(0).(func(context.Context, *agentpb.ServiceInfoRequest, uint32) *agentpb.ServiceInfoResponse); ok {
 		r0 = rf(ctx, req, id)
diff --git a/agent/client/mock_supervisor_test.go b/agent/client/mock_supervisor_test.go
index 1a53dbbae8..2654fd4bee 100644
--- a/agent/client/mock_supervisor_test.go
+++ b/agent/client/mock_supervisor_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package client
 
@@ -19,6 +19,10 @@ type mockSupervisor struct {
 func (_m *mockSupervisor) AgentLogByID(_a0 string) ([]string, uint) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for AgentLogByID")
+	}
+
 	var r0 []string
 	var r1 uint
 	if rf, ok := ret.Get(0).(func(string) ([]string, uint)); ok {
@@ -45,6 +49,10 @@ func (_m *mockSupervisor) AgentLogByID(_a0 string) ([]string, uint) {
 func (_m *mockSupervisor) AgentsList() []*agentlocalpb.AgentInfo {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for AgentsList")
+	}
+
 	var r0 []*agentlocalpb.AgentInfo
 	if rf, ok := ret.Get(0).(func() []*agentlocalpb.AgentInfo); ok {
 		r0 = rf()
@@ -61,6 +69,10 @@ func (_m *mockSupervisor) AgentsList() []*agentlocalpb.AgentInfo {
 func (_m *mockSupervisor) Changes() <-chan *agentpb.StateChangedRequest {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Changes")
+	}
+
 	var r0 <-chan *agentpb.StateChangedRequest
 	if rf, ok := ret.Get(0).(func() <-chan *agentpb.StateChangedRequest); ok {
 		r0 = rf()
@@ -92,6 +104,10 @@ func (_m *mockSupervisor) Describe(_a0 chan<- *prometheus.Desc) {
 func (_m *mockSupervisor) QANRequests() <-chan *agentpb.QANCollectRequest {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for QANRequests")
+	}
+
 	var r0 <-chan *agentpb.QANCollectRequest
 	if rf, ok := ret.Get(0).(func() <-chan *agentpb.QANCollectRequest); ok {
 		r0 = rf()
diff --git a/agent/versioner/mock_exec_functions_test.go b/agent/versioner/mock_exec_functions_test.go
index 4912d643ba..a6100a0014 100644
--- a/agent/versioner/mock_exec_functions_test.go
+++ b/agent/versioner/mock_exec_functions_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package versioner
 
@@ -24,6 +24,10 @@ func (_m *MockExecFunctions) CommandContext(ctx context.Context, name string, ar
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CommandContext")
+	}
+
 	var r0 CombinedOutputer
 	if rf, ok := ret.Get(0).(func(context.Context, string, ...string) CombinedOutputer); ok {
 		r0 = rf(ctx, name, arg...)
@@ -40,6 +44,10 @@ func (_m *MockExecFunctions) CommandContext(ctx context.Context, name string, ar
 func (_m *MockExecFunctions) LookPath(file string) (string, error) {
 	ret := _m.Called(file)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LookPath")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (string, error)); ok {
diff --git a/managed/services/backup/mock_agent_service_test.go b/managed/services/backup/mock_agent_service_test.go
index 7f3940e50b..a434a89a05 100644
--- a/managed/services/backup/mock_agent_service_test.go
+++ b/managed/services/backup/mock_agent_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -17,6 +17,10 @@ type mockAgentService struct {
 func (_m *mockAgentService) PBMSwitchPITR(pmmAgentID string, dsn string, files map[string]string, tdp *models.DelimiterPair, enabled bool) error {
 	ret := _m.Called(pmmAgentID, dsn, files, tdp, enabled)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PBMSwitchPITR")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string, map[string]string, *models.DelimiterPair, bool) error); ok {
 		r0 = rf(pmmAgentID, dsn, files, tdp, enabled)
diff --git a/managed/services/backup/mock_compatibility_service_test.go b/managed/services/backup/mock_compatibility_service_test.go
index a7db3db17c..e4280076f0 100644
--- a/managed/services/backup/mock_compatibility_service_test.go
+++ b/managed/services/backup/mock_compatibility_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -17,6 +17,10 @@ type mockCompatibilityService struct {
 func (_m *mockCompatibilityService) CheckArtifactCompatibility(artifactID string, targetDBVersion string) error {
 	ret := _m.Called(artifactID, targetDBVersion)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckArtifactCompatibility")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string) error); ok {
 		r0 = rf(artifactID, targetDBVersion)
@@ -31,6 +35,10 @@ func (_m *mockCompatibilityService) CheckArtifactCompatibility(artifactID string
 func (_m *mockCompatibilityService) CheckSoftwareCompatibilityForService(ctx context.Context, serviceID string) (string, error) {
 	ret := _m.Called(ctx, serviceID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckSoftwareCompatibilityForService")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (string, error)); ok {
diff --git a/managed/services/backup/mock_jobs_service_test.go b/managed/services/backup/mock_jobs_service_test.go
index cc048818f3..830fbb6f02 100644
--- a/managed/services/backup/mock_jobs_service_test.go
+++ b/managed/services/backup/mock_jobs_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -19,6 +19,10 @@ type mockJobsService struct {
 func (_m *mockJobsService) StartMongoDBBackupJob(service *models.Service, jobID string, pmmAgentID string, timeout time.Duration, name string, dbConfig *models.DBConfig, mode models.BackupMode, dataModel models.DataModel, locationConfig *models.BackupLocationConfig, folder string) error {
 	ret := _m.Called(service, jobID, pmmAgentID, timeout, name, dbConfig, mode, dataModel, locationConfig, folder)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMongoDBBackupJob")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*models.Service, string, string, time.Duration, string, *models.DBConfig, models.BackupMode, models.DataModel, *models.BackupLocationConfig, string) error); ok {
 		r0 = rf(service, jobID, pmmAgentID, timeout, name, dbConfig, mode, dataModel, locationConfig, folder)
@@ -33,6 +37,10 @@ func (_m *mockJobsService) StartMongoDBBackupJob(service *models.Service, jobID
 func (_m *mockJobsService) StartMongoDBRestoreBackupJob(service *models.Service, jobID string, pmmAgentID string, timeout time.Duration, name string, pbmBackupName string, dbConfig *models.DBConfig, dataModel models.DataModel, locationConfig *models.BackupLocationConfig, pitrTimestamp time.Time, folder string) error {
 	ret := _m.Called(service, jobID, pmmAgentID, timeout, name, pbmBackupName, dbConfig, dataModel, locationConfig, pitrTimestamp, folder)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMongoDBRestoreBackupJob")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*models.Service, string, string, time.Duration, string, string, *models.DBConfig, models.DataModel, *models.BackupLocationConfig, time.Time, string) error); ok {
 		r0 = rf(service, jobID, pmmAgentID, timeout, name, pbmBackupName, dbConfig, dataModel, locationConfig, pitrTimestamp, folder)
@@ -47,6 +55,10 @@ func (_m *mockJobsService) StartMongoDBRestoreBackupJob(service *models.Service,
 func (_m *mockJobsService) StartMySQLBackupJob(jobID string, pmmAgentID string, timeout time.Duration, name string, dbConfig *models.DBConfig, locationConfig *models.BackupLocationConfig, folder string) error {
 	ret := _m.Called(jobID, pmmAgentID, timeout, name, dbConfig, locationConfig, folder)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMySQLBackupJob")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string, time.Duration, string, *models.DBConfig, *models.BackupLocationConfig, string) error); ok {
 		r0 = rf(jobID, pmmAgentID, timeout, name, dbConfig, locationConfig, folder)
@@ -61,6 +73,10 @@ func (_m *mockJobsService) StartMySQLBackupJob(jobID string, pmmAgentID string,
 func (_m *mockJobsService) StartMySQLRestoreBackupJob(jobID string, pmmAgentID string, serviceID string, timeout time.Duration, name string, locationConfig *models.BackupLocationConfig, folder string) error {
 	ret := _m.Called(jobID, pmmAgentID, serviceID, timeout, name, locationConfig, folder)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMySQLRestoreBackupJob")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, string, string, time.Duration, string, *models.BackupLocationConfig, string) error); ok {
 		r0 = rf(jobID, pmmAgentID, serviceID, timeout, name, locationConfig, folder)
@@ -75,6 +91,10 @@ func (_m *mockJobsService) StartMySQLRestoreBackupJob(jobID string, pmmAgentID s
 func (_m *mockJobsService) StopJob(jobID string) error {
 	ret := _m.Called(jobID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StopJob")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(jobID)
diff --git a/managed/services/backup/mock_pbm_pitr_service_test.go b/managed/services/backup/mock_pbm_pitr_service_test.go
index c0255e8763..fc3274e3b0 100644
--- a/managed/services/backup/mock_pbm_pitr_service_test.go
+++ b/managed/services/backup/mock_pbm_pitr_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -20,6 +20,10 @@ type mockPbmPITRService struct {
 func (_m *mockPbmPITRService) GetPITRFiles(ctx context.Context, locationClient Storage, location *models.BackupLocation, artifact *models.Artifact, until *time.Time) ([]*oplogChunk, error) {
 	ret := _m.Called(ctx, locationClient, location, artifact, until)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPITRFiles")
+	}
+
 	var r0 []*oplogChunk
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, Storage, *models.BackupLocation, *models.Artifact, *time.Time) ([]*oplogChunk, error)); ok {
@@ -46,6 +50,10 @@ func (_m *mockPbmPITRService) GetPITRFiles(ctx context.Context, locationClient S
 func (_m *mockPbmPITRService) ListPITRTimeranges(ctx context.Context, locationClient Storage, location *models.BackupLocation, artifact *models.Artifact) ([]Timeline, error) {
 	ret := _m.Called(ctx, locationClient, location, artifact)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListPITRTimeranges")
+	}
+
 	var r0 []Timeline
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, Storage, *models.BackupLocation, *models.Artifact) ([]Timeline, error)); ok {
diff --git a/managed/services/backup/mock_removal_service_test.go b/managed/services/backup/mock_removal_service_test.go
index c3364c75c9..e106e6267a 100644
--- a/managed/services/backup/mock_removal_service_test.go
+++ b/managed/services/backup/mock_removal_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -13,6 +13,10 @@ type mockRemovalService struct {
 func (_m *mockRemovalService) DeleteArtifact(storage Storage, artifactID string, removeFiles bool) error {
 	ret := _m.Called(storage, artifactID, removeFiles)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteArtifact")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(Storage, string, bool) error); ok {
 		r0 = rf(storage, artifactID, removeFiles)
@@ -27,6 +31,10 @@ func (_m *mockRemovalService) DeleteArtifact(storage Storage, artifactID string,
 func (_m *mockRemovalService) TrimPITRArtifact(storage Storage, artifactID string, firstN int) error {
 	ret := _m.Called(storage, artifactID, firstN)
 
+	if len(ret) == 0 {
+		panic("no return value specified for TrimPITRArtifact")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(Storage, string, int) error); ok {
 		r0 = rf(storage, artifactID, firstN)
diff --git a/managed/services/backup/mock_storage_test.go b/managed/services/backup/mock_storage_test.go
index 150f5a7c21..8915b3d768 100644
--- a/managed/services/backup/mock_storage_test.go
+++ b/managed/services/backup/mock_storage_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -19,6 +19,10 @@ type MockStorage struct {
 func (_m *MockStorage) FileStat(ctx context.Context, endpoint string, accessKey string, secretKey string, bucketName string, name string) (minio.FileInfo, error) {
 	ret := _m.Called(ctx, endpoint, accessKey, secretKey, bucketName, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FileStat")
+	}
+
 	var r0 minio.FileInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, string) (minio.FileInfo, error)); ok {
@@ -43,6 +47,10 @@ func (_m *MockStorage) FileStat(ctx context.Context, endpoint string, accessKey
 func (_m *MockStorage) List(ctx context.Context, endpoint string, accessKey string, secretKey string, bucketName string, prefix string, suffix string) ([]minio.FileInfo, error) {
 	ret := _m.Called(ctx, endpoint, accessKey, secretKey, bucketName, prefix, suffix)
 
+	if len(ret) == 0 {
+		panic("no return value specified for List")
+	}
+
 	var r0 []minio.FileInfo
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, string, string) ([]minio.FileInfo, error)); ok {
@@ -69,6 +77,10 @@ func (_m *MockStorage) List(ctx context.Context, endpoint string, accessKey stri
 func (_m *MockStorage) Remove(ctx context.Context, endpoint string, accessKey string, secretKey string, bucketName string, objectName string) error {
 	ret := _m.Called(ctx, endpoint, accessKey, secretKey, bucketName, objectName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, string) error); ok {
 		r0 = rf(ctx, endpoint, accessKey, secretKey, bucketName, objectName)
@@ -83,6 +95,10 @@ func (_m *MockStorage) Remove(ctx context.Context, endpoint string, accessKey st
 func (_m *MockStorage) RemoveRecursive(ctx context.Context, endpoint string, accessKey string, secretKey string, bucketName string, prefix string) error {
 	ret := _m.Called(ctx, endpoint, accessKey, secretKey, bucketName, prefix)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveRecursive")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, string) error); ok {
 		r0 = rf(ctx, endpoint, accessKey, secretKey, bucketName, prefix)
diff --git a/managed/services/backup/mock_versioner_test.go b/managed/services/backup/mock_versioner_test.go
index bb1443d3c1..f7db4355bd 100644
--- a/managed/services/backup/mock_versioner_test.go
+++ b/managed/services/backup/mock_versioner_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -17,6 +17,10 @@ type mockVersioner struct {
 func (_m *mockVersioner) GetVersions(pmmAgentID string, softwares []agents.Software) ([]agents.Version, error) {
 	ret := _m.Called(pmmAgentID, softwares)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetVersions")
+	}
+
 	var r0 []agents.Version
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, []agents.Software) ([]agents.Version, error)); ok {
diff --git a/managed/services/checks/mock_agents_registry_test.go b/managed/services/checks/mock_agents_registry_test.go
index 52d0f05d89..369685f13b 100644
--- a/managed/services/checks/mock_agents_registry_test.go
+++ b/managed/services/checks/mock_agents_registry_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package checks
 
@@ -19,6 +19,10 @@ type mockAgentsRegistry struct {
 func (_m *mockAgentsRegistry) StartMongoDBQueryBuildInfoAction(ctx context.Context, id string, pmmAgentID string, dsn string, files map[string]string, tdp *models.DelimiterPair) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, files, tdp)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMongoDBQueryBuildInfoAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, map[string]string, *models.DelimiterPair) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, files, tdp)
@@ -33,6 +37,10 @@ func (_m *mockAgentsRegistry) StartMongoDBQueryBuildInfoAction(ctx context.Conte
 func (_m *mockAgentsRegistry) StartMongoDBQueryGetCmdLineOptsAction(ctx context.Context, id string, pmmAgentID string, dsn string, files map[string]string, tdp *models.DelimiterPair) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, files, tdp)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMongoDBQueryGetCmdLineOptsAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, map[string]string, *models.DelimiterPair) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, files, tdp)
@@ -47,6 +55,10 @@ func (_m *mockAgentsRegistry) StartMongoDBQueryGetCmdLineOptsAction(ctx context.
 func (_m *mockAgentsRegistry) StartMongoDBQueryGetDiagnosticDataAction(ctx context.Context, id string, pmmAgentID string, dsn string, files map[string]string, tdp *models.DelimiterPair) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, files, tdp)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMongoDBQueryGetDiagnosticDataAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, map[string]string, *models.DelimiterPair) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, files, tdp)
@@ -61,6 +73,10 @@ func (_m *mockAgentsRegistry) StartMongoDBQueryGetDiagnosticDataAction(ctx conte
 func (_m *mockAgentsRegistry) StartMongoDBQueryGetParameterAction(ctx context.Context, id string, pmmAgentID string, dsn string, files map[string]string, tdp *models.DelimiterPair) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, files, tdp)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMongoDBQueryGetParameterAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, map[string]string, *models.DelimiterPair) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, files, tdp)
@@ -75,6 +91,10 @@ func (_m *mockAgentsRegistry) StartMongoDBQueryGetParameterAction(ctx context.Co
 func (_m *mockAgentsRegistry) StartMongoDBQueryReplSetGetStatusAction(ctx context.Context, id string, pmmAgentID string, dsn string, files map[string]string, tdp *models.DelimiterPair) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, files, tdp)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMongoDBQueryReplSetGetStatusAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, map[string]string, *models.DelimiterPair) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, files, tdp)
@@ -89,6 +109,10 @@ func (_m *mockAgentsRegistry) StartMongoDBQueryReplSetGetStatusAction(ctx contex
 func (_m *mockAgentsRegistry) StartMySQLQuerySelectAction(ctx context.Context, id string, pmmAgentID string, dsn string, query string, files map[string]string, tdp *models.DelimiterPair, tlsSkipVerify bool) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, query, files, tdp, tlsSkipVerify)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMySQLQuerySelectAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, map[string]string, *models.DelimiterPair, bool) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, query, files, tdp, tlsSkipVerify)
@@ -103,6 +127,10 @@ func (_m *mockAgentsRegistry) StartMySQLQuerySelectAction(ctx context.Context, i
 func (_m *mockAgentsRegistry) StartMySQLQueryShowAction(ctx context.Context, id string, pmmAgentID string, dsn string, query string, files map[string]string, tdp *models.DelimiterPair, tlsSkipVerify bool) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, query, files, tdp, tlsSkipVerify)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMySQLQueryShowAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, map[string]string, *models.DelimiterPair, bool) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, query, files, tdp, tlsSkipVerify)
@@ -117,6 +145,10 @@ func (_m *mockAgentsRegistry) StartMySQLQueryShowAction(ctx context.Context, id
 func (_m *mockAgentsRegistry) StartPostgreSQLQuerySelectAction(ctx context.Context, id string, pmmAgentID string, dsn string, query string) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn, query)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartPostgreSQLQuerySelectAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn, query)
@@ -131,6 +163,10 @@ func (_m *mockAgentsRegistry) StartPostgreSQLQuerySelectAction(ctx context.Conte
 func (_m *mockAgentsRegistry) StartPostgreSQLQueryShowAction(ctx context.Context, id string, pmmAgentID string, dsn string) error {
 	ret := _m.Called(ctx, id, pmmAgentID, dsn)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartPostgreSQLQueryShowAction")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string) error); ok {
 		r0 = rf(ctx, id, pmmAgentID, dsn)
diff --git a/managed/services/dbaas/kubernetes/client/mock_kube_client_connector.go b/managed/services/dbaas/kubernetes/client/mock_kube_client_connector.go
index 8bddf6fb1e..94addfdb5c 100644
--- a/managed/services/dbaas/kubernetes/client/mock_kube_client_connector.go
+++ b/managed/services/dbaas/kubernetes/client/mock_kube_client_connector.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package client
 
@@ -30,6 +30,10 @@ type MockKubeClientConnector struct {
 func (_m *MockKubeClientConnector) ApplyFile(fileBytes []byte) error {
 	ret := _m.Called(fileBytes)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ApplyFile")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]byte) error); ok {
 		r0 = rf(fileBytes)
@@ -44,6 +48,10 @@ func (_m *MockKubeClientConnector) ApplyFile(fileBytes []byte) error {
 func (_m *MockKubeClientConnector) ApplyObject(obj runtime.Object) error {
 	ret := _m.Called(obj)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ApplyObject")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(runtime.Object) error); ok {
 		r0 = rf(obj)
@@ -58,6 +66,10 @@ func (_m *MockKubeClientConnector) ApplyObject(obj runtime.Object) error {
 func (_m *MockKubeClientConnector) CreateOperatorGroup(ctx context.Context, namespace string, name string) (*v1.OperatorGroup, error) {
 	ret := _m.Called(ctx, namespace, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateOperatorGroup")
+	}
+
 	var r0 *v1.OperatorGroup
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (*v1.OperatorGroup, error)); ok {
@@ -84,6 +96,10 @@ func (_m *MockKubeClientConnector) CreateOperatorGroup(ctx context.Context, name
 func (_m *MockKubeClientConnector) CreateSubscriptionForCatalog(ctx context.Context, namespace string, name string, catalogNamespace string, catalog string, packageName string, channel string, startingCSV string, approval v1alpha1.Approval) (*v1alpha1.Subscription, error) {
 	ret := _m.Called(ctx, namespace, name, catalogNamespace, catalog, packageName, channel, startingCSV, approval)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateSubscriptionForCatalog")
+	}
+
 	var r0 *v1alpha1.Subscription
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, string, string, string, v1alpha1.Approval) (*v1alpha1.Subscription, error)); ok {
@@ -110,6 +126,10 @@ func (_m *MockKubeClientConnector) CreateSubscriptionForCatalog(ctx context.Cont
 func (_m *MockKubeClientConnector) DeleteObject(obj runtime.Object) error {
 	ret := _m.Called(obj)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteObject")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(runtime.Object) error); ok {
 		r0 = rf(obj)
@@ -124,6 +144,10 @@ func (_m *MockKubeClientConnector) DeleteObject(obj runtime.Object) error {
 func (_m *MockKubeClientConnector) DoCSVWait(ctx context.Context, key types.NamespacedName) error {
 	ret := _m.Called(ctx, key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DoCSVWait")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, types.NamespacedName) error); ok {
 		r0 = rf(ctx, key)
@@ -138,6 +162,10 @@ func (_m *MockKubeClientConnector) DoCSVWait(ctx context.Context, key types.Name
 func (_m *MockKubeClientConnector) DoRolloutWait(ctx context.Context, key types.NamespacedName) error {
 	ret := _m.Called(ctx, key)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DoRolloutWait")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, types.NamespacedName) error); ok {
 		r0 = rf(ctx, key)
@@ -152,6 +180,10 @@ func (_m *MockKubeClientConnector) DoRolloutWait(ctx context.Context, key types.
 func (_m *MockKubeClientConnector) GenerateKubeConfig(secret *corev1.Secret) ([]byte, error) {
 	ret := _m.Called(secret)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GenerateKubeConfig")
+	}
+
 	var r0 []byte
 	var r1 error
 	if rf, ok := ret.Get(0).(func(*corev1.Secret) ([]byte, error)); ok {
@@ -178,6 +210,10 @@ func (_m *MockKubeClientConnector) GenerateKubeConfig(secret *corev1.Secret) ([]
 func (_m *MockKubeClientConnector) GetDatabaseCluster(ctx context.Context, name string) (*apiv1.DatabaseCluster, error) {
 	ret := _m.Called(ctx, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDatabaseCluster")
+	}
+
 	var r0 *apiv1.DatabaseCluster
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*apiv1.DatabaseCluster, error)); ok {
@@ -204,6 +240,10 @@ func (_m *MockKubeClientConnector) GetDatabaseCluster(ctx context.Context, name
 func (_m *MockKubeClientConnector) GetDeployment(ctx context.Context, name string) (*appsv1.Deployment, error) {
 	ret := _m.Called(ctx, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDeployment")
+	}
+
 	var r0 *appsv1.Deployment
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*appsv1.Deployment, error)); ok {
@@ -230,6 +270,10 @@ func (_m *MockKubeClientConnector) GetDeployment(ctx context.Context, name strin
 func (_m *MockKubeClientConnector) GetEvents(ctx context.Context, name string) (string, error) {
 	ret := _m.Called(ctx, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetEvents")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (string, error)); ok {
@@ -254,6 +298,10 @@ func (_m *MockKubeClientConnector) GetEvents(ctx context.Context, name string) (
 func (_m *MockKubeClientConnector) GetInstallPlan(ctx context.Context, namespace string, name string) (*v1alpha1.InstallPlan, error) {
 	ret := _m.Called(ctx, namespace, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetInstallPlan")
+	}
+
 	var r0 *v1alpha1.InstallPlan
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (*v1alpha1.InstallPlan, error)); ok {
@@ -280,6 +328,10 @@ func (_m *MockKubeClientConnector) GetInstallPlan(ctx context.Context, namespace
 func (_m *MockKubeClientConnector) GetLogs(ctx context.Context, pod string, container string) (string, error) {
 	ret := _m.Called(ctx, pod, container)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetLogs")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (string, error)); ok {
@@ -304,6 +356,10 @@ func (_m *MockKubeClientConnector) GetLogs(ctx context.Context, pod string, cont
 func (_m *MockKubeClientConnector) GetNodes(ctx context.Context) (*corev1.NodeList, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodes")
+	}
+
 	var r0 *corev1.NodeList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*corev1.NodeList, error)); ok {
@@ -330,6 +386,10 @@ func (_m *MockKubeClientConnector) GetNodes(ctx context.Context) (*corev1.NodeLi
 func (_m *MockKubeClientConnector) GetOperatorGroup(ctx context.Context, namespace string, name string) (*v1.OperatorGroup, error) {
 	ret := _m.Called(ctx, namespace, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetOperatorGroup")
+	}
+
 	var r0 *v1.OperatorGroup
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (*v1.OperatorGroup, error)); ok {
@@ -356,6 +416,10 @@ func (_m *MockKubeClientConnector) GetOperatorGroup(ctx context.Context, namespa
 func (_m *MockKubeClientConnector) GetPersistentVolumes(ctx context.Context) (*corev1.PersistentVolumeList, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPersistentVolumes")
+	}
+
 	var r0 *corev1.PersistentVolumeList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*corev1.PersistentVolumeList, error)); ok {
@@ -382,6 +446,10 @@ func (_m *MockKubeClientConnector) GetPersistentVolumes(ctx context.Context) (*c
 func (_m *MockKubeClientConnector) GetPods(ctx context.Context, namespace string, labelSelector *metav1.LabelSelector) (*corev1.PodList, error) {
 	ret := _m.Called(ctx, namespace, labelSelector)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPods")
+	}
+
 	var r0 *corev1.PodList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, *metav1.LabelSelector) (*corev1.PodList, error)); ok {
@@ -408,6 +476,10 @@ func (_m *MockKubeClientConnector) GetPods(ctx context.Context, namespace string
 func (_m *MockKubeClientConnector) GetSecret(ctx context.Context, name string) (*corev1.Secret, error) {
 	ret := _m.Called(ctx, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSecret")
+	}
+
 	var r0 *corev1.Secret
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*corev1.Secret, error)); ok {
@@ -434,6 +506,10 @@ func (_m *MockKubeClientConnector) GetSecret(ctx context.Context, name string) (
 func (_m *MockKubeClientConnector) GetSecretsForServiceAccount(ctx context.Context, accountName string) (*corev1.Secret, error) {
 	ret := _m.Called(ctx, accountName)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSecretsForServiceAccount")
+	}
+
 	var r0 *corev1.Secret
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*corev1.Secret, error)); ok {
@@ -460,6 +536,10 @@ func (_m *MockKubeClientConnector) GetSecretsForServiceAccount(ctx context.Conte
 func (_m *MockKubeClientConnector) GetServerVersion() (*version.Info, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetServerVersion")
+	}
+
 	var r0 *version.Info
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*version.Info, error)); ok {
@@ -486,6 +566,10 @@ func (_m *MockKubeClientConnector) GetServerVersion() (*version.Info, error) {
 func (_m *MockKubeClientConnector) GetStorageClasses(ctx context.Context) (*storagev1.StorageClassList, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStorageClasses")
+	}
+
 	var r0 *storagev1.StorageClassList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*storagev1.StorageClassList, error)); ok {
@@ -512,6 +596,10 @@ func (_m *MockKubeClientConnector) GetStorageClasses(ctx context.Context) (*stor
 func (_m *MockKubeClientConnector) GetSubscription(ctx context.Context, namespace string, name string) (*v1alpha1.Subscription, error) {
 	ret := _m.Called(ctx, namespace, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSubscription")
+	}
+
 	var r0 *v1alpha1.Subscription
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (*v1alpha1.Subscription, error)); ok {
@@ -538,6 +626,10 @@ func (_m *MockKubeClientConnector) GetSubscription(ctx context.Context, namespac
 func (_m *MockKubeClientConnector) GetSubscriptionCSV(ctx context.Context, subKey types.NamespacedName) (types.NamespacedName, error) {
 	ret := _m.Called(ctx, subKey)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSubscriptionCSV")
+	}
+
 	var r0 types.NamespacedName
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, types.NamespacedName) (types.NamespacedName, error)); ok {
@@ -562,6 +654,10 @@ func (_m *MockKubeClientConnector) GetSubscriptionCSV(ctx context.Context, subKe
 func (_m *MockKubeClientConnector) ListCRDs(ctx context.Context, labelSelector *metav1.LabelSelector) (*apiextensionsv1.CustomResourceDefinitionList, error) {
 	ret := _m.Called(ctx, labelSelector)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCRDs")
+	}
+
 	var r0 *apiextensionsv1.CustomResourceDefinitionList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *metav1.LabelSelector) (*apiextensionsv1.CustomResourceDefinitionList, error)); ok {
@@ -588,6 +684,10 @@ func (_m *MockKubeClientConnector) ListCRDs(ctx context.Context, labelSelector *
 func (_m *MockKubeClientConnector) ListCRs(ctx context.Context, namespace string, gvr schema.GroupVersionResource, labelSelector *metav1.LabelSelector) (*unstructured.UnstructuredList, error) {
 	ret := _m.Called(ctx, namespace, gvr, labelSelector)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListCRs")
+	}
+
 	var r0 *unstructured.UnstructuredList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, schema.GroupVersionResource, *metav1.LabelSelector) (*unstructured.UnstructuredList, error)); ok {
@@ -614,6 +714,10 @@ func (_m *MockKubeClientConnector) ListCRs(ctx context.Context, namespace string
 func (_m *MockKubeClientConnector) ListDatabaseClusters(ctx context.Context) (*apiv1.DatabaseClusterList, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListDatabaseClusters")
+	}
+
 	var r0 *apiv1.DatabaseClusterList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*apiv1.DatabaseClusterList, error)); ok {
@@ -640,6 +744,10 @@ func (_m *MockKubeClientConnector) ListDatabaseClusters(ctx context.Context) (*a
 func (_m *MockKubeClientConnector) ListSecrets(ctx context.Context) (*corev1.SecretList, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListSecrets")
+	}
+
 	var r0 *corev1.SecretList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*corev1.SecretList, error)); ok {
@@ -666,6 +774,10 @@ func (_m *MockKubeClientConnector) ListSecrets(ctx context.Context) (*corev1.Sec
 func (_m *MockKubeClientConnector) ListSubscriptions(ctx context.Context, namespace string) (*v1alpha1.SubscriptionList, error) {
 	ret := _m.Called(ctx, namespace)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListSubscriptions")
+	}
+
 	var r0 *v1alpha1.SubscriptionList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*v1alpha1.SubscriptionList, error)); ok {
@@ -692,6 +804,10 @@ func (_m *MockKubeClientConnector) ListSubscriptions(ctx context.Context, namesp
 func (_m *MockKubeClientConnector) UpdateInstallPlan(ctx context.Context, namespace string, installPlan *v1alpha1.InstallPlan) (*v1alpha1.InstallPlan, error) {
 	ret := _m.Called(ctx, namespace, installPlan)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateInstallPlan")
+	}
+
 	var r0 *v1alpha1.InstallPlan
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, *v1alpha1.InstallPlan) (*v1alpha1.InstallPlan, error)); ok {
diff --git a/managed/services/grafana/mock_aws_instance_checker_test.go b/managed/services/grafana/mock_aws_instance_checker_test.go
index 6fc2958b5c..f462d263f6 100644
--- a/managed/services/grafana/mock_aws_instance_checker_test.go
+++ b/managed/services/grafana/mock_aws_instance_checker_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package grafana
 
@@ -13,6 +13,10 @@ type mockAwsInstanceChecker struct {
 func (_m *mockAwsInstanceChecker) MustCheck() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for MustCheck")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
diff --git a/managed/services/inventory/mock_agent_service_test.go b/managed/services/inventory/mock_agent_service_test.go
index 9f84a3a905..79591e67b7 100644
--- a/managed/services/inventory/mock_agent_service_test.go
+++ b/managed/services/inventory/mock_agent_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
@@ -17,6 +17,10 @@ type mockAgentService struct {
 func (_m *mockAgentService) Logs(ctx context.Context, pmmAgentID string, agentID string, limit uint32) ([]string, uint32, error) {
 	ret := _m.Called(ctx, pmmAgentID, agentID, limit)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Logs")
+	}
+
 	var r0 []string
 	var r1 uint32
 	var r2 error
diff --git a/managed/services/inventory/mock_agents_registry_test.go b/managed/services/inventory/mock_agents_registry_test.go
index 63a6e4a727..98a66fbcdc 100644
--- a/managed/services/inventory/mock_agents_registry_test.go
+++ b/managed/services/inventory/mock_agents_registry_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
@@ -17,6 +17,10 @@ type mockAgentsRegistry struct {
 func (_m *mockAgentsRegistry) IsConnected(pmmAgentID string) bool {
 	ret := _m.Called(pmmAgentID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsConnected")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(string) bool); ok {
 		r0 = rf(pmmAgentID)
diff --git a/managed/services/inventory/mock_agents_state_updater_test.go b/managed/services/inventory/mock_agents_state_updater_test.go
index 368ec148e5..a407357807 100644
--- a/managed/services/inventory/mock_agents_state_updater_test.go
+++ b/managed/services/inventory/mock_agents_state_updater_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
diff --git a/managed/services/inventory/mock_connection_checker_test.go b/managed/services/inventory/mock_connection_checker_test.go
index d110f2d021..986c9f12ce 100644
--- a/managed/services/inventory/mock_connection_checker_test.go
+++ b/managed/services/inventory/mock_connection_checker_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
@@ -20,6 +20,10 @@ type mockConnectionChecker struct {
 func (_m *mockConnectionChecker) CheckConnectionToService(ctx context.Context, q *reform.Querier, service *models.Service, agent *models.Agent) error {
 	ret := _m.Called(ctx, q, service, agent)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckConnectionToService")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *reform.Querier, *models.Service, *models.Agent) error); ok {
 		r0 = rf(ctx, q, service, agent)
diff --git a/managed/services/inventory/mock_inventory_metrics_test.go b/managed/services/inventory/mock_inventory_metrics_test.go
index 8696dcf22c..7cbae013bb 100644
--- a/managed/services/inventory/mock_inventory_metrics_test.go
+++ b/managed/services/inventory/mock_inventory_metrics_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
@@ -17,6 +17,10 @@ type mockInventoryMetrics struct {
 func (_m *mockInventoryMetrics) GetAgentMetrics(ctx context.Context) ([]Metric, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAgentMetrics")
+	}
+
 	var r0 []Metric
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) ([]Metric, error)); ok {
@@ -43,6 +47,10 @@ func (_m *mockInventoryMetrics) GetAgentMetrics(ctx context.Context) ([]Metric,
 func (_m *mockInventoryMetrics) GetNodeMetrics(ctx context.Context) ([]Metric, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNodeMetrics")
+	}
+
 	var r0 []Metric
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) ([]Metric, error)); ok {
@@ -69,6 +77,10 @@ func (_m *mockInventoryMetrics) GetNodeMetrics(ctx context.Context) ([]Metric, e
 func (_m *mockInventoryMetrics) GetServiceMetrics(ctx context.Context) ([]Metric, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetServiceMetrics")
+	}
+
 	var r0 []Metric
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) ([]Metric, error)); ok {
diff --git a/managed/services/inventory/mock_prometheus_service_test.go b/managed/services/inventory/mock_prometheus_service_test.go
index b28b989a14..19337f99f4 100644
--- a/managed/services/inventory/mock_prometheus_service_test.go
+++ b/managed/services/inventory/mock_prometheus_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
diff --git a/managed/services/inventory/mock_service_info_broker_test.go b/managed/services/inventory/mock_service_info_broker_test.go
index b3e711ade0..bf0bc79916 100644
--- a/managed/services/inventory/mock_service_info_broker_test.go
+++ b/managed/services/inventory/mock_service_info_broker_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
@@ -20,6 +20,10 @@ type mockServiceInfoBroker struct {
 func (_m *mockServiceInfoBroker) GetInfoFromService(ctx context.Context, q *reform.Querier, service *models.Service, agent *models.Agent) error {
 	ret := _m.Called(ctx, q, service, agent)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetInfoFromService")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *reform.Querier, *models.Service, *models.Agent) error); ok {
 		r0 = rf(ctx, q, service, agent)
diff --git a/managed/services/inventory/mock_version_cache_test.go b/managed/services/inventory/mock_version_cache_test.go
index d72ad3d0d3..e363197cef 100644
--- a/managed/services/inventory/mock_version_cache_test.go
+++ b/managed/services/inventory/mock_version_cache_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package inventory
 
diff --git a/managed/services/management/alerting/mock_grafana_client_test.go b/managed/services/management/alerting/mock_grafana_client_test.go
index 8373945736..944b722e0d 100644
--- a/managed/services/management/alerting/mock_grafana_client_test.go
+++ b/managed/services/management/alerting/mock_grafana_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package alerting
 
@@ -20,6 +20,10 @@ type mockGrafanaClient struct {
 func (_m *mockGrafanaClient) CreateAlertRule(ctx context.Context, folderName string, groupName string, rule *services.Rule) error {
 	ret := _m.Called(ctx, folderName, groupName, rule)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateAlertRule")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, *services.Rule) error); ok {
 		r0 = rf(ctx, folderName, groupName, rule)
@@ -34,6 +38,10 @@ func (_m *mockGrafanaClient) CreateAlertRule(ctx context.Context, folderName str
 func (_m *mockGrafanaClient) GetDatasourceUIDByID(ctx context.Context, id int64) (string, error) {
 	ret := _m.Called(ctx, id)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDatasourceUIDByID")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) (string, error)); ok {
@@ -58,6 +66,10 @@ func (_m *mockGrafanaClient) GetDatasourceUIDByID(ctx context.Context, id int64)
 func (_m *mockGrafanaClient) GetFolderByUID(ctx context.Context, uid string) (*gapi.Folder, error) {
 	ret := _m.Called(ctx, uid)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFolderByUID")
+	}
+
 	var r0 *gapi.Folder
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*gapi.Folder, error)); ok {
diff --git a/managed/services/management/backup/mock_aws_s3_test.go b/managed/services/management/backup/mock_aws_s3_test.go
index 1247a97643..df5e9b7013 100644
--- a/managed/services/management/backup/mock_aws_s3_test.go
+++ b/managed/services/management/backup/mock_aws_s3_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -17,6 +17,10 @@ type mockAwsS3 struct {
 func (_m *mockAwsS3) BucketExists(ctx context.Context, host string, accessKey string, secretKey string, name string) (bool, error) {
 	ret := _m.Called(ctx, host, accessKey, secretKey, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for BucketExists")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string) (bool, error)); ok {
@@ -41,6 +45,10 @@ func (_m *mockAwsS3) BucketExists(ctx context.Context, host string, accessKey st
 func (_m *mockAwsS3) GetBucketLocation(ctx context.Context, host string, accessKey string, secretKey string, name string) (string, error) {
 	ret := _m.Called(ctx, host, accessKey, secretKey, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetBucketLocation")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string) (string, error)); ok {
@@ -65,6 +73,10 @@ func (_m *mockAwsS3) GetBucketLocation(ctx context.Context, host string, accessK
 func (_m *mockAwsS3) RemoveRecursive(ctx context.Context, endpoint string, accessKey string, secretKey string, bucketName string, prefix string) error {
 	ret := _m.Called(ctx, endpoint, accessKey, secretKey, bucketName, prefix)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveRecursive")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, string) error); ok {
 		r0 = rf(ctx, endpoint, accessKey, secretKey, bucketName, prefix)
diff --git a/managed/services/management/backup/mock_backup_service_test.go b/managed/services/management/backup/mock_backup_service_test.go
index 87354324b2..26dbcc8f3b 100644
--- a/managed/services/management/backup/mock_backup_service_test.go
+++ b/managed/services/management/backup/mock_backup_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -20,6 +20,10 @@ type mockBackupService struct {
 func (_m *mockBackupService) PerformBackup(ctx context.Context, params servicesbackup.PerformBackupParams) (string, error) {
 	ret := _m.Called(ctx, params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PerformBackup")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, servicesbackup.PerformBackupParams) (string, error)); ok {
@@ -44,6 +48,10 @@ func (_m *mockBackupService) PerformBackup(ctx context.Context, params servicesb
 func (_m *mockBackupService) RestoreBackup(ctx context.Context, serviceID string, artifactID string, pitrTimestamp time.Time) (string, error) {
 	ret := _m.Called(ctx, serviceID, artifactID, pitrTimestamp)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RestoreBackup")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, time.Time) (string, error)); ok {
@@ -68,6 +76,10 @@ func (_m *mockBackupService) RestoreBackup(ctx context.Context, serviceID string
 func (_m *mockBackupService) SwitchMongoPITR(ctx context.Context, serviceID string, enabled bool) error {
 	ret := _m.Called(ctx, serviceID, enabled)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SwitchMongoPITR")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, bool) error); ok {
 		r0 = rf(ctx, serviceID, enabled)
diff --git a/managed/services/management/backup/mock_pbm_pitr_service_test.go b/managed/services/management/backup/mock_pbm_pitr_service_test.go
index aa853c4746..96b8309fec 100644
--- a/managed/services/management/backup/mock_pbm_pitr_service_test.go
+++ b/managed/services/management/backup/mock_pbm_pitr_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -20,6 +20,10 @@ type mockPbmPITRService struct {
 func (_m *mockPbmPITRService) ListPITRTimeranges(ctx context.Context, locationClient servicesbackup.Storage, location *models.BackupLocation, artifact *models.Artifact) ([]servicesbackup.Timeline, error) {
 	ret := _m.Called(ctx, locationClient, location, artifact)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListPITRTimeranges")
+	}
+
 	var r0 []servicesbackup.Timeline
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, servicesbackup.Storage, *models.BackupLocation, *models.Artifact) ([]servicesbackup.Timeline, error)); ok {
diff --git a/managed/services/management/backup/mock_removal_service_test.go b/managed/services/management/backup/mock_removal_service_test.go
index 84786c689c..e165ffd4a2 100644
--- a/managed/services/management/backup/mock_removal_service_test.go
+++ b/managed/services/management/backup/mock_removal_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -17,6 +17,10 @@ type mockRemovalService struct {
 func (_m *mockRemovalService) DeleteArtifact(storage servicesbackup.Storage, artifactID string, removeFiles bool) error {
 	ret := _m.Called(storage, artifactID, removeFiles)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteArtifact")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(servicesbackup.Storage, string, bool) error); ok {
 		r0 = rf(storage, artifactID, removeFiles)
diff --git a/managed/services/management/backup/mock_schedule_service_test.go b/managed/services/management/backup/mock_schedule_service_test.go
index e85951c13f..c92dd3f981 100644
--- a/managed/services/management/backup/mock_schedule_service_test.go
+++ b/managed/services/management/backup/mock_schedule_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package backup
 
@@ -20,6 +20,10 @@ type mockScheduleService struct {
 func (_m *mockScheduleService) Add(task scheduler.Task, params scheduler.AddParams) (*models.ScheduledTask, error) {
 	ret := _m.Called(task, params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Add")
+	}
+
 	var r0 *models.ScheduledTask
 	var r1 error
 	if rf, ok := ret.Get(0).(func(scheduler.Task, scheduler.AddParams) (*models.ScheduledTask, error)); ok {
@@ -46,6 +50,10 @@ func (_m *mockScheduleService) Add(task scheduler.Task, params scheduler.AddPara
 func (_m *mockScheduleService) Remove(id string) error {
 	ret := _m.Called(id)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Remove")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(id)
@@ -65,6 +73,10 @@ func (_m *mockScheduleService) Run(ctx context.Context) {
 func (_m *mockScheduleService) Update(id string, params models.ChangeScheduledTaskParams) error {
 	ret := _m.Called(id, params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Update")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, models.ChangeScheduledTaskParams) error); ok {
 		r0 = rf(id, params)
diff --git a/managed/services/management/dbaas/mock_components_service_test.go b/managed/services/management/dbaas/mock_components_service_test.go
index 37b3f231a6..781f5084a5 100644
--- a/managed/services/management/dbaas/mock_components_service_test.go
+++ b/managed/services/management/dbaas/mock_components_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package dbaas
 
@@ -19,6 +19,10 @@ type mockComponentsService struct {
 func (_m *mockComponentsService) ChangePSMDBComponents(_a0 context.Context, _a1 *dbaasv1beta1.ChangePSMDBComponentsRequest) (*dbaasv1beta1.ChangePSMDBComponentsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ChangePSMDBComponents")
+	}
+
 	var r0 *dbaasv1beta1.ChangePSMDBComponentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *dbaasv1beta1.ChangePSMDBComponentsRequest) (*dbaasv1beta1.ChangePSMDBComponentsResponse, error)); ok {
@@ -45,6 +49,10 @@ func (_m *mockComponentsService) ChangePSMDBComponents(_a0 context.Context, _a1
 func (_m *mockComponentsService) ChangePXCComponents(_a0 context.Context, _a1 *dbaasv1beta1.ChangePXCComponentsRequest) (*dbaasv1beta1.ChangePXCComponentsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ChangePXCComponents")
+	}
+
 	var r0 *dbaasv1beta1.ChangePXCComponentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *dbaasv1beta1.ChangePXCComponentsRequest) (*dbaasv1beta1.ChangePXCComponentsResponse, error)); ok {
@@ -71,6 +79,10 @@ func (_m *mockComponentsService) ChangePXCComponents(_a0 context.Context, _a1 *d
 func (_m *mockComponentsService) CheckForOperatorUpdate(_a0 context.Context, _a1 *dbaasv1beta1.CheckForOperatorUpdateRequest) (*dbaasv1beta1.CheckForOperatorUpdateResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckForOperatorUpdate")
+	}
+
 	var r0 *dbaasv1beta1.CheckForOperatorUpdateResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *dbaasv1beta1.CheckForOperatorUpdateRequest) (*dbaasv1beta1.CheckForOperatorUpdateResponse, error)); ok {
@@ -97,6 +109,10 @@ func (_m *mockComponentsService) CheckForOperatorUpdate(_a0 context.Context, _a1
 func (_m *mockComponentsService) GetPSMDBComponents(_a0 context.Context, _a1 *dbaasv1beta1.GetPSMDBComponentsRequest) (*dbaasv1beta1.GetPSMDBComponentsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPSMDBComponents")
+	}
+
 	var r0 *dbaasv1beta1.GetPSMDBComponentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *dbaasv1beta1.GetPSMDBComponentsRequest) (*dbaasv1beta1.GetPSMDBComponentsResponse, error)); ok {
@@ -123,6 +139,10 @@ func (_m *mockComponentsService) GetPSMDBComponents(_a0 context.Context, _a1 *db
 func (_m *mockComponentsService) GetPXCComponents(_a0 context.Context, _a1 *dbaasv1beta1.GetPXCComponentsRequest) (*dbaasv1beta1.GetPXCComponentsResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPXCComponents")
+	}
+
 	var r0 *dbaasv1beta1.GetPXCComponentsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *dbaasv1beta1.GetPXCComponentsRequest) (*dbaasv1beta1.GetPXCComponentsResponse, error)); ok {
@@ -149,6 +169,10 @@ func (_m *mockComponentsService) GetPXCComponents(_a0 context.Context, _a1 *dbaa
 func (_m *mockComponentsService) InstallOperator(_a0 context.Context, _a1 *dbaasv1beta1.InstallOperatorRequest) (*dbaasv1beta1.InstallOperatorResponse, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InstallOperator")
+	}
+
 	var r0 *dbaasv1beta1.InstallOperatorResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *dbaasv1beta1.InstallOperatorRequest) (*dbaasv1beta1.InstallOperatorResponse, error)); ok {
diff --git a/managed/services/management/dbaas/mock_dbaas_client_test.go b/managed/services/management/dbaas/mock_dbaas_client_test.go
index e4f9f4162d..da1bd160f8 100644
--- a/managed/services/management/dbaas/mock_dbaas_client_test.go
+++ b/managed/services/management/dbaas/mock_dbaas_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package dbaas
 
@@ -19,6 +19,10 @@ type mockDbaasClient struct {
 func (_m *mockDbaasClient) Connect(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Connect")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
@@ -33,6 +37,10 @@ func (_m *mockDbaasClient) Connect(ctx context.Context) error {
 func (_m *mockDbaasClient) Disconnect() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Disconnect")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -54,6 +62,10 @@ func (_m *mockDbaasClient) GetKubeConfig(ctx context.Context, in *controllerv1be
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetKubeConfig")
+	}
+
 	var r0 *controllerv1beta1.GetKubeconfigResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *controllerv1beta1.GetKubeconfigRequest, ...grpc.CallOption) (*controllerv1beta1.GetKubeconfigResponse, error)); ok {
@@ -87,6 +99,10 @@ func (_m *mockDbaasClient) GetLogs(ctx context.Context, in *controllerv1beta1.Ge
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetLogs")
+	}
+
 	var r0 *controllerv1beta1.GetLogsResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *controllerv1beta1.GetLogsRequest, ...grpc.CallOption) (*controllerv1beta1.GetLogsResponse, error)); ok {
@@ -120,6 +136,10 @@ func (_m *mockDbaasClient) GetResources(ctx context.Context, in *controllerv1bet
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetResources")
+	}
+
 	var r0 *controllerv1beta1.GetResourcesResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *controllerv1beta1.GetResourcesRequest, ...grpc.CallOption) (*controllerv1beta1.GetResourcesResponse, error)); ok {
@@ -153,6 +173,10 @@ func (_m *mockDbaasClient) InstallPSMDBOperator(ctx context.Context, in *control
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InstallPSMDBOperator")
+	}
+
 	var r0 *controllerv1beta1.InstallPSMDBOperatorResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *controllerv1beta1.InstallPSMDBOperatorRequest, ...grpc.CallOption) (*controllerv1beta1.InstallPSMDBOperatorResponse, error)); ok {
@@ -186,6 +210,10 @@ func (_m *mockDbaasClient) InstallPXCOperator(ctx context.Context, in *controlle
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InstallPXCOperator")
+	}
+
 	var r0 *controllerv1beta1.InstallPXCOperatorResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *controllerv1beta1.InstallPXCOperatorRequest, ...grpc.CallOption) (*controllerv1beta1.InstallPXCOperatorResponse, error)); ok {
@@ -219,6 +247,10 @@ func (_m *mockDbaasClient) StartMonitoring(ctx context.Context, in *controllerv1
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartMonitoring")
+	}
+
 	var r0 *controllerv1beta1.StartMonitoringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *controllerv1beta1.StartMonitoringRequest, ...grpc.CallOption) (*controllerv1beta1.StartMonitoringResponse, error)); ok {
@@ -252,6 +284,10 @@ func (_m *mockDbaasClient) StopMonitoring(ctx context.Context, in *controllerv1b
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StopMonitoring")
+	}
+
 	var r0 *controllerv1beta1.StopMonitoringResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *controllerv1beta1.StopMonitoringRequest, ...grpc.CallOption) (*controllerv1beta1.StopMonitoringResponse, error)); ok {
diff --git a/managed/services/management/dbaas/mock_grafana_client_test.go b/managed/services/management/dbaas/mock_grafana_client_test.go
index 0f4e05622c..dd40fd82b2 100644
--- a/managed/services/management/dbaas/mock_grafana_client_test.go
+++ b/managed/services/management/dbaas/mock_grafana_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package dbaas
 
@@ -17,6 +17,10 @@ type mockGrafanaClient struct {
 func (_m *mockGrafanaClient) CreateAdminAPIKey(ctx context.Context, name string) (int64, string, error) {
 	ret := _m.Called(ctx, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateAdminAPIKey")
+	}
+
 	var r0 int64
 	var r1 string
 	var r2 error
@@ -48,6 +52,10 @@ func (_m *mockGrafanaClient) CreateAdminAPIKey(ctx context.Context, name string)
 func (_m *mockGrafanaClient) DeleteAPIKeyByID(ctx context.Context, id int64) error {
 	ret := _m.Called(ctx, id)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteAPIKeyByID")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, int64) error); ok {
 		r0 = rf(ctx, id)
@@ -62,6 +70,10 @@ func (_m *mockGrafanaClient) DeleteAPIKeyByID(ctx context.Context, id int64) err
 func (_m *mockGrafanaClient) DeleteAPIKeysWithPrefix(ctx context.Context, name string) error {
 	ret := _m.Called(ctx, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteAPIKeysWithPrefix")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
 		r0 = rf(ctx, name)
diff --git a/managed/services/management/dbaas/mock_kube_storage_manager_test.go b/managed/services/management/dbaas/mock_kube_storage_manager_test.go
index 82b07583e1..c5300a7f53 100644
--- a/managed/services/management/dbaas/mock_kube_storage_manager_test.go
+++ b/managed/services/management/dbaas/mock_kube_storage_manager_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package dbaas
 
@@ -13,6 +13,10 @@ type mockKubeStorageManager struct {
 func (_m *mockKubeStorageManager) DeleteClient(name string) error {
 	ret := _m.Called(name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteClient")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(name)
@@ -27,6 +31,10 @@ func (_m *mockKubeStorageManager) DeleteClient(name string) error {
 func (_m *mockKubeStorageManager) GetOrSetClient(name string) (kubernetesClient, error) {
 	ret := _m.Called(name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetOrSetClient")
+	}
+
 	var r0 kubernetesClient
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (kubernetesClient, error)); ok {
diff --git a/managed/services/management/dbaas/mock_kubernetes_client_test.go b/managed/services/management/dbaas/mock_kubernetes_client_test.go
index 8cd32a79ef..18eb140f70 100644
--- a/managed/services/management/dbaas/mock_kubernetes_client_test.go
+++ b/managed/services/management/dbaas/mock_kubernetes_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package dbaas
 
@@ -25,6 +25,10 @@ type mockKubernetesClient struct {
 func (_m *mockKubernetesClient) CreateDatabaseCluster(_a0 *v1.DatabaseCluster) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateDatabaseCluster")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*v1.DatabaseCluster) error); ok {
 		r0 = rf(_a0)
@@ -39,6 +43,10 @@ func (_m *mockKubernetesClient) CreateDatabaseCluster(_a0 *v1.DatabaseCluster) e
 func (_m *mockKubernetesClient) CreatePMMSecret(_a0 string, _a1 map[string][]byte) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreatePMMSecret")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string, map[string][]byte) error); ok {
 		r0 = rf(_a0, _a1)
@@ -53,6 +61,10 @@ func (_m *mockKubernetesClient) CreatePMMSecret(_a0 string, _a1 map[string][]byt
 func (_m *mockKubernetesClient) CreateRestore(_a0 *v1.DatabaseClusterRestore) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateRestore")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*v1.DatabaseClusterRestore) error); ok {
 		r0 = rf(_a0)
@@ -67,6 +79,10 @@ func (_m *mockKubernetesClient) CreateRestore(_a0 *v1.DatabaseClusterRestore) er
 func (_m *mockKubernetesClient) DeleteDatabaseCluster(_a0 context.Context, _a1 string) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteDatabaseCluster")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
 		r0 = rf(_a0, _a1)
@@ -81,6 +97,10 @@ func (_m *mockKubernetesClient) DeleteDatabaseCluster(_a0 context.Context, _a1 s
 func (_m *mockKubernetesClient) GetAllClusterResources(_a0 context.Context, _a1 kubernetes.ClusterType, _a2 *corev1.PersistentVolumeList) (uint64, uint64, uint64, error) {
 	ret := _m.Called(_a0, _a1, _a2)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAllClusterResources")
+	}
+
 	var r0 uint64
 	var r1 uint64
 	var r2 uint64
@@ -119,6 +139,10 @@ func (_m *mockKubernetesClient) GetAllClusterResources(_a0 context.Context, _a1
 func (_m *mockKubernetesClient) GetClusterType(_a0 context.Context) (kubernetes.ClusterType, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetClusterType")
+	}
+
 	var r0 kubernetes.ClusterType
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (kubernetes.ClusterType, error)); ok {
@@ -143,6 +167,10 @@ func (_m *mockKubernetesClient) GetClusterType(_a0 context.Context) (kubernetes.
 func (_m *mockKubernetesClient) GetConsumedCPUAndMemory(_a0 context.Context, _a1 string) (uint64, uint64, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetConsumedCPUAndMemory")
+	}
+
 	var r0 uint64
 	var r1 uint64
 	var r2 error
@@ -174,6 +202,10 @@ func (_m *mockKubernetesClient) GetConsumedCPUAndMemory(_a0 context.Context, _a1
 func (_m *mockKubernetesClient) GetConsumedDiskBytes(_a0 context.Context, _a1 kubernetes.ClusterType, _a2 *corev1.PersistentVolumeList) (uint64, error) {
 	ret := _m.Called(_a0, _a1, _a2)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetConsumedDiskBytes")
+	}
+
 	var r0 uint64
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, kubernetes.ClusterType, *corev1.PersistentVolumeList) (uint64, error)); ok {
@@ -198,6 +230,10 @@ func (_m *mockKubernetesClient) GetConsumedDiskBytes(_a0 context.Context, _a1 ku
 func (_m *mockKubernetesClient) GetDatabaseCluster(_a0 context.Context, _a1 string) (*v1.DatabaseCluster, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDatabaseCluster")
+	}
+
 	var r0 *v1.DatabaseCluster
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*v1.DatabaseCluster, error)); ok {
@@ -224,6 +260,10 @@ func (_m *mockKubernetesClient) GetDatabaseCluster(_a0 context.Context, _a1 stri
 func (_m *mockKubernetesClient) GetDefaultStorageClassName(_a0 context.Context) (string, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDefaultStorageClassName")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (string, error)); ok {
@@ -248,6 +288,10 @@ func (_m *mockKubernetesClient) GetDefaultStorageClassName(_a0 context.Context)
 func (_m *mockKubernetesClient) GetPSMDBOperatorVersion(_a0 context.Context) (string, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPSMDBOperatorVersion")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (string, error)); ok {
@@ -272,6 +316,10 @@ func (_m *mockKubernetesClient) GetPSMDBOperatorVersion(_a0 context.Context) (st
 func (_m *mockKubernetesClient) GetPXCOperatorVersion(_a0 context.Context) (string, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPXCOperatorVersion")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (string, error)); ok {
@@ -296,6 +344,10 @@ func (_m *mockKubernetesClient) GetPXCOperatorVersion(_a0 context.Context) (stri
 func (_m *mockKubernetesClient) GetPersistentVolumes(ctx context.Context) (*corev1.PersistentVolumeList, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetPersistentVolumes")
+	}
+
 	var r0 *corev1.PersistentVolumeList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*corev1.PersistentVolumeList, error)); ok {
@@ -322,6 +374,10 @@ func (_m *mockKubernetesClient) GetPersistentVolumes(ctx context.Context) (*core
 func (_m *mockKubernetesClient) GetSecret(_a0 context.Context, _a1 string) (*corev1.Secret, error) {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSecret")
+	}
+
 	var r0 *corev1.Secret
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*corev1.Secret, error)); ok {
@@ -348,6 +404,10 @@ func (_m *mockKubernetesClient) GetSecret(_a0 context.Context, _a1 string) (*cor
 func (_m *mockKubernetesClient) GetServerVersion() (*version.Info, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetServerVersion")
+	}
+
 	var r0 *version.Info
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (*version.Info, error)); ok {
@@ -374,6 +434,10 @@ func (_m *mockKubernetesClient) GetServerVersion() (*version.Info, error) {
 func (_m *mockKubernetesClient) GetStorageClasses(ctx context.Context) (*storagev1.StorageClassList, error) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetStorageClasses")
+	}
+
 	var r0 *storagev1.StorageClassList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*storagev1.StorageClassList, error)); ok {
@@ -400,6 +464,10 @@ func (_m *mockKubernetesClient) GetStorageClasses(ctx context.Context) (*storage
 func (_m *mockKubernetesClient) InstallOLMOperator(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InstallOLMOperator")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
@@ -414,6 +482,10 @@ func (_m *mockKubernetesClient) InstallOLMOperator(ctx context.Context) error {
 func (_m *mockKubernetesClient) InstallOperator(ctx context.Context, req kubernetes.InstallOperatorRequest) error {
 	ret := _m.Called(ctx, req)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InstallOperator")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, kubernetes.InstallOperatorRequest) error); ok {
 		r0 = rf(ctx, req)
@@ -428,6 +500,10 @@ func (_m *mockKubernetesClient) InstallOperator(ctx context.Context, req kuberne
 func (_m *mockKubernetesClient) ListDatabaseClusters(_a0 context.Context) (*v1.DatabaseClusterList, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListDatabaseClusters")
+	}
+
 	var r0 *v1.DatabaseClusterList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*v1.DatabaseClusterList, error)); ok {
@@ -454,6 +530,10 @@ func (_m *mockKubernetesClient) ListDatabaseClusters(_a0 context.Context) (*v1.D
 func (_m *mockKubernetesClient) ListSecrets(_a0 context.Context) (*corev1.SecretList, error) {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListSecrets")
+	}
+
 	var r0 *corev1.SecretList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context) (*corev1.SecretList, error)); ok {
@@ -480,6 +560,10 @@ func (_m *mockKubernetesClient) ListSecrets(_a0 context.Context) (*corev1.Secret
 func (_m *mockKubernetesClient) ListSubscriptions(ctx context.Context, namespace string) (*v1alpha1.SubscriptionList, error) {
 	ret := _m.Called(ctx, namespace)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListSubscriptions")
+	}
+
 	var r0 *v1alpha1.SubscriptionList
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (*v1alpha1.SubscriptionList, error)); ok {
@@ -506,6 +590,10 @@ func (_m *mockKubernetesClient) ListSubscriptions(ctx context.Context, namespace
 func (_m *mockKubernetesClient) ListTemplates(ctx context.Context, engine string, namespace string) ([]*dbaasv1beta1.Template, error) {
 	ret := _m.Called(ctx, engine, namespace)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ListTemplates")
+	}
+
 	var r0 []*dbaasv1beta1.Template
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) ([]*dbaasv1beta1.Template, error)); ok {
@@ -532,6 +620,10 @@ func (_m *mockKubernetesClient) ListTemplates(ctx context.Context, engine string
 func (_m *mockKubernetesClient) PatchDatabaseCluster(_a0 *v1.DatabaseCluster) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PatchDatabaseCluster")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*v1.DatabaseCluster) error); ok {
 		r0 = rf(_a0)
@@ -546,6 +638,10 @@ func (_m *mockKubernetesClient) PatchDatabaseCluster(_a0 *v1.DatabaseCluster) er
 func (_m *mockKubernetesClient) RestartDatabaseCluster(_a0 context.Context, _a1 string) error {
 	ret := _m.Called(_a0, _a1)
 
+	if len(ret) == 0 {
+		panic("no return value specified for RestartDatabaseCluster")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
 		r0 = rf(_a0, _a1)
@@ -560,6 +656,10 @@ func (_m *mockKubernetesClient) RestartDatabaseCluster(_a0 context.Context, _a1
 func (_m *mockKubernetesClient) SetKubeconfig(_a0 string) error {
 	ret := _m.Called(_a0)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SetKubeconfig")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(_a0)
@@ -574,6 +674,10 @@ func (_m *mockKubernetesClient) SetKubeconfig(_a0 string) error {
 func (_m *mockKubernetesClient) UpgradeOperator(ctx context.Context, namespace string, name string) error {
 	ret := _m.Called(ctx, namespace, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpgradeOperator")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) error); ok {
 		r0 = rf(ctx, namespace, name)
diff --git a/managed/services/management/dbaas/mock_version_service_test.go b/managed/services/management/dbaas/mock_version_service_test.go
index 5b6b387f8f..dff5c5d1af 100644
--- a/managed/services/management/dbaas/mock_version_service_test.go
+++ b/managed/services/management/dbaas/mock_version_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package dbaas
 
@@ -18,6 +18,10 @@ type mockVersionService struct {
 func (_m *mockVersionService) GetNextDatabaseImage(ctx context.Context, operatorType string, operatorVersion string, installedDBVersion string) (string, error) {
 	ret := _m.Called(ctx, operatorType, operatorVersion, installedDBVersion)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetNextDatabaseImage")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string) (string, error)); ok {
@@ -42,6 +46,10 @@ func (_m *mockVersionService) GetNextDatabaseImage(ctx context.Context, operator
 func (_m *mockVersionService) GetVersionServiceURL() string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetVersionServiceURL")
+	}
+
 	var r0 string
 	if rf, ok := ret.Get(0).(func() string); ok {
 		r0 = rf()
@@ -56,6 +64,10 @@ func (_m *mockVersionService) GetVersionServiceURL() string {
 func (_m *mockVersionService) IsDatabaseVersionSupportedByOperator(ctx context.Context, operatorType string, operatorVersion string, databaseVersion string) (bool, error) {
 	ret := _m.Called(ctx, operatorType, operatorVersion, databaseVersion)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsDatabaseVersionSupportedByOperator")
+	}
+
 	var r0 bool
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string, string) (bool, error)); ok {
@@ -80,6 +92,10 @@ func (_m *mockVersionService) IsDatabaseVersionSupportedByOperator(ctx context.C
 func (_m *mockVersionService) LatestOperatorVersion(ctx context.Context, pmmVersion string) (*version.Version, *version.Version, error) {
 	ret := _m.Called(ctx, pmmVersion)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LatestOperatorVersion")
+	}
+
 	var r0 *version.Version
 	var r1 *version.Version
 	var r2 error
@@ -115,6 +131,10 @@ func (_m *mockVersionService) LatestOperatorVersion(ctx context.Context, pmmVers
 func (_m *mockVersionService) Matrix(ctx context.Context, params componentsParams) (*VersionServiceResponse, error) {
 	ret := _m.Called(ctx, params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Matrix")
+	}
+
 	var r0 *VersionServiceResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, componentsParams) (*VersionServiceResponse, error)); ok {
@@ -141,6 +161,10 @@ func (_m *mockVersionService) Matrix(ctx context.Context, params componentsParam
 func (_m *mockVersionService) NextOperatorVersion(ctx context.Context, operatorType string, installedVersion string) (*version.Version, error) {
 	ret := _m.Called(ctx, operatorType, installedVersion)
 
+	if len(ret) == 0 {
+		panic("no return value specified for NextOperatorVersion")
+	}
+
 	var r0 *version.Version
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string, string) (*version.Version, error)); ok {
@@ -167,6 +191,10 @@ func (_m *mockVersionService) NextOperatorVersion(ctx context.Context, operatorT
 func (_m *mockVersionService) SupportedOperatorVersionsList(ctx context.Context, pmmVersion string) (map[string][]string, error) {
 	ret := _m.Called(ctx, pmmVersion)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SupportedOperatorVersionsList")
+	}
+
 	var r0 map[string][]string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) (map[string][]string, error)); ok {
diff --git a/managed/services/management/dump/mock_dump_service_test.go b/managed/services/management/dump/mock_dump_service_test.go
index 051bfa8db8..fedf81e426 100644
--- a/managed/services/management/dump/mock_dump_service_test.go
+++ b/managed/services/management/dump/mock_dump_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package dump
 
@@ -17,6 +17,10 @@ type mockDumpService struct {
 func (_m *mockDumpService) DeleteDump(dumpID string) error {
 	ret := _m.Called(dumpID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DeleteDump")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(dumpID)
@@ -31,6 +35,10 @@ func (_m *mockDumpService) DeleteDump(dumpID string) error {
 func (_m *mockDumpService) GetFilePathsForDumps(dumpIDs []string) (map[string]string, error) {
 	ret := _m.Called(dumpIDs)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetFilePathsForDumps")
+	}
+
 	var r0 map[string]string
 	var r1 error
 	if rf, ok := ret.Get(0).(func([]string) (map[string]string, error)); ok {
@@ -57,6 +65,10 @@ func (_m *mockDumpService) GetFilePathsForDumps(dumpIDs []string) (map[string]st
 func (_m *mockDumpService) StartDump(params *servicesdump.Params) (string, error) {
 	ret := _m.Called(params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartDump")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(*servicesdump.Params) (string, error)); ok {
diff --git a/managed/services/management/ia/mock_alert_manager_test.go b/managed/services/management/ia/mock_alert_manager_test.go
index 6860754429..69d37d4a86 100644
--- a/managed/services/management/ia/mock_alert_manager_test.go
+++ b/managed/services/management/ia/mock_alert_manager_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package ia
 
@@ -20,6 +20,10 @@ type mockAlertManager struct {
 func (_m *mockAlertManager) FindAlertsByID(ctx context.Context, params *services.FilterParams, ids []string) ([]*ammodels.GettableAlert, error) {
 	ret := _m.Called(ctx, params, ids)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FindAlertsByID")
+	}
+
 	var r0 []*ammodels.GettableAlert
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *services.FilterParams, []string) ([]*ammodels.GettableAlert, error)); ok {
@@ -46,6 +50,10 @@ func (_m *mockAlertManager) FindAlertsByID(ctx context.Context, params *services
 func (_m *mockAlertManager) GetAlerts(ctx context.Context, params *services.FilterParams) ([]*ammodels.GettableAlert, error) {
 	ret := _m.Called(ctx, params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAlerts")
+	}
+
 	var r0 []*ammodels.GettableAlert
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *services.FilterParams) ([]*ammodels.GettableAlert, error)); ok {
@@ -77,6 +85,10 @@ func (_m *mockAlertManager) RequestConfigurationUpdate() {
 func (_m *mockAlertManager) SilenceAlerts(ctx context.Context, alerts []*ammodels.GettableAlert) error {
 	ret := _m.Called(ctx, alerts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SilenceAlerts")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []*ammodels.GettableAlert) error); ok {
 		r0 = rf(ctx, alerts)
@@ -91,6 +103,10 @@ func (_m *mockAlertManager) SilenceAlerts(ctx context.Context, alerts []*ammodel
 func (_m *mockAlertManager) UnsilenceAlerts(ctx context.Context, alerts []*ammodels.GettableAlert) error {
 	ret := _m.Called(ctx, alerts)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UnsilenceAlerts")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, []*ammodels.GettableAlert) error); ok {
 		r0 = rf(ctx, alerts)
diff --git a/managed/services/management/ia/mock_templates_service_test.go b/managed/services/management/ia/mock_templates_service_test.go
index 96279fdd8c..f65fc87009 100644
--- a/managed/services/management/ia/mock_templates_service_test.go
+++ b/managed/services/management/ia/mock_templates_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package ia
 
@@ -17,6 +17,10 @@ type mockTemplatesService struct {
 func (_m *mockTemplatesService) GetTemplates() map[string]models.Template {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetTemplates")
+	}
+
 	var r0 map[string]models.Template
 	if rf, ok := ret.Get(0).(func() map[string]models.Template); ok {
 		r0 = rf()
diff --git a/managed/services/management/ia/mock_vm_alert_test.go b/managed/services/management/ia/mock_vm_alert_test.go
index 192f27e333..e4c6503d87 100644
--- a/managed/services/management/ia/mock_vm_alert_test.go
+++ b/managed/services/management/ia/mock_vm_alert_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package ia
 
diff --git a/managed/services/management/mock_agents_registry_test.go b/managed/services/management/mock_agents_registry_test.go
index 2432241b46..f236ad096c 100644
--- a/managed/services/management/mock_agents_registry_test.go
+++ b/managed/services/management/mock_agents_registry_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -17,6 +17,10 @@ type mockAgentsRegistry struct {
 func (_m *mockAgentsRegistry) IsConnected(pmmAgentID string) bool {
 	ret := _m.Called(pmmAgentID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsConnected")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(string) bool); ok {
 		r0 = rf(pmmAgentID)
diff --git a/managed/services/management/mock_agents_state_updater_test.go b/managed/services/management/mock_agents_state_updater_test.go
index a4b409e07c..25905e3945 100644
--- a/managed/services/management/mock_agents_state_updater_test.go
+++ b/managed/services/management/mock_agents_state_updater_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
diff --git a/managed/services/management/mock_api_key_provider_test.go b/managed/services/management/mock_api_key_provider_test.go
index b85f9fc04c..ef91fa18e4 100644
--- a/managed/services/management/mock_api_key_provider_test.go
+++ b/managed/services/management/mock_api_key_provider_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -18,6 +18,10 @@ type mockApiKeyProvider struct {
 func (_m *mockApiKeyProvider) CreateAdminAPIKey(ctx context.Context, name string) (int64, string, error) {
 	ret := _m.Called(ctx, name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateAdminAPIKey")
+	}
+
 	var r0 int64
 	var r1 string
 	var r2 error
@@ -49,6 +53,10 @@ func (_m *mockApiKeyProvider) CreateAdminAPIKey(ctx context.Context, name string
 func (_m *mockApiKeyProvider) IsAPIKeyAuth(headers http.Header) bool {
 	ret := _m.Called(headers)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsAPIKeyAuth")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func(http.Header) bool); ok {
 		r0 = rf(headers)
diff --git a/managed/services/management/mock_checks_service_test.go b/managed/services/management/mock_checks_service_test.go
index f6e0f6a91a..3723a54db9 100644
--- a/managed/services/management/mock_checks_service_test.go
+++ b/managed/services/management/mock_checks_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -20,6 +20,10 @@ type mockChecksService struct {
 func (_m *mockChecksService) ChangeInterval(params map[string]check.Interval) error {
 	ret := _m.Called(params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ChangeInterval")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(map[string]check.Interval) error); ok {
 		r0 = rf(params)
@@ -34,6 +38,10 @@ func (_m *mockChecksService) ChangeInterval(params map[string]check.Interval) er
 func (_m *mockChecksService) DisableChecks(checkNames []string) error {
 	ret := _m.Called(checkNames)
 
+	if len(ret) == 0 {
+		panic("no return value specified for DisableChecks")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(checkNames)
@@ -48,6 +56,10 @@ func (_m *mockChecksService) DisableChecks(checkNames []string) error {
 func (_m *mockChecksService) EnableChecks(checkNames []string) error {
 	ret := _m.Called(checkNames)
 
+	if len(ret) == 0 {
+		panic("no return value specified for EnableChecks")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(checkNames)
@@ -62,6 +74,10 @@ func (_m *mockChecksService) EnableChecks(checkNames []string) error {
 func (_m *mockChecksService) GetAdvisors() ([]check.Advisor, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetAdvisors")
+	}
+
 	var r0 []check.Advisor
 	var r1 error
 	if rf, ok := ret.Get(0).(func() ([]check.Advisor, error)); ok {
@@ -88,6 +104,10 @@ func (_m *mockChecksService) GetAdvisors() ([]check.Advisor, error) {
 func (_m *mockChecksService) GetChecks() (map[string]check.Check, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChecks")
+	}
+
 	var r0 map[string]check.Check
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (map[string]check.Check, error)); ok {
@@ -114,6 +134,10 @@ func (_m *mockChecksService) GetChecks() (map[string]check.Check, error) {
 func (_m *mockChecksService) GetChecksResults(ctx context.Context, serviceID string) ([]services.CheckResult, error) {
 	ret := _m.Called(ctx, serviceID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetChecksResults")
+	}
+
 	var r0 []services.CheckResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) ([]services.CheckResult, error)); ok {
@@ -140,6 +164,10 @@ func (_m *mockChecksService) GetChecksResults(ctx context.Context, serviceID str
 func (_m *mockChecksService) GetDisabledChecks() ([]string, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetDisabledChecks")
+	}
+
 	var r0 []string
 	var r1 error
 	if rf, ok := ret.Get(0).(func() ([]string, error)); ok {
@@ -166,6 +194,10 @@ func (_m *mockChecksService) GetDisabledChecks() ([]string, error) {
 func (_m *mockChecksService) GetSecurityCheckResults() ([]services.CheckResult, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSecurityCheckResults")
+	}
+
 	var r0 []services.CheckResult
 	var r1 error
 	if rf, ok := ret.Get(0).(func() ([]services.CheckResult, error)); ok {
@@ -192,6 +224,10 @@ func (_m *mockChecksService) GetSecurityCheckResults() ([]services.CheckResult,
 func (_m *mockChecksService) StartChecks(checkNames []string) error {
 	ret := _m.Called(checkNames)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartChecks")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(checkNames)
diff --git a/managed/services/management/mock_connection_checker_test.go b/managed/services/management/mock_connection_checker_test.go
index dd82f7775f..ed986d8f9f 100644
--- a/managed/services/management/mock_connection_checker_test.go
+++ b/managed/services/management/mock_connection_checker_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -20,6 +20,10 @@ type mockConnectionChecker struct {
 func (_m *mockConnectionChecker) CheckConnectionToService(ctx context.Context, q *reform.Querier, service *models.Service, agent *models.Agent) error {
 	ret := _m.Called(ctx, q, service, agent)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CheckConnectionToService")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *reform.Querier, *models.Service, *models.Agent) error); ok {
 		r0 = rf(ctx, q, service, agent)
diff --git a/managed/services/management/mock_grafana_client_test.go b/managed/services/management/mock_grafana_client_test.go
index c7c14095e6..3dcb1c4198 100644
--- a/managed/services/management/mock_grafana_client_test.go
+++ b/managed/services/management/mock_grafana_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -18,6 +18,10 @@ type mockGrafanaClient struct {
 func (_m *mockGrafanaClient) CreateAnnotation(_a0 context.Context, _a1 []string, _a2 time.Time, _a3 string, _a4 string) (string, error) {
 	ret := _m.Called(_a0, _a1, _a2, _a3, _a4)
 
+	if len(ret) == 0 {
+		panic("no return value specified for CreateAnnotation")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, []string, time.Time, string, string) (string, error)); ok {
diff --git a/managed/services/management/mock_jobs_service_test.go b/managed/services/management/mock_jobs_service_test.go
index 51f1f55ade..0b08f85e95 100644
--- a/managed/services/management/mock_jobs_service_test.go
+++ b/managed/services/management/mock_jobs_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -13,6 +13,10 @@ type mockJobsService struct {
 func (_m *mockJobsService) StopJob(jobID string) error {
 	ret := _m.Called(jobID)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StopJob")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(jobID)
diff --git a/managed/services/management/mock_prometheus_service_test.go b/managed/services/management/mock_prometheus_service_test.go
index 3a4447fe25..5112b1ebf7 100644
--- a/managed/services/management/mock_prometheus_service_test.go
+++ b/managed/services/management/mock_prometheus_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
diff --git a/managed/services/management/mock_service_info_broker_test.go b/managed/services/management/mock_service_info_broker_test.go
index c626e893d4..d352cae39d 100644
--- a/managed/services/management/mock_service_info_broker_test.go
+++ b/managed/services/management/mock_service_info_broker_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -20,6 +20,10 @@ type mockServiceInfoBroker struct {
 func (_m *mockServiceInfoBroker) GetInfoFromService(ctx context.Context, q *reform.Querier, service *models.Service, agent *models.Agent) error {
 	ret := _m.Called(ctx, q, service, agent)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetInfoFromService")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *reform.Querier, *models.Service, *models.Agent) error); ok {
 		r0 = rf(ctx, q, service, agent)
diff --git a/managed/services/management/mock_version_cache_test.go b/managed/services/management/mock_version_cache_test.go
index bc45f23e4f..3bd020e457 100644
--- a/managed/services/management/mock_version_cache_test.go
+++ b/managed/services/management/mock_version_cache_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
diff --git a/managed/services/management/mock_victoria_metrics_client_test.go b/managed/services/management/mock_victoria_metrics_client_test.go
index b19d56ef8b..89fa8b67b0 100644
--- a/managed/services/management/mock_victoria_metrics_client_test.go
+++ b/managed/services/management/mock_victoria_metrics_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package management
 
@@ -27,6 +27,10 @@ func (_m *mockVictoriaMetricsClient) Query(ctx context.Context, query string, ts
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Query")
+	}
+
 	var r0 model.Value
 	var r1 v1.Warnings
 	var r2 error
diff --git a/managed/services/qan/mock_qan_collector_client_test.go b/managed/services/qan/mock_qan_collector_client_test.go
index 05f0fa6fb4..0494befc64 100644
--- a/managed/services/qan/mock_qan_collector_client_test.go
+++ b/managed/services/qan/mock_qan_collector_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package qan
 
@@ -27,6 +27,10 @@ func (_m *mockQanCollectorClient) Collect(ctx context.Context, in *qanv1beta1.Co
 	_ca = append(_ca, _va...)
 	ret := _m.Called(_ca...)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Collect")
+	}
+
 	var r0 *qanv1beta1.CollectResponse
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, *qanv1beta1.CollectRequest, ...grpc.CallOption) (*qanv1beta1.CollectResponse, error)); ok {
diff --git a/managed/services/scheduler/mock_backup_service_test.go b/managed/services/scheduler/mock_backup_service_test.go
index db100056e9..4e8378e862 100644
--- a/managed/services/scheduler/mock_backup_service_test.go
+++ b/managed/services/scheduler/mock_backup_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package scheduler
 
@@ -19,6 +19,10 @@ type mockBackupService struct {
 func (_m *mockBackupService) PerformBackup(ctx context.Context, params backup.PerformBackupParams) (string, error) {
 	ret := _m.Called(ctx, params)
 
+	if len(ret) == 0 {
+		panic("no return value specified for PerformBackup")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, backup.PerformBackupParams) (string, error)); ok {
diff --git a/managed/services/server/mock_agents_state_updater_test.go b/managed/services/server/mock_agents_state_updater_test.go
index 9242bd4f73..b33b21a678 100644
--- a/managed/services/server/mock_agents_state_updater_test.go
+++ b/managed/services/server/mock_agents_state_updater_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -17,6 +17,10 @@ type mockAgentsStateUpdater struct {
 func (_m *mockAgentsStateUpdater) UpdateAgentsState(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateAgentsState")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
diff --git a/managed/services/server/mock_alertmanager_service_test.go b/managed/services/server/mock_alertmanager_service_test.go
index fe2984bd70..2e09cd797d 100644
--- a/managed/services/server/mock_alertmanager_service_test.go
+++ b/managed/services/server/mock_alertmanager_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -17,6 +17,10 @@ type mockAlertmanagerService struct {
 func (_m *mockAlertmanagerService) IsReady(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsReady")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
diff --git a/managed/services/server/mock_checks_service_test.go b/managed/services/server/mock_checks_service_test.go
index acb7e72b2a..33e24ab092 100644
--- a/managed/services/server/mock_checks_service_test.go
+++ b/managed/services/server/mock_checks_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -28,6 +28,10 @@ func (_m *mockChecksService) CollectAdvisors(ctx context.Context) {
 func (_m *mockChecksService) StartChecks(checkNames []string) error {
 	ret := _m.Called(checkNames)
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartChecks")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func([]string) error); ok {
 		r0 = rf(checkNames)
diff --git a/managed/services/server/mock_emailer_test.go b/managed/services/server/mock_emailer_test.go
index 8ac8f95bd9..c64f0d58a8 100644
--- a/managed/services/server/mock_emailer_test.go
+++ b/managed/services/server/mock_emailer_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -19,6 +19,10 @@ type mockEmailer struct {
 func (_m *mockEmailer) Send(ctx context.Context, settings *models.EmailAlertingSettings, emailTo string) error {
 	ret := _m.Called(ctx, settings, emailTo)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Send")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *models.EmailAlertingSettings, string) error); ok {
 		r0 = rf(ctx, settings, emailTo)
diff --git a/managed/services/server/mock_grafana_client_test.go b/managed/services/server/mock_grafana_client_test.go
index f7c18124f2..ea9d5f7495 100644
--- a/managed/services/server/mock_grafana_client_test.go
+++ b/managed/services/server/mock_grafana_client_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -17,6 +17,10 @@ type mockGrafanaClient struct {
 func (_m *mockGrafanaClient) IsReady(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsReady")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
diff --git a/managed/services/server/mock_prometheus_service_test.go b/managed/services/server/mock_prometheus_service_test.go
index b11015fc30..3213c9ba77 100644
--- a/managed/services/server/mock_prometheus_service_test.go
+++ b/managed/services/server/mock_prometheus_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -17,6 +17,10 @@ type mockPrometheusService struct {
 func (_m *mockPrometheusService) IsReady(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for IsReady")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
diff --git a/managed/services/server/mock_rules_service_test.go b/managed/services/server/mock_rules_service_test.go
index 8d5be04c2e..9e95e8be21 100644
--- a/managed/services/server/mock_rules_service_test.go
+++ b/managed/services/server/mock_rules_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -13,6 +13,10 @@ type mockRulesService struct {
 func (_m *mockRulesService) RemoveVMAlertRulesFiles() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveVMAlertRulesFiles")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
diff --git a/managed/services/server/mock_supervisord_service_test.go b/managed/services/server/mock_supervisord_service_test.go
index 64195d9f88..ac778ad82a 100644
--- a/managed/services/server/mock_supervisord_service_test.go
+++ b/managed/services/server/mock_supervisord_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -21,6 +21,10 @@ type mockSupervisordService struct {
 func (_m *mockSupervisordService) ForceCheckUpdates(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ForceCheckUpdates")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
@@ -35,6 +39,10 @@ func (_m *mockSupervisordService) ForceCheckUpdates(ctx context.Context) error {
 func (_m *mockSupervisordService) InstalledPMMVersion(ctx context.Context) *version.PackageInfo {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for InstalledPMMVersion")
+	}
+
 	var r0 *version.PackageInfo
 	if rf, ok := ret.Get(0).(func(context.Context) *version.PackageInfo); ok {
 		r0 = rf(ctx)
@@ -51,6 +59,10 @@ func (_m *mockSupervisordService) InstalledPMMVersion(ctx context.Context) *vers
 func (_m *mockSupervisordService) LastCheckUpdatesResult(ctx context.Context) (*version.UpdateCheckResult, time.Time) {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LastCheckUpdatesResult")
+	}
+
 	var r0 *version.UpdateCheckResult
 	var r1 time.Time
 	if rf, ok := ret.Get(0).(func(context.Context) (*version.UpdateCheckResult, time.Time)); ok {
@@ -77,6 +89,10 @@ func (_m *mockSupervisordService) LastCheckUpdatesResult(ctx context.Context) (*
 func (_m *mockSupervisordService) StartUpdate() (uint32, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for StartUpdate")
+	}
+
 	var r0 uint32
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (uint32, error)); ok {
@@ -101,6 +117,10 @@ func (_m *mockSupervisordService) StartUpdate() (uint32, error) {
 func (_m *mockSupervisordService) UpdateConfiguration(settings *models.Settings, ssoDetails *models.PerconaSSODetails) error {
 	ret := _m.Called(settings, ssoDetails)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateConfiguration")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(*models.Settings, *models.PerconaSSODetails) error); ok {
 		r0 = rf(settings, ssoDetails)
@@ -115,6 +135,10 @@ func (_m *mockSupervisordService) UpdateConfiguration(settings *models.Settings,
 func (_m *mockSupervisordService) UpdateLog(offset uint32) ([]string, uint32, error) {
 	ret := _m.Called(offset)
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateLog")
+	}
+
 	var r0 []string
 	var r1 uint32
 	var r2 error
@@ -148,6 +172,10 @@ func (_m *mockSupervisordService) UpdateLog(offset uint32) ([]string, uint32, er
 func (_m *mockSupervisordService) UpdateRunning() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for UpdateRunning")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
diff --git a/managed/services/server/mock_telemetry_service_test.go b/managed/services/server/mock_telemetry_service_test.go
index e8d72eb101..6b9ada9ad1 100644
--- a/managed/services/server/mock_telemetry_service_test.go
+++ b/managed/services/server/mock_telemetry_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -17,6 +17,10 @@ type mockTelemetryService struct {
 func (_m *mockTelemetryService) DistributionMethod() serverpb.DistributionMethod {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for DistributionMethod")
+	}
+
 	var r0 serverpb.DistributionMethod
 	if rf, ok := ret.Get(0).(func() serverpb.DistributionMethod); ok {
 		r0 = rf()
@@ -31,6 +35,10 @@ func (_m *mockTelemetryService) DistributionMethod() serverpb.DistributionMethod
 func (_m *mockTelemetryService) GetSummaries() []string {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetSummaries")
+	}
+
 	var r0 []string
 	if rf, ok := ret.Get(0).(func() []string); ok {
 		r0 = rf()
diff --git a/managed/services/server/mock_templates_service_test.go b/managed/services/server/mock_templates_service_test.go
index f9625aa42e..14b1abbaf9 100644
--- a/managed/services/server/mock_templates_service_test.go
+++ b/managed/services/server/mock_templates_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
diff --git a/managed/services/server/mock_vm_alert_external_rules_test.go b/managed/services/server/mock_vm_alert_external_rules_test.go
index 1c4d605dec..df5056dd65 100644
--- a/managed/services/server/mock_vm_alert_external_rules_test.go
+++ b/managed/services/server/mock_vm_alert_external_rules_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package server
 
@@ -17,6 +17,10 @@ type mockVmAlertExternalRules struct {
 func (_m *mockVmAlertExternalRules) ReadRules() (string, error) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for ReadRules")
+	}
+
 	var r0 string
 	var r1 error
 	if rf, ok := ret.Get(0).(func() (string, error)); ok {
@@ -41,6 +45,10 @@ func (_m *mockVmAlertExternalRules) ReadRules() (string, error) {
 func (_m *mockVmAlertExternalRules) RemoveRulesFile() error {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for RemoveRulesFile")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func() error); ok {
 		r0 = rf()
@@ -55,6 +63,10 @@ func (_m *mockVmAlertExternalRules) RemoveRulesFile() error {
 func (_m *mockVmAlertExternalRules) ValidateRules(ctx context.Context, rules string) error {
 	ret := _m.Called(ctx, rules)
 
+	if len(ret) == 0 {
+		panic("no return value specified for ValidateRules")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, string) error); ok {
 		r0 = rf(ctx, rules)
@@ -69,6 +81,10 @@ func (_m *mockVmAlertExternalRules) ValidateRules(ctx context.Context, rules str
 func (_m *mockVmAlertExternalRules) WriteRules(rules string) error {
 	ret := _m.Called(rules)
 
+	if len(ret) == 0 {
+		panic("no return value specified for WriteRules")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(string) error); ok {
 		r0 = rf(rules)
diff --git a/managed/services/telemetry/mock_data_source_locator_test.go b/managed/services/telemetry/mock_data_source_locator_test.go
index 4d4acc6924..e55479816b 100644
--- a/managed/services/telemetry/mock_data_source_locator_test.go
+++ b/managed/services/telemetry/mock_data_source_locator_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package telemetry
 
@@ -13,6 +13,10 @@ type MockDataSourceLocator struct {
 func (_m *MockDataSourceLocator) LocateTelemetryDataSource(name string) (DataSource, error) {
 	ret := _m.Called(name)
 
+	if len(ret) == 0 {
+		panic("no return value specified for LocateTelemetryDataSource")
+	}
+
 	var r0 DataSource
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string) (DataSource, error)); ok {
diff --git a/managed/services/telemetry/mock_data_source_test.go b/managed/services/telemetry/mock_data_source_test.go
index 6d6fad1407..6c7facc9e5 100644
--- a/managed/services/telemetry/mock_data_source_test.go
+++ b/managed/services/telemetry/mock_data_source_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package telemetry
 
@@ -18,6 +18,10 @@ type MockDataSource struct {
 func (_m *MockDataSource) Dispose(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Dispose")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
@@ -32,6 +36,10 @@ func (_m *MockDataSource) Dispose(ctx context.Context) error {
 func (_m *MockDataSource) Enabled() bool {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for Enabled")
+	}
+
 	var r0 bool
 	if rf, ok := ret.Get(0).(func() bool); ok {
 		r0 = rf()
@@ -46,6 +54,10 @@ func (_m *MockDataSource) Enabled() bool {
 func (_m *MockDataSource) FetchMetrics(ctx context.Context, config Config) ([]*pmmv1.ServerMetric_Metric, error) {
 	ret := _m.Called(ctx, config)
 
+	if len(ret) == 0 {
+		panic("no return value specified for FetchMetrics")
+	}
+
 	var r0 []*pmmv1.ServerMetric_Metric
 	var r1 error
 	if rf, ok := ret.Get(0).(func(context.Context, Config) ([]*pmmv1.ServerMetric_Metric, error)); ok {
@@ -72,6 +84,10 @@ func (_m *MockDataSource) FetchMetrics(ctx context.Context, config Config) ([]*p
 func (_m *MockDataSource) Init(ctx context.Context) error {
 	ret := _m.Called(ctx)
 
+	if len(ret) == 0 {
+		panic("no return value specified for Init")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context) error); ok {
 		r0 = rf(ctx)
diff --git a/managed/services/telemetry/mock_distribution_util_service_test.go b/managed/services/telemetry/mock_distribution_util_service_test.go
index fcc2a75f9b..09c703053a 100644
--- a/managed/services/telemetry/mock_distribution_util_service_test.go
+++ b/managed/services/telemetry/mock_distribution_util_service_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package telemetry
 
@@ -18,6 +18,10 @@ type mockDistributionUtilService struct {
 func (_m *mockDistributionUtilService) getDistributionMethodAndOS() (serverpb.DistributionMethod, pmmv1.DistributionMethod, string) {
 	ret := _m.Called()
 
+	if len(ret) == 0 {
+		panic("no return value specified for getDistributionMethodAndOS")
+	}
+
 	var r0 serverpb.DistributionMethod
 	var r1 pmmv1.DistributionMethod
 	var r2 string
diff --git a/managed/services/telemetry/mock_sender_test.go b/managed/services/telemetry/mock_sender_test.go
index 21dc086eb3..f6c36c843e 100644
--- a/managed/services/telemetry/mock_sender_test.go
+++ b/managed/services/telemetry/mock_sender_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package telemetry
 
@@ -18,6 +18,10 @@ type mockSender struct {
 func (_m *mockSender) SendTelemetry(ctx context.Context, report *reporterv1.ReportRequest) error {
 	ret := _m.Called(ctx, report)
 
+	if len(ret) == 0 {
+		panic("no return value specified for SendTelemetry")
+	}
+
 	var r0 error
 	if rf, ok := ret.Get(0).(func(context.Context, *reporterv1.ReportRequest) error); ok {
 		r0 = rf(ctx, report)
diff --git a/managed/services/versioncache/mock_versioner_test.go b/managed/services/versioncache/mock_versioner_test.go
index b1f487374c..02a3ad6b98 100644
--- a/managed/services/versioncache/mock_versioner_test.go
+++ b/managed/services/versioncache/mock_versioner_test.go
@@ -1,4 +1,4 @@
-// Code generated by mockery v2.36.0. DO NOT EDIT.
+// Code generated by mockery v2.39.1. DO NOT EDIT.
 
 package versioncache
 
@@ -17,6 +17,10 @@ type MockVersioner struct {
 func (_m *MockVersioner) GetVersions(pmmAgentID string, softwares []agents.Software) ([]agents.Version, error) {
 	ret := _m.Called(pmmAgentID, softwares)
 
+	if len(ret) == 0 {
+		panic("no return value specified for GetVersions")
+	}
+
 	var r0 []agents.Version
 	var r1 error
 	if rf, ok := ret.Get(0).(func(string, []agents.Software) ([]agents.Version, error)); ok {