mock

type AuditLogService struct {
	ListFn        func(context.Context, *ps.ListAuditLogsRequest, ...ps.ListOption) (*ps.CursorPaginatedResponse[*ps.AuditLog], error)
	ListFnInvoked bool
}

type BackupPoliciesService struct {
	ListFn        func(context.Context, *ps.ListBackupPoliciesRequest, ...ps.ListOption) ([]*ps.BackupPolicy, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetBackupPolicyRequest) (*ps.BackupPolicy, error)
	GetFnInvoked bool

	CreateFn        func(context.Context, *ps.CreateBackupPolicyRequest) (*ps.BackupPolicy, error)
	CreateFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateBackupPolicyRequest) (*ps.BackupPolicy, error)
	UpdateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeleteBackupPolicyRequest) error
	DeleteFnInvoked bool
}

type BackupsService struct {
	CreateFn        func(context.Context, *ps.CreateBackupRequest) (*ps.Backup, error)
	CreateFnInvoked bool

	GetFn        func(context.Context, *ps.GetBackupRequest) (*ps.Backup, error)
	GetFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateBackupRequest) (*ps.Backup, error)
	UpdateFnInvoked bool

	ListFn        func(context.Context, *ps.ListBackupsRequest) ([]*ps.Backup, error)
	ListFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeleteBackupRequest) error
	DeleteFnInvoked bool
}

type BillingPaymentMethodSetupsService struct {
	CreateFn        func(context.Context, *ps.CreateBillingPaymentMethodSetupRequest) (*ps.BillingPaymentMethodSetup, error)
	CreateFnInvoked bool
	GetFn           func(context.Context, *ps.GetBillingPaymentMethodSetupRequest) (*ps.BillingPaymentMethodSetup, error)
	GetFnInvoked    bool
}

type BillingPaymentMethodsService struct {
	GetFn           func(context.Context, *ps.GetBillingPaymentMethodRequest) (*ps.BillingPaymentMethod, error)
	GetFnInvoked    bool
	DeleteFn        func(context.Context, *ps.DeleteBillingPaymentMethodRequest) error
	DeleteFnInvoked bool
}

type BranchMaintenanceService struct {
	RunFn        func(context.Context, *ps.RunBranchMaintenanceRequest) error
	RunFnInvoked bool
}

type DataImportsService struct {
	TestDataImportSourceFn          func(ctx context.Context, request *ps.TestDataImportSourceRequest) (*ps.TestDataImportSourceResponse, error)
	TestDataImportSourceFnInvoked   bool
	StartDataImportFn               func(ctx context.Context, request *ps.StartDataImportRequest) (*ps.DataImport, error)
	StartDataImportFnInvoked        bool
	CancelDataImportFn              func(ctx context.Context, request *ps.CancelDataImportRequest) error
	CancelDataImportFnInvoked       bool
	GetDataImportStatusFn           func(ctx context.Context, request *ps.GetImportStatusRequest) (*ps.DataImport, error)
	GetDataImportStatusFnInvoked    bool
	MakePlanetScalePrimaryFn        func(ctx context.Context, request *ps.MakePlanetScalePrimaryRequest) (*ps.DataImport, error)
	MakePlanetScalePrimaryFnInvoked bool
	MakePlanetScaleReplicaFn        func(ctx context.Context, request *ps.MakePlanetScaleReplicaRequest) (*ps.DataImport, error)
	MakePlanetScaleReplicaFnInvoked bool
	DetachExternalDatabaseFn        func(ctx context.Context, request *ps.DetachExternalDatabaseRequest) (*ps.DataImport, error)
	DetachExternalDatabaseFnInvoked bool
}

