mock

type AuditLogService struct {
	ListFn        func(context.Context, *ps.ListAuditLogsRequest, ...ps.ListOption) (*ps.CursorPaginatedResponse[*ps.AuditLog], error)
	ListFnInvoked 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

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

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

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

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

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

	UpdateVSchemaFn        func(context.Context, *ps.UpdateKeyspaceVSchemaRequest) (*ps.VSchema, error)
	UpdateVSchemaFnInvoked 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.DatabaseBranch, error)
	ListFnInvoked bool

	GetFn        func(context.Context, *ps.GetDatabaseBranchRequest) (*ps.DatabaseBranch, error)
	GetFnInvoked 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

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

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

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

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

	KeyspacesFn        func(context.Context, *ps.BranchKeyspacesRequest) ([]*ps.Keyspace, error)
	KeyspacesFnInvoked 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
}

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

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

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

	AutoApplyFn        func(context.Context, *ps.AutoApplyDeployRequestRequest) (*ps.DeployRequest, error)
	AutoApplyFnInvoked 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
}

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

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

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

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

type ServiceTokenService struct {
	CreateFn        func(context.Context, *ps.CreateServiceTokenRequest) (*ps.ServiceToken, error)
	CreateFnInvoked bool

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

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

	GetAccessFn        func(context.Context, *ps.GetServiceTokenAccessRequest) ([]*ps.ServiceTokenAccess, error)
	GetAccessFnInvoked bool

	AddAccessFn        func(context.Context, *ps.AddServiceTokenAccessRequest) ([]*ps.ServiceTokenAccess, error)
	AddAccessFnInvoked bool

	DeleteAccessFn        func(context.Context, *ps.DeleteServiceTokenAccessRequest) error
	DeleteAccessFnInvoked bool

	ListGrantsFn        func(context.Context, *ps.ListServiceTokenGrantsRequest) ([]*ps.ServiceTokenGrant, error)
	ListGrantsFnInvoked bool
}