type DatabaseBranchesService struct {
	CreateFn        func(context.Context, *ps.CreateDatabaseBranchRequest) (*ps.DatabaseBranch, error)
	CreateFnInvoked bool

	ListFn        func(context.Context, *ps.ListDatabaseBranchesRequest, ...ps.ListOption) ([]*ps.DatabaseBranch, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetDatabaseBranchRequest) (*ps.DatabaseBranch, error)
	GetFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateDatabaseBranchRequest) (*ps.DatabaseBranch, error)
	UpdateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeleteDatabaseBranchRequest) error
	DeleteFnInvoked bool

	DiffFn        func(context.Context, *ps.DiffBranchRequest) ([]*ps.Diff, error)
	DiffFnInvoked bool

	SchemaFn        func(context.Context, *ps.BranchSchemaRequest) ([]*ps.Diff, error)
	SchemaFnInvoked bool

	RoutingRulesFn        func(context.Context, *ps.BranchRoutingRulesRequest) (*ps.RoutingRules, error)
	RoutingRulesFnInvoked bool

	UpdateRoutingRulesFn        func(context.Context, *ps.UpdateBranchRoutingRulesRequest) (*ps.RoutingRules, error)
	UpdateRoutingRulesFnInvoked bool

	DataTopologyFn        func(context.Context, *ps.BranchDataTopologyRequest) (*ps.DataTopology, error)
	DataTopologyFnInvoked bool

	UpdateDataTopologyFn        func(context.Context, *ps.UpdateBranchDataTopologyRequest) (*ps.DataTopology, error)
	UpdateDataTopologyFnInvoked bool

	RefreshSchemaFn        func(context.Context, *ps.RefreshSchemaRequest) error
	RefreshSchemaFnInvoked bool

	DemoteFn        func(context.Context, *ps.DemoteRequest) (*ps.DatabaseBranch, error)
	DemoteFnInvoked bool

	EnableSafeMigrationsFn        func(context.Context, *ps.EnableSafeMigrationsRequest) (*ps.DatabaseBranch, error)
	EnableSafeMigrationsFnInvoked bool

	DisableSafeMigrationsFn        func(context.Context, *ps.DisableSafeMigrationsRequest) (*ps.DatabaseBranch, error)
	DisableSafeMigrationsFnInvoked bool

	PromoteFn        func(context.Context, *ps.PromoteRequest) (*ps.DatabaseBranch, error)
	PromoteFnInvoked bool

	LintSchemaFn        func(context.Context, *ps.LintSchemaRequest) ([]*ps.SchemaLintError, error)
	LintSchemaFnInvoked bool

	ListClusterSKUsFn        func(context.Context, *ps.ListBranchClusterSKUsRequest, ...ps.ListOption) ([]*ps.ClusterSKU, error)
	ListClusterSKUsFnInvoked bool

	ResizeFn        func(context.Context, *ps.ResizeBranchRequest) (*ps.BranchResizeRequest, error)
	ResizeFnInvoked bool

	ListResizesFn        func(context.Context, *ps.ListBranchResizesRequest) ([]*ps.BranchResizeRequest, error)
	ListResizesFnInvoked bool

	CancelResizeFn        func(context.Context, *ps.CancelBranchResizeRequest) error
	CancelResizeFnInvoked bool

	ResizeStatusFn        func(context.Context, *ps.BranchResizeStatusRequest) (*ps.BranchResizeRequest, error)
	ResizeStatusFnInvoked bool
}

type DatabaseService struct {
	CreateFn        func(context.Context, *ps.CreateDatabaseRequest) (*ps.Database, error)
	CreateFnInvoked bool

	GetFn        func(context.Context, *ps.GetDatabaseRequest) (*ps.Database, error)
	GetFnInvoked bool

	ListFn        func(context.Context, *ps.ListDatabasesRequest, ...ps.ListOption) ([]*ps.Database, error)
	ListFnInvoked bool

	ListRegionsFn        func(context.Context, *ps.ListDatabaseRegionsRequest, ...ps.ListOption) ([]*ps.Region, error)
	ListRegionsFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeleteDatabaseRequest) (*ps.DatabaseDeletionRequest, error)
	DeleteFnInvoked bool

	UpdateSettingsFn        func(context.Context, *ps.UpdateDatabaseSettingsRequest) (*ps.Database, error)
	UpdateSettingsFnInvoked bool

	GetThrottlerFn        func(context.Context, *ps.GetDatabaseThrottlerRequest) (*ps.DatabaseThrottler, error)
	GetThrottlerFnInvoked bool

	UpdateThrottlerFn        func(context.Context, *ps.UpdateDatabaseThrottlerRequest) (*ps.DatabaseThrottler, error)
	UpdateThrottlerFnInvoked bool

	GetAggressiveCutoverFn        func(context.Context, *ps.AggressiveCutoverRequest) (*ps.AggressiveCutover, error)
	GetAggressiveCutoverFnInvoked bool

	EnableAggressiveCutoverFn        func(context.Context, *ps.AggressiveCutoverRequest) (*ps.AggressiveCutover, error)
	EnableAggressiveCutoverFnInvoked bool

	DisableAggressiveCutoverFn        func(context.Context, *ps.AggressiveCutoverRequest) (*ps.AggressiveCutover, error)
	DisableAggressiveCutoverFnInvoked bool
}

type DeployRequestsService struct {
	ApplyFn        func(context.Context, *ps.ApplyDeployRequestRequest) (*ps.DeployRequest, error)
	ApplyFnInvoked bool

	ForceCutoverFn        func(context.Context, *ps.ForceCutoverDeployRequestRequest) (*ps.DeployRequest, error)
	ForceCutoverFnInvoked bool

	UnblockFn        func(context.Context, *ps.UnblockDeployRequestRequest) (*ps.DeployRequest, error)
	UnblockFnInvoked bool

	AutoApplyFn        func(context.Context, *ps.AutoApplyDeployRequestRequest) (*ps.DeployRequest, error)
	AutoApplyFnInvoked bool

	AutoDeleteBranchFn        func(context.Context, *ps.AutoDeleteBranchRequest) (*ps.DeployRequest, error)
	AutoDeleteBranchFnInvoked bool

	CancelFn        func(context.Context, *ps.CancelDeployRequestRequest) (*ps.DeployRequest, error)
	CancelFnInvoked bool

	CloseFn        func(context.Context, *ps.CloseDeployRequestRequest) (*ps.DeployRequest, error)
	CloseFnInvoked bool

	CreateFn        func(context.Context, *ps.CreateDeployRequestRequest) (*ps.DeployRequest, error)
	CreateFnInvoked bool

	CreateReviewFn        func(context.Context, *ps.ReviewDeployRequestRequest) (*ps.DeployRequestReview, error)
	CreateReviewFnInvoked bool

	DeployFn        func(context.Context, *ps.PerformDeployRequest) (*ps.DeployRequest, error)
	DeployFnInvoked bool

	DiffFn        func(context.Context, *ps.DiffRequest) ([]*ps.Diff, error)
	DiffFnInvoked bool

	GetFn        func(context.Context, *ps.GetDeployRequestRequest) (*ps.DeployRequest, error)
	GetFnInvoked bool

	ListFn        func(context.Context, *ps.ListDeployRequestsRequest) ([]*ps.DeployRequest, error)
	ListFnInvoked bool

	RevertDeployFn        func(context.Context, *ps.RevertDeployRequestRequest) (*ps.DeployRequest, error)
	RevertDeployFnInvoked bool

	SkipRevertDeployFn        func(context.Context, *ps.SkipRevertDeployRequestRequest) (*ps.DeployRequest, error)
	SkipRevertDeployFnInvoked bool

	AutoApplyDeployFn        func(context.Context, *ps.AutoApplyDeployRequestRequest) (*ps.DeployRequest, error)
	AutoApplyDeployFnInvoked bool

	GetDeployOperationsFn        func(context.Context, *ps.GetDeployOperationsRequest) ([]*ps.DeployOperation, error)
	GetDeployOperationsFnInvoked bool

	GetDeployQueueFn        func(context.Context, *ps.GetDeployQueueRequest) ([]*ps.Deployment, error)
	GetDeployQueueFnInvoked bool

	GetDeploymentFn        func(context.Context, *ps.GetDeploymentRequest) (*ps.Deployment, error)
	GetDeploymentFnInvoked bool

	ListReviewsFn        func(context.Context, *ps.ListDeployRequestReviewsRequest) ([]*ps.DeployRequestReview, error)
	ListReviewsFnInvoked bool

	CheckStorageFn        func(context.Context, *ps.CheckDeployRequestStorageRequest) (*ps.DeployRequestStorageCheck, error)
	CheckStorageFnInvoked bool

	GetThrottlerFn        func(context.Context, *ps.GetDeployRequestThrottlerRequest) (*ps.DeployRequestThrottler, error)
	GetThrottlerFnInvoked bool

	UpdateThrottlerFn        func(context.Context, *ps.UpdateDeployRequestThrottlerRequest) (*ps.DeployRequestThrottler, error)
	UpdateThrottlerFnInvoked bool
}

type InvoicesService struct {
	ListFn                 func(context.Context, *ps.ListInvoicesRequest, ...ps.ListOption) (*ps.InvoicePage, error)
	ListFnInvoked          bool
	GetFn                  func(context.Context, *ps.GetInvoiceRequest) (*ps.Invoice, error)
	GetFnInvoked           bool
	ListLineItemsFn        func(context.Context, *ps.ListInvoiceLineItemsRequest, ...ps.ListOption) (*ps.InvoiceLineItemPage, error)
	ListLineItemsFnInvoked bool
}

type KeyspacesService struct {
	ListFn        func(context.Context, *ps.ListKeyspacesRequest) ([]*ps.Keyspace, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetKeyspaceRequest) (*ps.Keyspace, error)
	GetFnInvoked bool

	UpdateReadOnlyRegionsFn        func(context.Context, *ps.UpdateReadOnlyRegionsRequest) ([]*ps.ReadOnlyRegionKeyspace, error)
	UpdateReadOnlyRegionsFnInvoked bool

	VSchemaFn        func(context.Context, *ps.GetKeyspaceVSchemaRequest) (*ps.VSchema, error)
	VSchemaFnInvoked bool

	UpdateVSchemaFn        func(context.Context, *ps.UpdateKeyspaceVSchemaRequest) (*ps.VSchema, error)
	UpdateVSchemaFnInvoked bool

	CreateFn        func(context.Context, *ps.CreateKeyspaceRequest) (*ps.Keyspace, error)
	CreateFnInvoked bool

	CreateExternalFn        func(context.Context, *ps.CreateExternalKeyspaceRequest) (*ps.Keyspace, error)
	CreateExternalFnInvoked bool

	LintExternalFn        func(context.Context, *ps.LintExternalKeyspaceRequest) (*ps.LintExternalKeyspaceResponse, error)
	LintExternalFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeleteKeyspaceRequest) error
	DeleteFnInvoked bool

	ResizeFn        func(context.Context, *ps.ResizeKeyspaceRequest) (*ps.KeyspaceResizeRequest, error)
	ResizeFnInvoked bool

	CancelResizeFn        func(context.Context, *ps.CancelKeyspaceResizeRequest) error
	CancelResizeFnInvoked bool

	ResizeStatusFn        func(context.Context, *ps.KeyspaceResizeStatusRequest) (*ps.KeyspaceResizeRequest, error)
	ResizeStatusFnInvoked bool

	RolloutStatusFn        func(context.Context, *ps.KeyspaceRolloutStatusRequest) (*ps.KeyspaceRollout, error)
	RolloutStatusFnInvoked bool

	UpdateSettingsFn        func(context.Context, *ps.UpdateKeyspaceSettingsRequest) (*ps.Keyspace, error)
	UpdateSettingsFnInvoked bool
}

type LookupVindexService struct {
	CreateFn        func(context.Context, *ps.LookupVindexCreateRequest) (json.RawMessage, error)
	CreateFnInvoked bool

	ShowFn        func(context.Context, *ps.LookupVindexShowRequest) (json.RawMessage, error)
	ShowFnInvoked bool

	ExternalizeFn        func(context.Context, *ps.LookupVindexExternalizeRequest) (json.RawMessage, error)
	ExternalizeFnInvoked bool

	InternalizeFn        func(context.Context, *ps.LookupVindexInternalizeRequest) (json.RawMessage, error)
	InternalizeFnInvoked bool

	CancelFn        func(context.Context, *ps.LookupVindexCancelRequest) (json.RawMessage, error)
	CancelFnInvoked bool

	CompleteFn        func(context.Context, *ps.LookupVindexCompleteRequest) (json.RawMessage, error)
	CompleteFnInvoked bool
}

type MaintenanceSchedulesService struct {
	ListFn        func(context.Context, *ps.ListMaintenanceSchedulesRequest, ...ps.ListOption) ([]*ps.MaintenanceSchedule, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetMaintenanceScheduleRequest) (*ps.MaintenanceSchedule, error)
	GetFnInvoked bool

	ListWindowsFn        func(context.Context, *ps.ListMaintenanceWindowsRequest, ...ps.ListOption) ([]*ps.MaintenanceWindow, error)
	ListWindowsFnInvoked bool
}

type MaterializeService struct {
	CreateFn        func(context.Context, *ps.MaterializeCreateRequest) (json.RawMessage, error)
	CreateFnInvoked bool

	ShowFn        func(context.Context, *ps.MaterializeShowRequest) (json.RawMessage, error)
	ShowFnInvoked bool

	StartFn        func(context.Context, *ps.MaterializeStartRequest) (json.RawMessage, error)
	StartFnInvoked bool

	StopFn        func(context.Context, *ps.MaterializeStopRequest) (json.RawMessage, error)
	StopFnInvoked bool

	CancelFn        func(context.Context, *ps.MaterializeCancelRequest) (json.RawMessage, error)
	CancelFnInvoked bool
}

type MetricsService struct {
	GetSeriesFn        func(context.Context, *ps.GetMetricSeriesRequest) (*ps.MetricSeries, error)
	GetSeriesFnInvoked bool

	GetInstantFn        func(context.Context, *ps.GetInstantMetricsRequest) (*ps.InstantMetrics, error)
	GetInstantFnInvoked bool

	GetQuerySeriesFn        func(context.Context, *ps.GetQueryMetricSeriesRequest) (*ps.MetricSeries, error)
	GetQuerySeriesFnInvoked bool

	GetTablesFn        func(context.Context, *ps.GetBranchMetricsRequest) (json.RawMessage, error)
	GetTablesFnInvoked bool

	GetKeyspaceTablesFn        func(context.Context, *ps.GetBranchMetricsRequest) (json.RawMessage, error)
	GetKeyspaceTablesFnInvoked bool

	GetTabletSeriesFn        func(context.Context, *ps.GetTabletMetricSeriesRequest) (*ps.MetricSeries, error)
	GetTabletSeriesFnInvoked bool

	GetInstantTabletsFn        func(context.Context, *ps.GetInstantTabletMetricsRequest) (*ps.InstantMetrics, error)
	GetInstantTabletsFnInvoked bool

	GetTagSeriesFn        func(context.Context, *ps.GetTagMetricSeriesRequest) (*ps.MetricSeries, error)
	GetTagSeriesFnInvoked bool
}

type MoveTablesService struct {
	ListFn        func(context.Context, *ps.MoveTablesListRequest) (json.RawMessage, error)
	ListFnInvoked bool

	CreateFn        func(context.Context, *ps.MoveTablesCreateRequest) (*ps.VtctldOperationReference, error)
	CreateFnInvoked bool

	ShowFn        func(context.Context, *ps.MoveTablesShowRequest) (json.RawMessage, error)
	ShowFnInvoked bool

	StatusFn        func(context.Context, *ps.MoveTablesStatusRequest) (json.RawMessage, error)
	StatusFnInvoked bool

	SwitchTrafficFn        func(context.Context, *ps.MoveTablesSwitchTrafficRequest) (*ps.VtctldOperationReference, error)
	SwitchTrafficFnInvoked bool

	ReverseTrafficFn        func(context.Context, *ps.MoveTablesReverseTrafficRequest) (*ps.VtctldOperationReference, error)
	ReverseTrafficFnInvoked bool

	CancelFn        func(context.Context, *ps.MoveTablesCancelRequest) (*ps.VtctldOperationReference, error)
	CancelFnInvoked bool

	CompleteFn        func(context.Context, *ps.MoveTablesCompleteRequest) (*ps.VtctldOperationReference, error)
	CompleteFnInvoked bool
}

type NekiAdminsService struct {
	GetFn        func(context.Context, *ps.GetNekiAdminRequest) (*ps.NekiAdmin, error)
	GetFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateNekiAdminRequest) (*ps.NekiAdmin, error)
	UpdateFnInvoked bool

	ListParametersFn        func(context.Context, *ps.ListNekiAdminParametersRequest) ([]*ps.NekiParameter, error)
	ListParametersFnInvoked bool

	ListSizeSKUsFn        func(context.Context, *ps.ListNekiAdminSizeSKUsRequest) ([]*ps.NekiAdminSKU, error)
	ListSizeSKUsFnInvoked bool

	ListChangesFn        func(context.Context, *ps.ListNekiAdminChangesRequest) ([]*ps.NekiAdminChange, error)
	ListChangesFnInvoked bool

	GetChangeFn        func(context.Context, *ps.GetNekiAdminChangeRequest) (*ps.NekiAdminChange, error)
	GetChangeFnInvoked bool

	CancelChangeFn        func(context.Context, *ps.CancelNekiAdminChangeRequest) error
	CancelChangeFnInvoked bool
}

type NekiChangesService struct {
	ListFn        func(context.Context, *ps.ListNekiChangesRequest) ([]*ps.NekiChange, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetNekiChangeRequest) (*ps.NekiChange, error)
	GetFnInvoked bool

	CancelFn        func(context.Context, *ps.CancelNekiChangeRequest) error
	CancelFnInvoked bool
}

type NekiRoutersService struct {
	ListFn        func(context.Context, *ps.ListNekiRoutersRequest) ([]*ps.NekiRouter, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetNekiRouterRequest) (*ps.NekiRouter, error)
	GetFnInvoked bool

	CreateFn        func(context.Context, *ps.CreateNekiRouterRequest) (*ps.NekiRouter, error)
	CreateFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateNekiRouterRequest) (*ps.NekiRouter, error)
	UpdateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeleteNekiRouterRequest) error
	DeleteFnInvoked bool

	ListParametersFn        func(context.Context, *ps.ListNekiRouterParametersRequest) ([]*ps.NekiParameter, error)
	ListParametersFnInvoked bool

	ListSizeSKUsFn        func(context.Context, *ps.ListNekiRouterSizeSKUsRequest) ([]*ps.NekiRouterSKU, error)
	ListSizeSKUsFnInvoked bool

	ListChangesFn        func(context.Context, *ps.ListNekiRouterChangesRequest) ([]*ps.NekiRouterChange, error)
	ListChangesFnInvoked bool

	GetChangeFn        func(context.Context, *ps.GetNekiRouterChangeRequest) (*ps.NekiRouterChange, error)
	GetChangeFnInvoked bool

	CancelChangeFn        func(context.Context, *ps.CancelNekiRouterChangeRequest) error
	CancelChangeFnInvoked bool
}

type NekiShardConfigurationProfilesService struct {
	ListFn               func(context.Context, *ps.ListNekiShardConfigurationProfilesRequest) ([]*ps.NekiShardConfigurationProfile, error)
	GetFn                func(context.Context, *ps.GetNekiShardConfigurationProfileRequest) (*ps.NekiShardConfigurationProfile, error)
	CreateFn             func(context.Context, *ps.CreateNekiShardConfigurationProfileRequest) (*ps.NekiShardConfigurationProfile, error)
	UpdateFn             func(context.Context, *ps.UpdateNekiShardConfigurationProfileRequest) (*ps.NekiShardConfigurationProfile, error)
	DeleteFn             func(context.Context, *ps.DeleteNekiShardConfigurationProfileRequest) error
	GetDefaultFn         func(context.Context, *ps.GetDefaultNekiShardConfigurationProfileRequest) (*ps.NekiShardConfigurationProfile, error)
	SetDefaultFn         func(context.Context, *ps.SetDefaultNekiShardConfigurationProfileRequest) (*ps.NekiShardConfigurationProfile, error)
	ListParametersFn     func(context.Context, *ps.ListNekiShardConfigurationProfileParametersRequest) ([]*ps.NekiParameter, error)
	ListExtensionsFn     func(context.Context, *ps.ListNekiShardConfigurationProfileExtensionsRequest) ([]*ps.NekiExtension, error)
	UpdateExtensionFn    func(context.Context, *ps.UpdateNekiShardConfigurationProfileExtensionRequest) (*ps.NekiExtension, error)
	RunMaintenanceFn     func(context.Context, *ps.RunNekiShardConfigurationProfileMaintenanceRequest) error
	RunBulkMaintenanceFn func(context.Context, *ps.RunNekiShardConfigurationProfilesMaintenanceRequest) error
	ListChangesFn        func(context.Context, *ps.ListNekiShardConfigurationProfileChangesRequest) ([]*ps.NekiShardConfigurationProfileChange, error)
	GetChangeFn          func(context.Context, *ps.GetNekiShardConfigurationProfileChangeRequest) (*ps.NekiShardConfigurationProfileChange, error)
	CancelChangeFn       func(context.Context, *ps.CancelNekiShardConfigurationProfileChangeRequest) error

	ListFnInvoked, GetFnInvoked, CreateFnInvoked, UpdateFnInvoked, DeleteFnInvoked bool
	GetDefaultFnInvoked, SetDefaultFnInvoked                                       bool
	ListParametersFnInvoked, ListExtensionsFnInvoked, UpdateExtensionFnInvoked     bool
	RunMaintenanceFnInvoked, RunBulkMaintenanceFnInvoked                           bool
	ListChangesFnInvoked, GetChangeFnInvoked, CancelChangeFnInvoked                bool
}

type NekiShardsService struct {
	ListFn        func(context.Context, *ps.ListNekiShardsRequest) ([]*ps.NekiShard, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetNekiShardRequest) (*ps.NekiShard, error)
	GetFnInvoked bool

	CreateFn        func(context.Context, *ps.CreateNekiShardsRequest) (*ps.CreateNekiShardsResponse, error)
	CreateFnInvoked bool

	AssignFn        func(context.Context, *ps.AssignNekiShardsRequest) ([]*ps.NekiShardAssignment, error)
	AssignFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateNekiShardRequest) (*ps.NekiShard, error)
	UpdateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeleteNekiShardRequest) error
	DeleteFnInvoked bool
}

type NekiSidecarsService struct {
	ListFn        func(context.Context, *ps.ListNekiSidecarsRequest) ([]*ps.NekiSidecar, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetNekiSidecarRequest) (*ps.NekiSidecar, error)
	GetFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateNekiSidecarRequest) (*ps.NekiSidecar, error)
	UpdateFnInvoked bool

	ListParametersFn        func(context.Context, *ps.ListNekiSidecarParametersRequest) ([]*ps.NekiParameter, error)
	ListParametersFnInvoked bool

	ListChangesFn        func(context.Context, *ps.ListNekiSidecarChangesRequest) ([]*ps.NekiSidecarChange, error)
	ListChangesFnInvoked bool

	GetChangeFn        func(context.Context, *ps.GetNekiSidecarChangeRequest) (*ps.NekiSidecarChange, error)
	GetChangeFnInvoked bool

	CancelChangeFn        func(context.Context, *ps.CancelNekiSidecarChangeRequest) error
	CancelChangeFnInvoked bool
}

type OrganizationSSOService struct {
	GetFn        func(context.Context, *ps.OrganizationSSORequest) (*ps.OrganizationSSO, error)
	GetFnInvoked bool

	EnableFn        func(context.Context, *ps.OrganizationSSORequest) (*ps.OrganizationSSO, error)
	EnableFnInvoked bool

	DisableFn        func(context.Context, *ps.OrganizationSSORequest) (*ps.OrganizationSSO, error)
	DisableFnInvoked bool

	ConfigureFn        func(context.Context, *ps.OrganizationSSORequest) (*ps.SSOPortal, error)
	ConfigureFnInvoked bool

	EnableDirectoryFn        func(context.Context, *ps.OrganizationSSORequest) (*ps.SSOPortal, error)
	EnableDirectoryFnInvoked bool

	DisableDirectoryFn        func(context.Context, *ps.OrganizationSSORequest) (*ps.OrganizationSSO, error)
	DisableDirectoryFnInvoked bool

	ListDomainsFn        func(context.Context, *ps.OrganizationSSORequest) ([]*ps.OrganizationDomain, error)
	ListDomainsFnInvoked bool

	GetDomainFn        func(context.Context, *ps.OrganizationSSODomainRequest) (*ps.OrganizationDomain, error)
	GetDomainFnInvoked bool

	VerifyDomainFn        func(context.Context, *ps.OrganizationSSORequest) (*ps.SSOPortal, error)
	VerifyDomainFnInvoked bool

	DeleteDomainFn        func(context.Context, *ps.OrganizationSSODomainRequest) error
	DeleteDomainFnInvoked bool
}

type OrganizationsService struct {
	GetFn        func(context.Context, *ps.GetOrganizationRequest) (*ps.Organization, error)
	GetFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdateOrganizationRequest) (*ps.Organization, error)
	UpdateFnInvoked bool

	ListFn        func(context.Context) ([]*ps.Organization, error)
	ListFnInvoked bool

	ListRegionsFn        func(context.Context, *ps.ListOrganizationRegionsRequest) ([]*ps.Region, error)
	ListRegionsFnInvoked bool

	ListClusterSKUsFn        func(context.Context, *ps.ListOrganizationClusterSKUsRequest, ...ps.ListOption) ([]*ps.ClusterSKU, error)
	ListClusterSKUsFnInvoked bool

	ListMembersFn        func(context.Context, *ps.ListOrganizationMembersRequest, ...ps.ListOption) ([]*ps.OrganizationMembership, error)
	ListMembersFnInvoked bool

	GetMemberFn        func(context.Context, *ps.GetOrganizationMemberRequest) (*ps.OrganizationMembership, error)
	GetMemberFnInvoked bool

	UpdateMemberFn        func(context.Context, *ps.UpdateOrganizationMemberRequest) (*ps.OrganizationMembership, error)
	UpdateMemberFnInvoked bool

	RemoveMemberFn        func(context.Context, *ps.RemoveOrganizationMemberRequest) error
	RemoveMemberFnInvoked bool

	ListTeamsFn        func(context.Context, *ps.ListOrganizationTeamsRequest, ...ps.ListOption) ([]*ps.OrganizationTeam, error)
	ListTeamsFnInvoked bool

	GetTeamFn        func(context.Context, *ps.GetOrganizationTeamRequest) (*ps.OrganizationTeam, error)
	GetTeamFnInvoked bool

	CreateTeamFn        func(context.Context, *ps.CreateOrganizationTeamRequest) (*ps.OrganizationTeam, error)
	CreateTeamFnInvoked bool

	UpdateTeamFn        func(context.Context, *ps.UpdateOrganizationTeamRequest) (*ps.OrganizationTeam, error)
	UpdateTeamFnInvoked bool

	DeleteTeamFn        func(context.Context, *ps.DeleteOrganizationTeamRequest) error
	DeleteTeamFnInvoked bool

	ListTeamMembersFn        func(context.Context, *ps.ListOrganizationTeamMembersRequest, ...ps.ListOption) ([]*ps.OrganizationTeamMembership, error)
	ListTeamMembersFnInvoked bool

	AddTeamMemberFn        func(context.Context, *ps.AddOrganizationTeamMemberRequest) (*ps.OrganizationTeamMembership, error)
	AddTeamMemberFnInvoked bool

	RemoveTeamMemberFn        func(context.Context, *ps.RemoveOrganizationTeamMemberRequest) error
	RemoveTeamMemberFnInvoked bool
}

type PasswordsService struct {
	CreateFn        func(context.Context, *ps.DatabaseBranchPasswordRequest) (*ps.DatabaseBranchPassword, error)
	CreateFnInvoked bool
	ListFn          func(context.Context, *ps.ListDatabaseBranchPasswordRequest, ...ps.ListOption) ([]*ps.DatabaseBranchPassword, error)
	ListFnInvoked   bool
	GetFn           func(context.Context, *ps.GetDatabaseBranchPasswordRequest) (*ps.DatabaseBranchPassword, error)
	GetFnInvoked    bool
	UpdateFn        func(context.Context, *ps.UpdateDatabaseBranchPasswordRequest) (*ps.DatabaseBranchPassword, error)
	UpdateFnInvoked bool
	DeleteFn        func(context.Context, *ps.DeleteDatabaseBranchPasswordRequest) error
	DeleteFnInvoked bool
	RenewFn         func(context.Context, *ps.RenewDatabaseBranchPasswordRequest) (*ps.DatabaseBranchPassword, error)
	RenewFnInvoked  bool
}

type PlannedReparentShardService struct {
	CreateFn        func(context.Context, *ps.PlannedReparentShardRequest) (*ps.VtctldOperation, error)
	CreateFnInvoked bool

	GetFn        func(context.Context, *ps.GetPlannedReparentShardRequest) (*ps.VtctldOperation, error)
	GetFnInvoked bool
}

type PostgresBouncersService struct {
	ListFn        func(context.Context, *ps.ListPostgresBouncersRequest, ...ps.ListOption) ([]*ps.PostgresBouncer, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetPostgresBouncerRequest) (*ps.PostgresBouncer, error)
	GetFnInvoked bool

	CreateFn        func(context.Context, *ps.CreatePostgresBouncerRequest) (*ps.PostgresBouncer, error)
	CreateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeletePostgresBouncerRequest) error
	DeleteFnInvoked bool

	ListResizesFn        func(context.Context, *ps.ListPostgresBouncerResizesRequest, ...ps.ListOption) ([]*ps.PostgresBouncerResizeRequest, error)
	ListResizesFnInvoked bool

	ResizeFn        func(context.Context, *ps.ResizePostgresBouncerRequest) (*ps.PostgresBouncerResizeRequest, error)
	ResizeFnInvoked bool

	CancelResizesFn        func(context.Context, *ps.CancelPostgresBouncerResizesRequest) error
	CancelResizesFnInvoked bool
}

type PostgresBranchesService struct {
	CreateFn        func(context.Context, *ps.CreatePostgresBranchRequest) (*ps.PostgresBranch, error)
	CreateFnInvoked bool

	ListFn        func(context.Context, *ps.ListPostgresBranchesRequest, ...ps.ListOption) ([]*ps.PostgresBranch, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetPostgresBranchRequest) (*ps.PostgresBranch, error)
	GetFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdatePostgresBranchRequest) (*ps.PostgresBranch, error)
	UpdateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeletePostgresBranchRequest) error
	DeleteFnInvoked bool

	SchemaFn        func(context.Context, *ps.PostgresBranchSchemaRequest) ([]*ps.PostgresBranchSchema, error)
	SchemaFnInvoked bool

	ListClusterSKUsFn        func(context.Context, *ps.ListBranchClusterSKUsRequest, ...ps.ListOption) ([]*ps.ClusterSKU, error)
	ListClusterSKUsFnInvoked bool

	ResizeFn        func(context.Context, *ps.ResizePostgresBranchRequest) (*ps.PostgresBranchClusterResizeRequest, error)
	ResizeFnInvoked bool

	ListChangesFn        func(context.Context, *ps.ListPostgresBranchChangesRequest) ([]*ps.PostgresBranchClusterResizeRequest, error)
	ListChangesFnInvoked bool

	GetChangeFn        func(context.Context, *ps.GetPostgresBranchChangeRequest) (*ps.PostgresBranchClusterResizeRequest, error)
	GetChangeFnInvoked bool

	CancelChangesFn        func(context.Context, *ps.CancelPostgresBranchChangesRequest) error
	CancelChangesFnInvoked bool

	ListParametersFn        func(context.Context, *ps.ListPostgresParametersRequest) ([]*ps.PostgresParameter, error)
	ListParametersFnInvoked bool

	ListExtensionsFn        func(context.Context, *ps.ListPostgresExtensionsRequest) ([]*ps.PostgresExtension, error)
	ListExtensionsFnInvoked bool
}

type PostgresCIDRsService struct {
	ListFn        func(context.Context, *ps.ListPostgresCIDRsRequest, ...ps.ListOption) ([]*ps.PostgresCIDR, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetPostgresCIDRRequest) (*ps.PostgresCIDR, error)
	GetFnInvoked bool

	CreateFn        func(context.Context, *ps.CreatePostgresCIDRRequest) (*ps.PostgresCIDR, error)
	CreateFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdatePostgresCIDRRequest) (*ps.PostgresCIDR, error)
	UpdateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeletePostgresCIDRRequest) error
	DeleteFnInvoked bool
}

type PostgresReadOnlyReplicasService struct {
	ListFn        func(context.Context, *ps.ListPostgresReadOnlyReplicasRequest) ([]*ps.PostgresReadOnlyReplica, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetPostgresReadOnlyReplicaRequest) (*ps.PostgresReadOnlyReplica, error)
	GetFnInvoked bool

	CreateFn        func(context.Context, *ps.CreatePostgresReadOnlyReplicaRequest) (*ps.PostgresReadOnlyReplica, error)
	CreateFnInvoked bool

	UpdateFn        func(context.Context, *ps.UpdatePostgresReadOnlyReplicaRequest) (*ps.PostgresReadOnlyReplica, error)
	UpdateFnInvoked bool

	DeleteFn        func(context.Context, *ps.DeletePostgresReadOnlyReplicaRequest) error
	DeleteFnInvoked bool
}

type PostgresRolesService struct {
	ListFn                    func(context.Context, *ps.ListPostgresRolesRequest, ...ps.ListOption) ([]*ps.PostgresRole, error)
	ListFnInvoked             bool
	GetFn                     func(context.Context, *ps.GetPostgresRoleRequest) (*ps.PostgresRole, error)
	GetFnInvoked              bool
	CreateFn                  func(context.Context, *ps.CreatePostgresRoleRequest) (*ps.PostgresRole, error)
	CreateFnInvoked           bool
	UpdateFn                  func(context.Context, *ps.UpdatePostgresRoleRequest) (*ps.PostgresRole, error)
	UpdateFnInvoked           bool
	RenewFn                   func(context.Context, *ps.RenewPostgresRoleRequest) (*ps.PostgresRole, error)
	RenewFnInvoked            bool
	DeleteFn                  func(context.Context, *ps.DeletePostgresRoleRequest) error
	DeleteFnInvoked           bool
	ResetDefaultRoleFn        func(context.Context, *ps.ResetDefaultRoleRequest) (*ps.PostgresRole, error)
	ResetDefaultRoleFnInvoked bool
	GetDefaultRoleFn          func(context.Context, *ps.GetDefaultPostgresRoleRequest) (*ps.PostgresRole, error)
	GetDefaultRoleFnInvoked   bool
	ResetPasswordFn           func(context.Context, *ps.ResetPostgresRolePasswordRequest) (*ps.PostgresRole, error)
	ResetPasswordFnInvoked    bool
	ReassignObjectsFn         func(context.Context, *ps.ReassignPostgresRoleObjectsRequest) error
	ReassignObjectsFnInvoked  bool
}