// Automatically generated by mockimpl. DO NOT EDIT! package mock import ( "context" "encoding/json" "sync" "time" "github.com/fleetdm/fleet/v4/server/config" "github.com/fleetdm/fleet/v4/server/fleet" "github.com/micromdm/nanodep/godep" ) var _ fleet.Datastore = (*DataStore)(nil) type HealthCheckFunc func() error type NewCarveFunc func(ctx context.Context, metadata *fleet.CarveMetadata) (*fleet.CarveMetadata, error) type UpdateCarveFunc func(ctx context.Context, metadata *fleet.CarveMetadata) error type CarveFunc func(ctx context.Context, carveId int64) (*fleet.CarveMetadata, error) type CarveBySessionIdFunc func(ctx context.Context, sessionId string) (*fleet.CarveMetadata, error) type CarveByNameFunc func(ctx context.Context, name string) (*fleet.CarveMetadata, error) type ListCarvesFunc func(ctx context.Context, opt fleet.CarveListOptions) ([]*fleet.CarveMetadata, error) type NewBlockFunc func(ctx context.Context, metadata *fleet.CarveMetadata, blockId int64, data []byte) error type GetBlockFunc func(ctx context.Context, metadata *fleet.CarveMetadata, blockId int64) ([]byte, error) type CleanupCarvesFunc func(ctx context.Context, now time.Time) (expired int, err error) type NewUserFunc func(ctx context.Context, user *fleet.User) (*fleet.User, error) type ListUsersFunc func(ctx context.Context, opt fleet.UserListOptions) ([]*fleet.User, error) type UserByEmailFunc func(ctx context.Context, email string) (*fleet.User, error) type UserByIDFunc func(ctx context.Context, id uint) (*fleet.User, error) type SaveUserFunc func(ctx context.Context, user *fleet.User) error type SaveUsersFunc func(ctx context.Context, users []*fleet.User) error type DeleteUserFunc func(ctx context.Context, id uint) error type PendingEmailChangeFunc func(ctx context.Context, userID uint, newEmail string, token string) error type ConfirmPendingEmailChangeFunc func(ctx context.Context, userID uint, token string) (string, error) type ApplyQueriesFunc func(ctx context.Context, authorID uint, queries []*fleet.Query) error type NewQueryFunc func(ctx context.Context, query *fleet.Query, opts ...fleet.OptionalArg) (*fleet.Query, error) type SaveQueryFunc func(ctx context.Context, query *fleet.Query) error type DeleteQueryFunc func(ctx context.Context, name string) error type DeleteQueriesFunc func(ctx context.Context, ids []uint) (uint, error) type QueryFunc func(ctx context.Context, id uint) (*fleet.Query, error) type ListQueriesFunc func(ctx context.Context, opt fleet.ListQueryOptions) ([]*fleet.Query, error) type QueryByNameFunc func(ctx context.Context, name string, opts ...fleet.OptionalArg) (*fleet.Query, error) type ObserverCanRunQueryFunc func(ctx context.Context, queryID uint) (bool, error) type NewDistributedQueryCampaignFunc func(ctx context.Context, camp *fleet.DistributedQueryCampaign) (*fleet.DistributedQueryCampaign, error) type DistributedQueryCampaignFunc func(ctx context.Context, id uint) (*fleet.DistributedQueryCampaign, error) type SaveDistributedQueryCampaignFunc func(ctx context.Context, camp *fleet.DistributedQueryCampaign) error type DistributedQueryCampaignTargetIDsFunc func(ctx context.Context, id uint) (targets *fleet.HostTargets, err error) type NewDistributedQueryCampaignTargetFunc func(ctx context.Context, target *fleet.DistributedQueryCampaignTarget) (*fleet.DistributedQueryCampaignTarget, error) type CleanupDistributedQueryCampaignsFunc func(ctx context.Context, now time.Time) (expired uint, err error) type DistributedQueryCampaignsForQueryFunc func(ctx context.Context, queryID uint) ([]*fleet.DistributedQueryCampaign, error) type ApplyPackSpecsFunc func(ctx context.Context, specs []*fleet.PackSpec) error type GetPackSpecsFunc func(ctx context.Context) ([]*fleet.PackSpec, error) type GetPackSpecFunc func(ctx context.Context, name string) (*fleet.PackSpec, error) type NewPackFunc func(ctx context.Context, pack *fleet.Pack, opts ...fleet.OptionalArg) (*fleet.Pack, error) type SavePackFunc func(ctx context.Context, pack *fleet.Pack) error type DeletePackFunc func(ctx context.Context, name string) error type PackFunc func(ctx context.Context, pid uint) (*fleet.Pack, error) type ListPacksFunc func(ctx context.Context, opt fleet.PackListOptions) ([]*fleet.Pack, error) type PackByNameFunc func(ctx context.Context, name string, opts ...fleet.OptionalArg) (*fleet.Pack, bool, error) type ListPacksForHostFunc func(ctx context.Context, hid uint) (packs []*fleet.Pack, err error) type EnsureGlobalPackFunc func(ctx context.Context) (*fleet.Pack, error) type EnsureTeamPackFunc func(ctx context.Context, teamID uint) (*fleet.Pack, error) type ApplyLabelSpecsFunc func(ctx context.Context, specs []*fleet.LabelSpec) error type GetLabelSpecsFunc func(ctx context.Context) ([]*fleet.LabelSpec, error) type GetLabelSpecFunc func(ctx context.Context, name string) (*fleet.LabelSpec, error) type NewLabelFunc func(ctx context.Context, Label *fleet.Label, opts ...fleet.OptionalArg) (*fleet.Label, error) type SaveLabelFunc func(ctx context.Context, label *fleet.Label) (*fleet.Label, error) type DeleteLabelFunc func(ctx context.Context, name string) error type LabelFunc func(ctx context.Context, lid uint) (*fleet.Label, error) type ListLabelsFunc func(ctx context.Context, filter fleet.TeamFilter, opt fleet.ListOptions) ([]*fleet.Label, error) type LabelsSummaryFunc func(ctx context.Context) ([]*fleet.LabelSummary, error) type LabelQueriesForHostFunc func(ctx context.Context, host *fleet.Host) (map[string]string, error) type ListLabelsForHostFunc func(ctx context.Context, hid uint) ([]*fleet.Label, error) type ListHostsInLabelFunc func(ctx context.Context, filter fleet.TeamFilter, lid uint, opt fleet.HostListOptions) ([]*fleet.Host, error) type ListUniqueHostsInLabelsFunc func(ctx context.Context, filter fleet.TeamFilter, labels []uint) ([]*fleet.Host, error) type SearchLabelsFunc func(ctx context.Context, filter fleet.TeamFilter, query string, omit ...uint) ([]*fleet.Label, error) type LabelIDsByNameFunc func(ctx context.Context, labels []string) ([]uint, error) type AsyncBatchInsertLabelMembershipFunc func(ctx context.Context, batch [][2]uint) error type AsyncBatchDeleteLabelMembershipFunc func(ctx context.Context, batch [][2]uint) error type AsyncBatchUpdateLabelTimestampFunc func(ctx context.Context, ids []uint, ts time.Time) error type NewHostFunc func(ctx context.Context, host *fleet.Host) (*fleet.Host, error) type DeleteHostFunc func(ctx context.Context, hid uint) error type HostFunc func(ctx context.Context, id uint) (*fleet.Host, error) type ListHostsFunc func(ctx context.Context, filter fleet.TeamFilter, opt fleet.HostListOptions) ([]*fleet.Host, error) type MarkHostsSeenFunc func(ctx context.Context, hostIDs []uint, t time.Time) error type SearchHostsFunc func(ctx context.Context, filter fleet.TeamFilter, query string, omit ...uint) ([]*fleet.Host, error) type EnrolledHostIDsFunc func(ctx context.Context) ([]uint, error) type CountEnrolledHostsFunc func(ctx context.Context) (int, error) type CleanupIncomingHostsFunc func(ctx context.Context, now time.Time) ([]uint, error) type GenerateHostStatusStatisticsFunc func(ctx context.Context, filter fleet.TeamFilter, now time.Time, platform *string, lowDiskSpace *int) (*fleet.HostSummary, error) type HostIDsByNameFunc func(ctx context.Context, filter fleet.TeamFilter, hostnames []string) ([]uint, error) type HostIDsByOSIDFunc func(ctx context.Context, osID uint, offset int, limit int) ([]uint, error) type HostIDsByOSVersionFunc func(ctx context.Context, osVersion fleet.OSVersion, offset int, limit int) ([]uint, error) type HostByIdentifierFunc func(ctx context.Context, identifier string) (*fleet.Host, error) type AddHostsToTeamFunc func(ctx context.Context, teamID *uint, hostIDs []uint) error type TotalAndUnseenHostsSinceFunc func(ctx context.Context, daysCount int) (total int, unseen int, err error) type DeleteHostsFunc func(ctx context.Context, ids []uint) error type CountHostsFunc func(ctx context.Context, filter fleet.TeamFilter, opt fleet.HostListOptions) (int, error) type CountHostsInLabelFunc func(ctx context.Context, filter fleet.TeamFilter, lid uint, opt fleet.HostListOptions) (int, error) type ListHostDeviceMappingFunc func(ctx context.Context, id uint) ([]*fleet.HostDeviceMapping, error) type ListHostBatteriesFunc func(ctx context.Context, id uint) ([]*fleet.HostBattery, error) type LoadHostByDeviceAuthTokenFunc func(ctx context.Context, authToken string, tokenTTL time.Duration) (*fleet.Host, error) type SetOrUpdateDeviceAuthTokenFunc func(ctx context.Context, hostID uint, authToken string) error type FailingPoliciesCountFunc func(ctx context.Context, host *fleet.Host) (uint, error) type ListPoliciesForHostFunc func(ctx context.Context, host *fleet.Host) ([]*fleet.HostPolicy, error) type GetHostMunkiVersionFunc func(ctx context.Context, hostID uint) (string, error) type GetHostMunkiIssuesFunc func(ctx context.Context, hostID uint) ([]*fleet.HostMunkiIssue, error) type GetHostMDMFunc func(ctx context.Context, hostID uint) (*fleet.HostMDM, error) type GetHostMDMCheckinInfoFunc func(ctx context.Context, hostUUID string) (*fleet.HostMDMCheckinInfo, error) type AggregatedMunkiVersionFunc func(ctx context.Context, teamID *uint) ([]fleet.AggregatedMunkiVersion, time.Time, error) type AggregatedMunkiIssuesFunc func(ctx context.Context, teamID *uint) ([]fleet.AggregatedMunkiIssue, time.Time, error) type AggregatedMDMStatusFunc func(ctx context.Context, teamID *uint, platform string) (fleet.AggregatedMDMStatus, time.Time, error) type AggregatedMDMSolutionsFunc func(ctx context.Context, teamID *uint, platform string) ([]fleet.AggregatedMDMSolutions, time.Time, error) type GenerateAggregatedMunkiAndMDMFunc func(ctx context.Context) error type GetMunkiIssueFunc func(ctx context.Context, munkiIssueID uint) (*fleet.MunkiIssue, error) type GetMDMSolutionFunc func(ctx context.Context, mdmID uint) (*fleet.MDMSolution, error) type OSVersionsFunc func(ctx context.Context, teamID *uint, platform *string, name *string, version *string) (*fleet.OSVersions, error) type UpdateOSVersionsFunc func(ctx context.Context) error type CountHostsInTargetsFunc func(ctx context.Context, filter fleet.TeamFilter, targets fleet.HostTargets, now time.Time) (fleet.TargetMetrics, error) type HostIDsInTargetsFunc func(ctx context.Context, filter fleet.TeamFilter, targets fleet.HostTargets) ([]uint, error) type NewPasswordResetRequestFunc func(ctx context.Context, req *fleet.PasswordResetRequest) (*fleet.PasswordResetRequest, error) type DeletePasswordResetRequestsForUserFunc func(ctx context.Context, userID uint) error type FindPasswordResetByTokenFunc func(ctx context.Context, token string) (*fleet.PasswordResetRequest, error) type CleanupExpiredPasswordResetRequestsFunc func(ctx context.Context) error type SessionByKeyFunc func(ctx context.Context, key string) (*fleet.Session, error) type SessionByIDFunc func(ctx context.Context, id uint) (*fleet.Session, error) type ListSessionsForUserFunc func(ctx context.Context, id uint) ([]*fleet.Session, error) type NewSessionFunc func(ctx context.Context, userID uint, sessionKey string) (*fleet.Session, error) type DestroySessionFunc func(ctx context.Context, session *fleet.Session) error type DestroyAllSessionsForUserFunc func(ctx context.Context, id uint) error type MarkSessionAccessedFunc func(ctx context.Context, session *fleet.Session) error type NewAppConfigFunc func(ctx context.Context, info *fleet.AppConfig) (*fleet.AppConfig, error) type AppConfigFunc func(ctx context.Context) (*fleet.AppConfig, error) type SaveAppConfigFunc func(ctx context.Context, info *fleet.AppConfig) error type GetEnrollSecretsFunc func(ctx context.Context, teamID *uint) ([]*fleet.EnrollSecret, error) type ApplyEnrollSecretsFunc func(ctx context.Context, teamID *uint, secrets []*fleet.EnrollSecret) error type NewInviteFunc func(ctx context.Context, i *fleet.Invite) (*fleet.Invite, error) type ListInvitesFunc func(ctx context.Context, opt fleet.ListOptions) ([]*fleet.Invite, error) type InviteFunc func(ctx context.Context, id uint) (*fleet.Invite, error) type InviteByEmailFunc func(ctx context.Context, email string) (*fleet.Invite, error) type InviteByTokenFunc func(ctx context.Context, token string) (*fleet.Invite, error) type DeleteInviteFunc func(ctx context.Context, id uint) error type UpdateInviteFunc func(ctx context.Context, id uint, i *fleet.Invite) (*fleet.Invite, error) type ListScheduledQueriesInPackWithStatsFunc func(ctx context.Context, id uint, opts fleet.ListOptions) ([]*fleet.ScheduledQuery, error) type NewScheduledQueryFunc func(ctx context.Context, sq *fleet.ScheduledQuery, opts ...fleet.OptionalArg) (*fleet.ScheduledQuery, error) type SaveScheduledQueryFunc func(ctx context.Context, sq *fleet.ScheduledQuery) (*fleet.ScheduledQuery, error) type DeleteScheduledQueryFunc func(ctx context.Context, id uint) error type ScheduledQueryFunc func(ctx context.Context, id uint) (*fleet.ScheduledQuery, error) type CleanupExpiredHostsFunc func(ctx context.Context) ([]uint, error) type ScheduledQueryIDsByNameFunc func(ctx context.Context, batchSize int, packAndSchedQueryNames ...[2]string) ([]uint, error) type NewTeamFunc func(ctx context.Context, team *fleet.Team) (*fleet.Team, error) type SaveTeamFunc func(ctx context.Context, team *fleet.Team) (*fleet.Team, error) type TeamFunc func(ctx context.Context, tid uint) (*fleet.Team, error) type DeleteTeamFunc func(ctx context.Context, tid uint) error type TeamByNameFunc func(ctx context.Context, name string) (*fleet.Team, error) type ListTeamsFunc func(ctx context.Context, filter fleet.TeamFilter, opt fleet.ListOptions) ([]*fleet.Team, error) type TeamsSummaryFunc func(ctx context.Context) ([]*fleet.TeamSummary, error) type SearchTeamsFunc func(ctx context.Context, filter fleet.TeamFilter, matchQuery string, omit ...uint) ([]*fleet.Team, error) type TeamEnrollSecretsFunc func(ctx context.Context, teamID uint) ([]*fleet.EnrollSecret, error) type DeleteIntegrationsFromTeamsFunc func(ctx context.Context, deletedIntgs fleet.Integrations) error type ListSoftwareForVulnDetectionFunc func(ctx context.Context, hostID uint) ([]fleet.Software, error) type ListSoftwareVulnerabilitiesByHostIDsSourceFunc func(ctx context.Context, hostIDs []uint, source fleet.VulnerabilitySource) (map[uint][]fleet.SoftwareVulnerability, error) type LoadHostSoftwareFunc func(ctx context.Context, host *fleet.Host, includeCVEScores bool) error type ListSoftwareBySourceIterFunc func(ctx context.Context, sources []string) (fleet.SoftwareIterator, error) type AllSoftwareWithoutCPEIteratorFunc func(ctx context.Context, excludedPlatforms []string) (fleet.SoftwareIterator, error) type AddCPEForSoftwareFunc func(ctx context.Context, software fleet.Software, cpe string) error type ListSoftwareCPEsFunc func(ctx context.Context) ([]fleet.SoftwareCPE, error) type InsertSoftwareVulnerabilitiesFunc func(ctx context.Context, vulns []fleet.SoftwareVulnerability, source fleet.VulnerabilitySource) (int64, error) type SoftwareByIDFunc func(ctx context.Context, id uint, includeCVEScores bool) (*fleet.Software, error) type ListSoftwareByHostIDShortFunc func(ctx context.Context, hostID uint) ([]fleet.Software, error) type SyncHostsSoftwareFunc func(ctx context.Context, updatedAt time.Time) error type HostsBySoftwareIDsFunc func(ctx context.Context, softwareIDs []uint) ([]*fleet.HostShort, error) type HostsByCVEFunc func(ctx context.Context, cve string) ([]*fleet.HostShort, error) type InsertCVEMetaFunc func(ctx context.Context, cveMeta []fleet.CVEMeta) error type ListCVEsFunc func(ctx context.Context, maxAge time.Duration) ([]fleet.CVEMeta, error) type ListOperatingSystemsFunc func(ctx context.Context) ([]fleet.OperatingSystem, error) type UpdateHostOperatingSystemFunc func(ctx context.Context, hostID uint, hostOS fleet.OperatingSystem) error type CleanupHostOperatingSystemsFunc func(ctx context.Context) error type UpdateHostTablesOnMDMUnenrollFunc func(ctx context.Context, uuid string) error type NewActivityFunc func(ctx context.Context, user *fleet.User, activity fleet.ActivityDetails) error type ListActivitiesFunc func(ctx context.Context, opt fleet.ListActivitiesOptions) ([]*fleet.Activity, *fleet.PaginationMetadata, error) type MarkActivitiesAsStreamedFunc func(ctx context.Context, activityIDs []uint) error type ShouldSendStatisticsFunc func(ctx context.Context, frequency time.Duration, config config.FleetConfig) (fleet.StatisticsPayload, bool, error) type RecordStatisticsSentFunc func(ctx context.Context) error type CleanupStatisticsFunc func(ctx context.Context) error type ApplyPolicySpecsFunc func(ctx context.Context, authorID uint, specs []*fleet.PolicySpec) error type NewGlobalPolicyFunc func(ctx context.Context, authorID *uint, args fleet.PolicyPayload) (*fleet.Policy, error) type PolicyFunc func(ctx context.Context, id uint) (*fleet.Policy, error) type SavePolicyFunc func(ctx context.Context, p *fleet.Policy) error type ListGlobalPoliciesFunc func(ctx context.Context) ([]*fleet.Policy, error) type PoliciesByIDFunc func(ctx context.Context, ids []uint) (map[uint]*fleet.Policy, error) type DeleteGlobalPoliciesFunc func(ctx context.Context, ids []uint) ([]uint, error) type PolicyQueriesForHostFunc func(ctx context.Context, host *fleet.Host) (map[string]string, error) type AsyncBatchInsertPolicyMembershipFunc func(ctx context.Context, batch []fleet.PolicyMembershipResult) error type AsyncBatchUpdatePolicyTimestampFunc func(ctx context.Context, ids []uint, ts time.Time) error type MigrateTablesFunc func(ctx context.Context) error type MigrateDataFunc func(ctx context.Context) error type MigrationStatusFunc func(ctx context.Context) (*fleet.MigrationStatus, error) type ListSoftwareFunc func(ctx context.Context, opt fleet.SoftwareListOptions) ([]fleet.Software, error) type CountSoftwareFunc func(ctx context.Context, opt fleet.SoftwareListOptions) (int, error) type DeleteSoftwareVulnerabilitiesFunc func(ctx context.Context, vulnerabilities []fleet.SoftwareVulnerability) error type NewTeamPolicyFunc func(ctx context.Context, teamID uint, authorID *uint, args fleet.PolicyPayload) (*fleet.Policy, error) type ListTeamPoliciesFunc func(ctx context.Context, teamID uint) (teamPolicies []*fleet.Policy, inheritedPolicies []*fleet.Policy, err error) type DeleteTeamPoliciesFunc func(ctx context.Context, teamID uint, ids []uint) ([]uint, error) type TeamPolicyFunc func(ctx context.Context, teamID uint, policyID uint) (*fleet.Policy, error) type CleanupPolicyMembershipFunc func(ctx context.Context, now time.Time) error type IncrementPolicyViolationDaysFunc func(ctx context.Context) error type InitializePolicyViolationDaysFunc func(ctx context.Context) error type LockFunc func(ctx context.Context, name string, owner string, expiration time.Duration) (bool, error) type UnlockFunc func(ctx context.Context, name string, owner string) error type DBLocksFunc func(ctx context.Context) ([]*fleet.DBLock, error) type GetLatestCronStatsFunc func(ctx context.Context, name string) ([]fleet.CronStats, error) type InsertCronStatsFunc func(ctx context.Context, statsType fleet.CronStatsType, name string, instance string, status fleet.CronStatsStatus) (int, error) type UpdateCronStatsFunc func(ctx context.Context, id int, status fleet.CronStatsStatus) error type UpdateAllCronStatsForInstanceFunc func(ctx context.Context, instance string, fromStatus fleet.CronStatsStatus, toStatus fleet.CronStatsStatus) error type CleanupCronStatsFunc func(ctx context.Context) error type UpdateScheduledQueryAggregatedStatsFunc func(ctx context.Context) error type UpdateQueryAggregatedStatsFunc func(ctx context.Context) error type LoadHostByNodeKeyFunc func(ctx context.Context, nodeKey string) (*fleet.Host, error) type LoadHostByOrbitNodeKeyFunc func(ctx context.Context, nodeKey string) (*fleet.Host, error) type HostLiteFunc func(ctx context.Context, hostID uint) (*fleet.Host, error) type UpdateHostOsqueryIntervalsFunc func(ctx context.Context, hostID uint, intervals fleet.HostOsqueryIntervals) error type TeamAgentOptionsFunc func(ctx context.Context, teamID uint) (*json.RawMessage, error) type TeamFeaturesFunc func(ctx context.Context, teamID uint) (*fleet.Features, error) type TeamMDMConfigFunc func(ctx context.Context, teamID uint) (*fleet.TeamMDM, error) type SaveHostPackStatsFunc func(ctx context.Context, hostID uint, stats []fleet.PackStats) error type AsyncBatchSaveHostsScheduledQueryStatsFunc func(ctx context.Context, stats map[uint][]fleet.ScheduledQueryStats, batchSize int) (int, error) type UpdateHostSoftwareFunc func(ctx context.Context, hostID uint, software []fleet.Software) error type UpdateHostFunc func(ctx context.Context, host *fleet.Host) error type ListScheduledQueriesInPackFunc func(ctx context.Context, packID uint) (fleet.ScheduledQueryList, error) type UpdateHostRefetchRequestedFunc func(ctx context.Context, hostID uint, value bool) error type FlippingPoliciesForHostFunc func(ctx context.Context, hostID uint, incomingResults map[uint]*bool) (newFailing []uint, newPassing []uint, err error) type RecordPolicyQueryExecutionsFunc func(ctx context.Context, host *fleet.Host, results map[uint]*bool, updated time.Time, deferredSaveHost bool) error type RecordLabelQueryExecutionsFunc func(ctx context.Context, host *fleet.Host, results map[uint]*bool, t time.Time, deferredSaveHost bool) error type SaveHostUsersFunc func(ctx context.Context, hostID uint, users []fleet.HostUser) error type SaveHostAdditionalFunc func(ctx context.Context, hostID uint, additional *json.RawMessage) error type SetOrUpdateMunkiInfoFunc func(ctx context.Context, hostID uint, version string, errors []string, warnings []string) error type SetOrUpdateMDMDataFunc func(ctx context.Context, hostID uint, isServer bool, enrolled bool, serverURL string, installedFromDep bool, name string) error type SetOrUpdateHostDisksSpaceFunc func(ctx context.Context, hostID uint, gigsAvailable float64, percentAvailable float64) error type SetOrUpdateHostDisksEncryptionFunc func(ctx context.Context, hostID uint, encrypted bool) error type SetOrUpdateHostDiskEncryptionKeyFunc func(ctx context.Context, hostID uint, encryptedBase64Key string) error type GetUnverifiedDiskEncryptionKeysFunc func(ctx context.Context) ([]fleet.HostDiskEncryptionKey, error) type SetHostsDiskEncryptionKeyStatusFunc func(ctx context.Context, hostIDs []uint, encryptable bool, threshold time.Time) error type GetHostDiskEncryptionKeyFunc func(ctx context.Context, hostID uint) (*fleet.HostDiskEncryptionKey, error) type SetOrUpdateHostOrbitInfoFunc func(ctx context.Context, hostID uint, version string) error type ReplaceHostDeviceMappingFunc func(ctx context.Context, id uint, mappings []*fleet.HostDeviceMapping) error type ReplaceHostBatteriesFunc func(ctx context.Context, id uint, mappings []*fleet.HostBattery) error type VerifyEnrollSecretFunc func(ctx context.Context, secret string) (*fleet.EnrollSecret, error) type EnrollHostFunc func(ctx context.Context, osqueryHostId string, nodeKey string, teamID *uint, cooldown time.Duration) (*fleet.Host, error) type EnrollOrbitFunc func(ctx context.Context, hardwareUUID string, orbitNodeKey string, teamID *uint) (*fleet.Host, error) type SerialUpdateHostFunc func(ctx context.Context, host *fleet.Host) error type NewJobFunc func(ctx context.Context, job *fleet.Job) (*fleet.Job, error) type GetQueuedJobsFunc func(ctx context.Context, maxNumJobs int) ([]*fleet.Job, error) type UpdateJobFunc func(ctx context.Context, id uint, job *fleet.Job) (*fleet.Job, error) type InnoDBStatusFunc func(ctx context.Context) (string, error) type ProcessListFunc func(ctx context.Context) ([]fleet.MySQLProcess, error) type ListWindowsUpdatesByHostIDFunc func(ctx context.Context, hostID uint) ([]fleet.WindowsUpdate, error) type InsertWindowsUpdatesFunc func(ctx context.Context, hostID uint, updates []fleet.WindowsUpdate) error type ListOSVulnerabilitiesFunc func(ctx context.Context, hostID []uint) ([]fleet.OSVulnerability, error) type InsertOSVulnerabilitiesFunc func(ctx context.Context, vulnerabilities []fleet.OSVulnerability, source fleet.VulnerabilitySource) (int64, error) type DeleteOSVulnerabilitiesFunc func(ctx context.Context, vulnerabilities []fleet.OSVulnerability) error type NewMDMAppleConfigProfileFunc func(ctx context.Context, p fleet.MDMAppleConfigProfile) (*fleet.MDMAppleConfigProfile, error) type GetMDMAppleConfigProfileFunc func(ctx context.Context, profileID uint) (*fleet.MDMAppleConfigProfile, error) type ListMDMAppleConfigProfilesFunc func(ctx context.Context, teamID *uint) ([]*fleet.MDMAppleConfigProfile, error) type DeleteMDMAppleConfigProfileFunc func(ctx context.Context, profileID uint) error type GetHostMDMProfilesFunc func(ctx context.Context, hostUUID string) ([]fleet.HostMDMAppleProfile, error) type NewMDMAppleEnrollmentProfileFunc func(ctx context.Context, enrollmentPayload fleet.MDMAppleEnrollmentProfilePayload) (*fleet.MDMAppleEnrollmentProfile, error) type GetMDMAppleEnrollmentProfileByTokenFunc func(ctx context.Context, token string) (*fleet.MDMAppleEnrollmentProfile, error) type ListMDMAppleEnrollmentProfilesFunc func(ctx context.Context) ([]*fleet.MDMAppleEnrollmentProfile, error) type GetMDMAppleCommandResultsFunc func(ctx context.Context, commandUUID string) (map[string]*fleet.MDMAppleCommandResult, error) type NewMDMAppleInstallerFunc func(ctx context.Context, name string, size int64, manifest string, installer []byte, urlToken string) (*fleet.MDMAppleInstaller, error) type MDMAppleInstallerFunc func(ctx context.Context, token string) (*fleet.MDMAppleInstaller, error) type MDMAppleInstallerDetailsByIDFunc func(ctx context.Context, id uint) (*fleet.MDMAppleInstaller, error) type DeleteMDMAppleInstallerFunc func(ctx context.Context, id uint) error type MDMAppleInstallerDetailsByTokenFunc func(ctx context.Context, token string) (*fleet.MDMAppleInstaller, error) type ListMDMAppleInstallersFunc func(ctx context.Context) ([]fleet.MDMAppleInstaller, error) type BatchSetMDMAppleProfilesFunc func(ctx context.Context, tmID *uint, profiles []*fleet.MDMAppleConfigProfile) error type MDMAppleListDevicesFunc func(ctx context.Context) ([]fleet.MDMAppleDevice, error) type IngestMDMAppleDevicesFromDEPSyncFunc func(ctx context.Context, devices []godep.Device) (int64, error) type IngestMDMAppleDeviceFromCheckinFunc func(ctx context.Context, mdmHost fleet.MDMAppleHostDetails) error type GetNanoMDMEnrollmentStatusFunc func(ctx context.Context, id string) (bool, error) type IncreasePolicyAutomationIterationFunc func(ctx context.Context, policyID uint) error type OutdatedAutomationBatchFunc func(ctx context.Context) ([]fleet.PolicyFailure, error) type ListMDMAppleProfilesToInstallFunc func(ctx context.Context) ([]*fleet.MDMAppleProfilePayload, error) type ListMDMAppleProfilesToRemoveFunc func(ctx context.Context) ([]*fleet.MDMAppleProfilePayload, error) type BulkUpsertMDMAppleHostProfilesFunc func(ctx context.Context, payload []*fleet.MDMAppleBulkUpsertHostProfilePayload) error type GetMDMAppleProfilesContentsFunc func(ctx context.Context, profileIDs []uint) (map[uint]fleet.Mobileconfig, error) type UpdateHostMDMAppleProfileFunc func(ctx context.Context, profile *fleet.HostMDMAppleProfile) error type GetMDMAppleCommandRequestTypeFunc func(ctx context.Context, commandUUID string) (string, error) type DataStore struct { HealthCheckFunc HealthCheckFunc HealthCheckFuncInvoked bool NewCarveFunc NewCarveFunc NewCarveFuncInvoked bool UpdateCarveFunc UpdateCarveFunc UpdateCarveFuncInvoked bool CarveFunc CarveFunc CarveFuncInvoked bool CarveBySessionIdFunc CarveBySessionIdFunc CarveBySessionIdFuncInvoked bool CarveByNameFunc CarveByNameFunc CarveByNameFuncInvoked bool ListCarvesFunc ListCarvesFunc ListCarvesFuncInvoked bool NewBlockFunc NewBlockFunc NewBlockFuncInvoked bool GetBlockFunc GetBlockFunc GetBlockFuncInvoked bool CleanupCarvesFunc CleanupCarvesFunc CleanupCarvesFuncInvoked bool NewUserFunc NewUserFunc NewUserFuncInvoked bool ListUsersFunc ListUsersFunc ListUsersFuncInvoked bool UserByEmailFunc UserByEmailFunc UserByEmailFuncInvoked bool UserByIDFunc UserByIDFunc UserByIDFuncInvoked bool SaveUserFunc SaveUserFunc SaveUserFuncInvoked bool SaveUsersFunc SaveUsersFunc SaveUsersFuncInvoked bool DeleteUserFunc DeleteUserFunc DeleteUserFuncInvoked bool PendingEmailChangeFunc PendingEmailChangeFunc PendingEmailChangeFuncInvoked bool ConfirmPendingEmailChangeFunc ConfirmPendingEmailChangeFunc ConfirmPendingEmailChangeFuncInvoked bool ApplyQueriesFunc ApplyQueriesFunc ApplyQueriesFuncInvoked bool NewQueryFunc NewQueryFunc NewQueryFuncInvoked bool SaveQueryFunc SaveQueryFunc SaveQueryFuncInvoked bool DeleteQueryFunc DeleteQueryFunc DeleteQueryFuncInvoked bool DeleteQueriesFunc DeleteQueriesFunc DeleteQueriesFuncInvoked bool QueryFunc QueryFunc QueryFuncInvoked bool ListQueriesFunc ListQueriesFunc ListQueriesFuncInvoked bool QueryByNameFunc QueryByNameFunc QueryByNameFuncInvoked bool ObserverCanRunQueryFunc ObserverCanRunQueryFunc ObserverCanRunQueryFuncInvoked bool NewDistributedQueryCampaignFunc NewDistributedQueryCampaignFunc NewDistributedQueryCampaignFuncInvoked bool DistributedQueryCampaignFunc DistributedQueryCampaignFunc DistributedQueryCampaignFuncInvoked bool SaveDistributedQueryCampaignFunc SaveDistributedQueryCampaignFunc SaveDistributedQueryCampaignFuncInvoked bool DistributedQueryCampaignTargetIDsFunc DistributedQueryCampaignTargetIDsFunc DistributedQueryCampaignTargetIDsFuncInvoked bool NewDistributedQueryCampaignTargetFunc NewDistributedQueryCampaignTargetFunc NewDistributedQueryCampaignTargetFuncInvoked bool CleanupDistributedQueryCampaignsFunc CleanupDistributedQueryCampaignsFunc CleanupDistributedQueryCampaignsFuncInvoked bool DistributedQueryCampaignsForQueryFunc DistributedQueryCampaignsForQueryFunc DistributedQueryCampaignsForQueryFuncInvoked bool ApplyPackSpecsFunc ApplyPackSpecsFunc ApplyPackSpecsFuncInvoked bool GetPackSpecsFunc GetPackSpecsFunc GetPackSpecsFuncInvoked bool GetPackSpecFunc GetPackSpecFunc GetPackSpecFuncInvoked bool NewPackFunc NewPackFunc NewPackFuncInvoked bool SavePackFunc SavePackFunc SavePackFuncInvoked bool DeletePackFunc DeletePackFunc DeletePackFuncInvoked bool PackFunc PackFunc PackFuncInvoked bool ListPacksFunc ListPacksFunc ListPacksFuncInvoked bool PackByNameFunc PackByNameFunc PackByNameFuncInvoked bool ListPacksForHostFunc ListPacksForHostFunc ListPacksForHostFuncInvoked bool EnsureGlobalPackFunc EnsureGlobalPackFunc EnsureGlobalPackFuncInvoked bool EnsureTeamPackFunc EnsureTeamPackFunc EnsureTeamPackFuncInvoked bool ApplyLabelSpecsFunc ApplyLabelSpecsFunc ApplyLabelSpecsFuncInvoked bool GetLabelSpecsFunc GetLabelSpecsFunc GetLabelSpecsFuncInvoked bool GetLabelSpecFunc GetLabelSpecFunc GetLabelSpecFuncInvoked bool NewLabelFunc NewLabelFunc NewLabelFuncInvoked bool SaveLabelFunc SaveLabelFunc SaveLabelFuncInvoked bool DeleteLabelFunc DeleteLabelFunc DeleteLabelFuncInvoked bool LabelFunc LabelFunc LabelFuncInvoked bool ListLabelsFunc ListLabelsFunc ListLabelsFuncInvoked bool LabelsSummaryFunc LabelsSummaryFunc LabelsSummaryFuncInvoked bool LabelQueriesForHostFunc LabelQueriesForHostFunc LabelQueriesForHostFuncInvoked bool ListLabelsForHostFunc ListLabelsForHostFunc ListLabelsForHostFuncInvoked bool ListHostsInLabelFunc ListHostsInLabelFunc ListHostsInLabelFuncInvoked bool ListUniqueHostsInLabelsFunc ListUniqueHostsInLabelsFunc ListUniqueHostsInLabelsFuncInvoked bool SearchLabelsFunc SearchLabelsFunc SearchLabelsFuncInvoked bool LabelIDsByNameFunc LabelIDsByNameFunc LabelIDsByNameFuncInvoked bool AsyncBatchInsertLabelMembershipFunc AsyncBatchInsertLabelMembershipFunc AsyncBatchInsertLabelMembershipFuncInvoked bool AsyncBatchDeleteLabelMembershipFunc AsyncBatchDeleteLabelMembershipFunc AsyncBatchDeleteLabelMembershipFuncInvoked bool AsyncBatchUpdateLabelTimestampFunc AsyncBatchUpdateLabelTimestampFunc AsyncBatchUpdateLabelTimestampFuncInvoked bool NewHostFunc NewHostFunc NewHostFuncInvoked bool DeleteHostFunc DeleteHostFunc DeleteHostFuncInvoked bool HostFunc HostFunc HostFuncInvoked bool ListHostsFunc ListHostsFunc ListHostsFuncInvoked bool MarkHostsSeenFunc MarkHostsSeenFunc MarkHostsSeenFuncInvoked bool SearchHostsFunc SearchHostsFunc SearchHostsFuncInvoked bool EnrolledHostIDsFunc EnrolledHostIDsFunc EnrolledHostIDsFuncInvoked bool CountEnrolledHostsFunc CountEnrolledHostsFunc CountEnrolledHostsFuncInvoked bool CleanupIncomingHostsFunc CleanupIncomingHostsFunc CleanupIncomingHostsFuncInvoked bool GenerateHostStatusStatisticsFunc GenerateHostStatusStatisticsFunc GenerateHostStatusStatisticsFuncInvoked bool HostIDsByNameFunc HostIDsByNameFunc HostIDsByNameFuncInvoked bool HostIDsByOSIDFunc HostIDsByOSIDFunc HostIDsByOSIDFuncInvoked bool HostIDsByOSVersionFunc HostIDsByOSVersionFunc HostIDsByOSVersionFuncInvoked bool HostByIdentifierFunc HostByIdentifierFunc HostByIdentifierFuncInvoked bool AddHostsToTeamFunc AddHostsToTeamFunc AddHostsToTeamFuncInvoked bool TotalAndUnseenHostsSinceFunc TotalAndUnseenHostsSinceFunc TotalAndUnseenHostsSinceFuncInvoked bool DeleteHostsFunc DeleteHostsFunc DeleteHostsFuncInvoked bool CountHostsFunc CountHostsFunc CountHostsFuncInvoked bool CountHostsInLabelFunc CountHostsInLabelFunc CountHostsInLabelFuncInvoked bool ListHostDeviceMappingFunc ListHostDeviceMappingFunc ListHostDeviceMappingFuncInvoked bool ListHostBatteriesFunc ListHostBatteriesFunc ListHostBatteriesFuncInvoked bool LoadHostByDeviceAuthTokenFunc LoadHostByDeviceAuthTokenFunc LoadHostByDeviceAuthTokenFuncInvoked bool SetOrUpdateDeviceAuthTokenFunc SetOrUpdateDeviceAuthTokenFunc SetOrUpdateDeviceAuthTokenFuncInvoked bool FailingPoliciesCountFunc FailingPoliciesCountFunc FailingPoliciesCountFuncInvoked bool ListPoliciesForHostFunc ListPoliciesForHostFunc ListPoliciesForHostFuncInvoked bool GetHostMunkiVersionFunc GetHostMunkiVersionFunc GetHostMunkiVersionFuncInvoked bool GetHostMunkiIssuesFunc GetHostMunkiIssuesFunc GetHostMunkiIssuesFuncInvoked bool GetHostMDMFunc GetHostMDMFunc GetHostMDMFuncInvoked bool GetHostMDMCheckinInfoFunc GetHostMDMCheckinInfoFunc GetHostMDMCheckinInfoFuncInvoked bool AggregatedMunkiVersionFunc AggregatedMunkiVersionFunc AggregatedMunkiVersionFuncInvoked bool AggregatedMunkiIssuesFunc AggregatedMunkiIssuesFunc AggregatedMunkiIssuesFuncInvoked bool AggregatedMDMStatusFunc AggregatedMDMStatusFunc AggregatedMDMStatusFuncInvoked bool AggregatedMDMSolutionsFunc AggregatedMDMSolutionsFunc AggregatedMDMSolutionsFuncInvoked bool GenerateAggregatedMunkiAndMDMFunc GenerateAggregatedMunkiAndMDMFunc GenerateAggregatedMunkiAndMDMFuncInvoked bool GetMunkiIssueFunc GetMunkiIssueFunc GetMunkiIssueFuncInvoked bool GetMDMSolutionFunc GetMDMSolutionFunc GetMDMSolutionFuncInvoked bool OSVersionsFunc OSVersionsFunc OSVersionsFuncInvoked bool UpdateOSVersionsFunc UpdateOSVersionsFunc UpdateOSVersionsFuncInvoked bool CountHostsInTargetsFunc CountHostsInTargetsFunc CountHostsInTargetsFuncInvoked bool HostIDsInTargetsFunc HostIDsInTargetsFunc HostIDsInTargetsFuncInvoked bool NewPasswordResetRequestFunc NewPasswordResetRequestFunc NewPasswordResetRequestFuncInvoked bool DeletePasswordResetRequestsForUserFunc DeletePasswordResetRequestsForUserFunc DeletePasswordResetRequestsForUserFuncInvoked bool FindPasswordResetByTokenFunc FindPasswordResetByTokenFunc FindPasswordResetByTokenFuncInvoked bool CleanupExpiredPasswordResetRequestsFunc CleanupExpiredPasswordResetRequestsFunc CleanupExpiredPasswordResetRequestsFuncInvoked bool SessionByKeyFunc SessionByKeyFunc SessionByKeyFuncInvoked bool SessionByIDFunc SessionByIDFunc SessionByIDFuncInvoked bool ListSessionsForUserFunc ListSessionsForUserFunc ListSessionsForUserFuncInvoked bool NewSessionFunc NewSessionFunc NewSessionFuncInvoked bool DestroySessionFunc DestroySessionFunc DestroySessionFuncInvoked bool DestroyAllSessionsForUserFunc DestroyAllSessionsForUserFunc DestroyAllSessionsForUserFuncInvoked bool MarkSessionAccessedFunc MarkSessionAccessedFunc MarkSessionAccessedFuncInvoked bool NewAppConfigFunc NewAppConfigFunc NewAppConfigFuncInvoked bool AppConfigFunc AppConfigFunc AppConfigFuncInvoked bool SaveAppConfigFunc SaveAppConfigFunc SaveAppConfigFuncInvoked bool GetEnrollSecretsFunc GetEnrollSecretsFunc GetEnrollSecretsFuncInvoked bool ApplyEnrollSecretsFunc ApplyEnrollSecretsFunc ApplyEnrollSecretsFuncInvoked bool NewInviteFunc NewInviteFunc NewInviteFuncInvoked bool ListInvitesFunc ListInvitesFunc ListInvitesFuncInvoked bool InviteFunc InviteFunc InviteFuncInvoked bool InviteByEmailFunc InviteByEmailFunc InviteByEmailFuncInvoked bool InviteByTokenFunc InviteByTokenFunc InviteByTokenFuncInvoked bool DeleteInviteFunc DeleteInviteFunc DeleteInviteFuncInvoked bool UpdateInviteFunc UpdateInviteFunc UpdateInviteFuncInvoked bool ListScheduledQueriesInPackWithStatsFunc ListScheduledQueriesInPackWithStatsFunc ListScheduledQueriesInPackWithStatsFuncInvoked bool NewScheduledQueryFunc NewScheduledQueryFunc NewScheduledQueryFuncInvoked bool SaveScheduledQueryFunc SaveScheduledQueryFunc SaveScheduledQueryFuncInvoked bool DeleteScheduledQueryFunc DeleteScheduledQueryFunc DeleteScheduledQueryFuncInvoked bool ScheduledQueryFunc ScheduledQueryFunc ScheduledQueryFuncInvoked bool CleanupExpiredHostsFunc CleanupExpiredHostsFunc CleanupExpiredHostsFuncInvoked bool ScheduledQueryIDsByNameFunc ScheduledQueryIDsByNameFunc ScheduledQueryIDsByNameFuncInvoked bool NewTeamFunc NewTeamFunc NewTeamFuncInvoked bool SaveTeamFunc SaveTeamFunc SaveTeamFuncInvoked bool TeamFunc TeamFunc TeamFuncInvoked bool DeleteTeamFunc DeleteTeamFunc DeleteTeamFuncInvoked bool TeamByNameFunc TeamByNameFunc TeamByNameFuncInvoked bool ListTeamsFunc ListTeamsFunc ListTeamsFuncInvoked bool TeamsSummaryFunc TeamsSummaryFunc TeamsSummaryFuncInvoked bool SearchTeamsFunc SearchTeamsFunc SearchTeamsFuncInvoked bool TeamEnrollSecretsFunc TeamEnrollSecretsFunc TeamEnrollSecretsFuncInvoked bool DeleteIntegrationsFromTeamsFunc DeleteIntegrationsFromTeamsFunc DeleteIntegrationsFromTeamsFuncInvoked bool ListSoftwareForVulnDetectionFunc ListSoftwareForVulnDetectionFunc ListSoftwareForVulnDetectionFuncInvoked bool ListSoftwareVulnerabilitiesByHostIDsSourceFunc ListSoftwareVulnerabilitiesByHostIDsSourceFunc ListSoftwareVulnerabilitiesByHostIDsSourceFuncInvoked bool LoadHostSoftwareFunc LoadHostSoftwareFunc LoadHostSoftwareFuncInvoked bool ListSoftwareBySourceIterFunc ListSoftwareBySourceIterFunc ListSoftwareBySourceIterFuncInvoked bool AllSoftwareWithoutCPEIteratorFunc AllSoftwareWithoutCPEIteratorFunc AllSoftwareWithoutCPEIteratorFuncInvoked bool AddCPEForSoftwareFunc AddCPEForSoftwareFunc AddCPEForSoftwareFuncInvoked bool ListSoftwareCPEsFunc ListSoftwareCPEsFunc ListSoftwareCPEsFuncInvoked bool InsertSoftwareVulnerabilitiesFunc InsertSoftwareVulnerabilitiesFunc InsertSoftwareVulnerabilitiesFuncInvoked bool SoftwareByIDFunc SoftwareByIDFunc SoftwareByIDFuncInvoked bool ListSoftwareByHostIDShortFunc ListSoftwareByHostIDShortFunc ListSoftwareByHostIDShortFuncInvoked bool SyncHostsSoftwareFunc SyncHostsSoftwareFunc SyncHostsSoftwareFuncInvoked bool HostsBySoftwareIDsFunc HostsBySoftwareIDsFunc HostsBySoftwareIDsFuncInvoked bool HostsByCVEFunc HostsByCVEFunc HostsByCVEFuncInvoked bool InsertCVEMetaFunc InsertCVEMetaFunc InsertCVEMetaFuncInvoked bool ListCVEsFunc ListCVEsFunc ListCVEsFuncInvoked bool ListOperatingSystemsFunc ListOperatingSystemsFunc ListOperatingSystemsFuncInvoked bool UpdateHostOperatingSystemFunc UpdateHostOperatingSystemFunc UpdateHostOperatingSystemFuncInvoked bool CleanupHostOperatingSystemsFunc CleanupHostOperatingSystemsFunc CleanupHostOperatingSystemsFuncInvoked bool UpdateHostTablesOnMDMUnenrollFunc UpdateHostTablesOnMDMUnenrollFunc UpdateHostTablesOnMDMUnenrollFuncInvoked bool NewActivityFunc NewActivityFunc NewActivityFuncInvoked bool ListActivitiesFunc ListActivitiesFunc ListActivitiesFuncInvoked bool MarkActivitiesAsStreamedFunc MarkActivitiesAsStreamedFunc MarkActivitiesAsStreamedFuncInvoked bool ShouldSendStatisticsFunc ShouldSendStatisticsFunc ShouldSendStatisticsFuncInvoked bool RecordStatisticsSentFunc RecordStatisticsSentFunc RecordStatisticsSentFuncInvoked bool CleanupStatisticsFunc CleanupStatisticsFunc CleanupStatisticsFuncInvoked bool ApplyPolicySpecsFunc ApplyPolicySpecsFunc ApplyPolicySpecsFuncInvoked bool NewGlobalPolicyFunc NewGlobalPolicyFunc NewGlobalPolicyFuncInvoked bool PolicyFunc PolicyFunc PolicyFuncInvoked bool SavePolicyFunc SavePolicyFunc SavePolicyFuncInvoked bool ListGlobalPoliciesFunc ListGlobalPoliciesFunc ListGlobalPoliciesFuncInvoked bool PoliciesByIDFunc PoliciesByIDFunc PoliciesByIDFuncInvoked bool DeleteGlobalPoliciesFunc DeleteGlobalPoliciesFunc DeleteGlobalPoliciesFuncInvoked bool PolicyQueriesForHostFunc PolicyQueriesForHostFunc PolicyQueriesForHostFuncInvoked bool AsyncBatchInsertPolicyMembershipFunc AsyncBatchInsertPolicyMembershipFunc AsyncBatchInsertPolicyMembershipFuncInvoked bool AsyncBatchUpdatePolicyTimestampFunc AsyncBatchUpdatePolicyTimestampFunc AsyncBatchUpdatePolicyTimestampFuncInvoked bool MigrateTablesFunc MigrateTablesFunc MigrateTablesFuncInvoked bool MigrateDataFunc MigrateDataFunc MigrateDataFuncInvoked bool MigrationStatusFunc MigrationStatusFunc MigrationStatusFuncInvoked bool ListSoftwareFunc ListSoftwareFunc ListSoftwareFuncInvoked bool CountSoftwareFunc CountSoftwareFunc CountSoftwareFuncInvoked bool DeleteSoftwareVulnerabilitiesFunc DeleteSoftwareVulnerabilitiesFunc DeleteSoftwareVulnerabilitiesFuncInvoked bool NewTeamPolicyFunc NewTeamPolicyFunc NewTeamPolicyFuncInvoked bool ListTeamPoliciesFunc ListTeamPoliciesFunc ListTeamPoliciesFuncInvoked bool DeleteTeamPoliciesFunc DeleteTeamPoliciesFunc DeleteTeamPoliciesFuncInvoked bool TeamPolicyFunc TeamPolicyFunc TeamPolicyFuncInvoked bool CleanupPolicyMembershipFunc CleanupPolicyMembershipFunc CleanupPolicyMembershipFuncInvoked bool IncrementPolicyViolationDaysFunc IncrementPolicyViolationDaysFunc IncrementPolicyViolationDaysFuncInvoked bool InitializePolicyViolationDaysFunc InitializePolicyViolationDaysFunc InitializePolicyViolationDaysFuncInvoked bool LockFunc LockFunc LockFuncInvoked bool UnlockFunc UnlockFunc UnlockFuncInvoked bool DBLocksFunc DBLocksFunc DBLocksFuncInvoked bool GetLatestCronStatsFunc GetLatestCronStatsFunc GetLatestCronStatsFuncInvoked bool InsertCronStatsFunc InsertCronStatsFunc InsertCronStatsFuncInvoked bool UpdateCronStatsFunc UpdateCronStatsFunc UpdateCronStatsFuncInvoked bool UpdateAllCronStatsForInstanceFunc UpdateAllCronStatsForInstanceFunc UpdateAllCronStatsForInstanceFuncInvoked bool CleanupCronStatsFunc CleanupCronStatsFunc CleanupCronStatsFuncInvoked bool UpdateScheduledQueryAggregatedStatsFunc UpdateScheduledQueryAggregatedStatsFunc UpdateScheduledQueryAggregatedStatsFuncInvoked bool UpdateQueryAggregatedStatsFunc UpdateQueryAggregatedStatsFunc UpdateQueryAggregatedStatsFuncInvoked bool LoadHostByNodeKeyFunc LoadHostByNodeKeyFunc LoadHostByNodeKeyFuncInvoked bool LoadHostByOrbitNodeKeyFunc LoadHostByOrbitNodeKeyFunc LoadHostByOrbitNodeKeyFuncInvoked bool HostLiteFunc HostLiteFunc HostLiteFuncInvoked bool UpdateHostOsqueryIntervalsFunc UpdateHostOsqueryIntervalsFunc UpdateHostOsqueryIntervalsFuncInvoked bool TeamAgentOptionsFunc TeamAgentOptionsFunc TeamAgentOptionsFuncInvoked bool TeamFeaturesFunc TeamFeaturesFunc TeamFeaturesFuncInvoked bool TeamMDMConfigFunc TeamMDMConfigFunc TeamMDMConfigFuncInvoked bool SaveHostPackStatsFunc SaveHostPackStatsFunc SaveHostPackStatsFuncInvoked bool AsyncBatchSaveHostsScheduledQueryStatsFunc AsyncBatchSaveHostsScheduledQueryStatsFunc AsyncBatchSaveHostsScheduledQueryStatsFuncInvoked bool UpdateHostSoftwareFunc UpdateHostSoftwareFunc UpdateHostSoftwareFuncInvoked bool UpdateHostFunc UpdateHostFunc UpdateHostFuncInvoked bool ListScheduledQueriesInPackFunc ListScheduledQueriesInPackFunc ListScheduledQueriesInPackFuncInvoked bool UpdateHostRefetchRequestedFunc UpdateHostRefetchRequestedFunc UpdateHostRefetchRequestedFuncInvoked bool FlippingPoliciesForHostFunc FlippingPoliciesForHostFunc FlippingPoliciesForHostFuncInvoked bool RecordPolicyQueryExecutionsFunc RecordPolicyQueryExecutionsFunc RecordPolicyQueryExecutionsFuncInvoked bool RecordLabelQueryExecutionsFunc RecordLabelQueryExecutionsFunc RecordLabelQueryExecutionsFuncInvoked bool SaveHostUsersFunc SaveHostUsersFunc SaveHostUsersFuncInvoked bool SaveHostAdditionalFunc SaveHostAdditionalFunc SaveHostAdditionalFuncInvoked bool SetOrUpdateMunkiInfoFunc SetOrUpdateMunkiInfoFunc SetOrUpdateMunkiInfoFuncInvoked bool SetOrUpdateMDMDataFunc SetOrUpdateMDMDataFunc SetOrUpdateMDMDataFuncInvoked bool SetOrUpdateHostDisksSpaceFunc SetOrUpdateHostDisksSpaceFunc SetOrUpdateHostDisksSpaceFuncInvoked bool SetOrUpdateHostDisksEncryptionFunc SetOrUpdateHostDisksEncryptionFunc SetOrUpdateHostDisksEncryptionFuncInvoked bool SetOrUpdateHostDiskEncryptionKeyFunc SetOrUpdateHostDiskEncryptionKeyFunc SetOrUpdateHostDiskEncryptionKeyFuncInvoked bool GetUnverifiedDiskEncryptionKeysFunc GetUnverifiedDiskEncryptionKeysFunc GetUnverifiedDiskEncryptionKeysFuncInvoked bool SetHostsDiskEncryptionKeyStatusFunc SetHostsDiskEncryptionKeyStatusFunc SetHostsDiskEncryptionKeyStatusFuncInvoked bool GetHostDiskEncryptionKeyFunc GetHostDiskEncryptionKeyFunc GetHostDiskEncryptionKeyFuncInvoked bool SetOrUpdateHostOrbitInfoFunc SetOrUpdateHostOrbitInfoFunc SetOrUpdateHostOrbitInfoFuncInvoked bool ReplaceHostDeviceMappingFunc ReplaceHostDeviceMappingFunc ReplaceHostDeviceMappingFuncInvoked bool ReplaceHostBatteriesFunc ReplaceHostBatteriesFunc ReplaceHostBatteriesFuncInvoked bool VerifyEnrollSecretFunc VerifyEnrollSecretFunc VerifyEnrollSecretFuncInvoked bool EnrollHostFunc EnrollHostFunc EnrollHostFuncInvoked bool EnrollOrbitFunc EnrollOrbitFunc EnrollOrbitFuncInvoked bool SerialUpdateHostFunc SerialUpdateHostFunc SerialUpdateHostFuncInvoked bool NewJobFunc NewJobFunc NewJobFuncInvoked bool GetQueuedJobsFunc GetQueuedJobsFunc GetQueuedJobsFuncInvoked bool UpdateJobFunc UpdateJobFunc UpdateJobFuncInvoked bool InnoDBStatusFunc InnoDBStatusFunc InnoDBStatusFuncInvoked bool ProcessListFunc ProcessListFunc ProcessListFuncInvoked bool ListWindowsUpdatesByHostIDFunc ListWindowsUpdatesByHostIDFunc ListWindowsUpdatesByHostIDFuncInvoked bool InsertWindowsUpdatesFunc InsertWindowsUpdatesFunc InsertWindowsUpdatesFuncInvoked bool ListOSVulnerabilitiesFunc ListOSVulnerabilitiesFunc ListOSVulnerabilitiesFuncInvoked bool InsertOSVulnerabilitiesFunc InsertOSVulnerabilitiesFunc InsertOSVulnerabilitiesFuncInvoked bool DeleteOSVulnerabilitiesFunc DeleteOSVulnerabilitiesFunc DeleteOSVulnerabilitiesFuncInvoked bool NewMDMAppleConfigProfileFunc NewMDMAppleConfigProfileFunc NewMDMAppleConfigProfileFuncInvoked bool GetMDMAppleConfigProfileFunc GetMDMAppleConfigProfileFunc GetMDMAppleConfigProfileFuncInvoked bool ListMDMAppleConfigProfilesFunc ListMDMAppleConfigProfilesFunc ListMDMAppleConfigProfilesFuncInvoked bool DeleteMDMAppleConfigProfileFunc DeleteMDMAppleConfigProfileFunc DeleteMDMAppleConfigProfileFuncInvoked bool GetHostMDMProfilesFunc GetHostMDMProfilesFunc GetHostMDMProfilesFuncInvoked bool NewMDMAppleEnrollmentProfileFunc NewMDMAppleEnrollmentProfileFunc NewMDMAppleEnrollmentProfileFuncInvoked bool GetMDMAppleEnrollmentProfileByTokenFunc GetMDMAppleEnrollmentProfileByTokenFunc GetMDMAppleEnrollmentProfileByTokenFuncInvoked bool ListMDMAppleEnrollmentProfilesFunc ListMDMAppleEnrollmentProfilesFunc ListMDMAppleEnrollmentProfilesFuncInvoked bool GetMDMAppleCommandResultsFunc GetMDMAppleCommandResultsFunc GetMDMAppleCommandResultsFuncInvoked bool NewMDMAppleInstallerFunc NewMDMAppleInstallerFunc NewMDMAppleInstallerFuncInvoked bool MDMAppleInstallerFunc MDMAppleInstallerFunc MDMAppleInstallerFuncInvoked bool MDMAppleInstallerDetailsByIDFunc MDMAppleInstallerDetailsByIDFunc MDMAppleInstallerDetailsByIDFuncInvoked bool DeleteMDMAppleInstallerFunc DeleteMDMAppleInstallerFunc DeleteMDMAppleInstallerFuncInvoked bool MDMAppleInstallerDetailsByTokenFunc MDMAppleInstallerDetailsByTokenFunc MDMAppleInstallerDetailsByTokenFuncInvoked bool ListMDMAppleInstallersFunc ListMDMAppleInstallersFunc ListMDMAppleInstallersFuncInvoked bool BatchSetMDMAppleProfilesFunc BatchSetMDMAppleProfilesFunc BatchSetMDMAppleProfilesFuncInvoked bool MDMAppleListDevicesFunc MDMAppleListDevicesFunc MDMAppleListDevicesFuncInvoked bool IngestMDMAppleDevicesFromDEPSyncFunc IngestMDMAppleDevicesFromDEPSyncFunc IngestMDMAppleDevicesFromDEPSyncFuncInvoked bool IngestMDMAppleDeviceFromCheckinFunc IngestMDMAppleDeviceFromCheckinFunc IngestMDMAppleDeviceFromCheckinFuncInvoked bool GetNanoMDMEnrollmentStatusFunc GetNanoMDMEnrollmentStatusFunc GetNanoMDMEnrollmentStatusFuncInvoked bool IncreasePolicyAutomationIterationFunc IncreasePolicyAutomationIterationFunc IncreasePolicyAutomationIterationFuncInvoked bool OutdatedAutomationBatchFunc OutdatedAutomationBatchFunc OutdatedAutomationBatchFuncInvoked bool ListMDMAppleProfilesToInstallFunc ListMDMAppleProfilesToInstallFunc ListMDMAppleProfilesToInstallFuncInvoked bool ListMDMAppleProfilesToRemoveFunc ListMDMAppleProfilesToRemoveFunc ListMDMAppleProfilesToRemoveFuncInvoked bool BulkUpsertMDMAppleHostProfilesFunc BulkUpsertMDMAppleHostProfilesFunc BulkUpsertMDMAppleHostProfilesFuncInvoked bool GetMDMAppleProfilesContentsFunc GetMDMAppleProfilesContentsFunc GetMDMAppleProfilesContentsFuncInvoked bool UpdateHostMDMAppleProfileFunc UpdateHostMDMAppleProfileFunc UpdateHostMDMAppleProfileFuncInvoked bool GetMDMAppleCommandRequestTypeFunc GetMDMAppleCommandRequestTypeFunc GetMDMAppleCommandRequestTypeFuncInvoked bool mu sync.Mutex } func (s *DataStore) HealthCheck() error { s.mu.Lock() s.HealthCheckFuncInvoked = true s.mu.Unlock() return s.HealthCheckFunc() } func (s *DataStore) NewCarve(ctx context.Context, metadata *fleet.CarveMetadata) (*fleet.CarveMetadata, error) { s.mu.Lock() s.NewCarveFuncInvoked = true s.mu.Unlock() return s.NewCarveFunc(ctx, metadata) } func (s *DataStore) UpdateCarve(ctx context.Context, metadata *fleet.CarveMetadata) error { s.mu.Lock() s.UpdateCarveFuncInvoked = true s.mu.Unlock() return s.UpdateCarveFunc(ctx, metadata) } func (s *DataStore) Carve(ctx context.Context, carveId int64) (*fleet.CarveMetadata, error) { s.mu.Lock() s.CarveFuncInvoked = true s.mu.Unlock() return s.CarveFunc(ctx, carveId) } func (s *DataStore) CarveBySessionId(ctx context.Context, sessionId string) (*fleet.CarveMetadata, error) { s.mu.Lock() s.CarveBySessionIdFuncInvoked = true s.mu.Unlock() return s.CarveBySessionIdFunc(ctx, sessionId) } func (s *DataStore) CarveByName(ctx context.Context, name string) (*fleet.CarveMetadata, error) { s.mu.Lock() s.CarveByNameFuncInvoked = true s.mu.Unlock() return s.CarveByNameFunc(ctx, name) } func (s *DataStore) ListCarves(ctx context.Context, opt fleet.CarveListOptions) ([]*fleet.CarveMetadata, error) { s.mu.Lock() s.ListCarvesFuncInvoked = true s.mu.Unlock() return s.ListCarvesFunc(ctx, opt) } func (s *DataStore) NewBlock(ctx context.Context, metadata *fleet.CarveMetadata, blockId int64, data []byte) error { s.mu.Lock() s.NewBlockFuncInvoked = true s.mu.Unlock() return s.NewBlockFunc(ctx, metadata, blockId, data) } func (s *DataStore) GetBlock(ctx context.Context, metadata *fleet.CarveMetadata, blockId int64) ([]byte, error) { s.mu.Lock() s.GetBlockFuncInvoked = true s.mu.Unlock() return s.GetBlockFunc(ctx, metadata, blockId) } func (s *DataStore) CleanupCarves(ctx context.Context, now time.Time) (expired int, err error) { s.mu.Lock() s.CleanupCarvesFuncInvoked = true s.mu.Unlock() return s.CleanupCarvesFunc(ctx, now) } func (s *DataStore) NewUser(ctx context.Context, user *fleet.User) (*fleet.User, error) { s.mu.Lock() s.NewUserFuncInvoked = true s.mu.Unlock() return s.NewUserFunc(ctx, user) } func (s *DataStore) ListUsers(ctx context.Context, opt fleet.UserListOptions) ([]*fleet.User, error) { s.mu.Lock() s.ListUsersFuncInvoked = true s.mu.Unlock() return s.ListUsersFunc(ctx, opt) } func (s *DataStore) UserByEmail(ctx context.Context, email string) (*fleet.User, error) { s.mu.Lock() s.UserByEmailFuncInvoked = true s.mu.Unlock() return s.UserByEmailFunc(ctx, email) } func (s *DataStore) UserByID(ctx context.Context, id uint) (*fleet.User, error) { s.mu.Lock() s.UserByIDFuncInvoked = true s.mu.Unlock() return s.UserByIDFunc(ctx, id) } func (s *DataStore) SaveUser(ctx context.Context, user *fleet.User) error { s.mu.Lock() s.SaveUserFuncInvoked = true s.mu.Unlock() return s.SaveUserFunc(ctx, user) } func (s *DataStore) SaveUsers(ctx context.Context, users []*fleet.User) error { s.mu.Lock() s.SaveUsersFuncInvoked = true s.mu.Unlock() return s.SaveUsersFunc(ctx, users) } func (s *DataStore) DeleteUser(ctx context.Context, id uint) error { s.mu.Lock() s.DeleteUserFuncInvoked = true s.mu.Unlock() return s.DeleteUserFunc(ctx, id) } func (s *DataStore) PendingEmailChange(ctx context.Context, userID uint, newEmail string, token string) error { s.mu.Lock() s.PendingEmailChangeFuncInvoked = true s.mu.Unlock() return s.PendingEmailChangeFunc(ctx, userID, newEmail, token) } func (s *DataStore) ConfirmPendingEmailChange(ctx context.Context, userID uint, token string) (string, error) { s.mu.Lock() s.ConfirmPendingEmailChangeFuncInvoked = true s.mu.Unlock() return s.ConfirmPendingEmailChangeFunc(ctx, userID, token) } func (s *DataStore) ApplyQueries(ctx context.Context, authorID uint, queries []*fleet.Query) error { s.mu.Lock() s.ApplyQueriesFuncInvoked = true s.mu.Unlock() return s.ApplyQueriesFunc(ctx, authorID, queries) } func (s *DataStore) NewQuery(ctx context.Context, query *fleet.Query, opts ...fleet.OptionalArg) (*fleet.Query, error) { s.mu.Lock() s.NewQueryFuncInvoked = true s.mu.Unlock() return s.NewQueryFunc(ctx, query, opts...) } func (s *DataStore) SaveQuery(ctx context.Context, query *fleet.Query) error { s.mu.Lock() s.SaveQueryFuncInvoked = true s.mu.Unlock() return s.SaveQueryFunc(ctx, query) } func (s *DataStore) DeleteQuery(ctx context.Context, name string) error { s.mu.Lock() s.DeleteQueryFuncInvoked = true s.mu.Unlock() return s.DeleteQueryFunc(ctx, name) } func (s *DataStore) DeleteQueries(ctx context.Context, ids []uint) (uint, error) { s.mu.Lock() s.DeleteQueriesFuncInvoked = true s.mu.Unlock() return s.DeleteQueriesFunc(ctx, ids) } func (s *DataStore) Query(ctx context.Context, id uint) (*fleet.Query, error) { s.mu.Lock() s.QueryFuncInvoked = true s.mu.Unlock() return s.QueryFunc(ctx, id) } func (s *DataStore) ListQueries(ctx context.Context, opt fleet.ListQueryOptions) ([]*fleet.Query, error) { s.mu.Lock() s.ListQueriesFuncInvoked = true s.mu.Unlock() return s.ListQueriesFunc(ctx, opt) } func (s *DataStore) QueryByName(ctx context.Context, name string, opts ...fleet.OptionalArg) (*fleet.Query, error) { s.mu.Lock() s.QueryByNameFuncInvoked = true s.mu.Unlock() return s.QueryByNameFunc(ctx, name, opts...) } func (s *DataStore) ObserverCanRunQuery(ctx context.Context, queryID uint) (bool, error) { s.mu.Lock() s.ObserverCanRunQueryFuncInvoked = true s.mu.Unlock() return s.ObserverCanRunQueryFunc(ctx, queryID) } func (s *DataStore) NewDistributedQueryCampaign(ctx context.Context, camp *fleet.DistributedQueryCampaign) (*fleet.DistributedQueryCampaign, error) { s.mu.Lock() s.NewDistributedQueryCampaignFuncInvoked = true s.mu.Unlock() return s.NewDistributedQueryCampaignFunc(ctx, camp) } func (s *DataStore) DistributedQueryCampaign(ctx context.Context, id uint) (*fleet.DistributedQueryCampaign, error) { s.mu.Lock() s.DistributedQueryCampaignFuncInvoked = true s.mu.Unlock() return s.DistributedQueryCampaignFunc(ctx, id) } func (s *DataStore) SaveDistributedQueryCampaign(ctx context.Context, camp *fleet.DistributedQueryCampaign) error { s.mu.Lock() s.SaveDistributedQueryCampaignFuncInvoked = true s.mu.Unlock() return s.SaveDistributedQueryCampaignFunc(ctx, camp) } func (s *DataStore) DistributedQueryCampaignTargetIDs(ctx context.Context, id uint) (targets *fleet.HostTargets, err error) { s.mu.Lock() s.DistributedQueryCampaignTargetIDsFuncInvoked = true s.mu.Unlock() return s.DistributedQueryCampaignTargetIDsFunc(ctx, id) } func (s *DataStore) NewDistributedQueryCampaignTarget(ctx context.Context, target *fleet.DistributedQueryCampaignTarget) (*fleet.DistributedQueryCampaignTarget, error) { s.mu.Lock() s.NewDistributedQueryCampaignTargetFuncInvoked = true s.mu.Unlock() return s.NewDistributedQueryCampaignTargetFunc(ctx, target) } func (s *DataStore) CleanupDistributedQueryCampaigns(ctx context.Context, now time.Time) (expired uint, err error) { s.mu.Lock() s.CleanupDistributedQueryCampaignsFuncInvoked = true s.mu.Unlock() return s.CleanupDistributedQueryCampaignsFunc(ctx, now) } func (s *DataStore) DistributedQueryCampaignsForQuery(ctx context.Context, queryID uint) ([]*fleet.DistributedQueryCampaign, error) { s.mu.Lock() s.DistributedQueryCampaignsForQueryFuncInvoked = true s.mu.Unlock() return s.DistributedQueryCampaignsForQueryFunc(ctx, queryID) } func (s *DataStore) ApplyPackSpecs(ctx context.Context, specs []*fleet.PackSpec) error { s.mu.Lock() s.ApplyPackSpecsFuncInvoked = true s.mu.Unlock() return s.ApplyPackSpecsFunc(ctx, specs) } func (s *DataStore) GetPackSpecs(ctx context.Context) ([]*fleet.PackSpec, error) { s.mu.Lock() s.GetPackSpecsFuncInvoked = true s.mu.Unlock() return s.GetPackSpecsFunc(ctx) } func (s *DataStore) GetPackSpec(ctx context.Context, name string) (*fleet.PackSpec, error) { s.mu.Lock() s.GetPackSpecFuncInvoked = true s.mu.Unlock() return s.GetPackSpecFunc(ctx, name) } func (s *DataStore) NewPack(ctx context.Context, pack *fleet.Pack, opts ...fleet.OptionalArg) (*fleet.Pack, error) { s.mu.Lock() s.NewPackFuncInvoked = true s.mu.Unlock() return s.NewPackFunc(ctx, pack, opts...) } func (s *DataStore) SavePack(ctx context.Context, pack *fleet.Pack) error { s.mu.Lock() s.SavePackFuncInvoked = true s.mu.Unlock() return s.SavePackFunc(ctx, pack) } func (s *DataStore) DeletePack(ctx context.Context, name string) error { s.mu.Lock() s.DeletePackFuncInvoked = true s.mu.Unlock() return s.DeletePackFunc(ctx, name) } func (s *DataStore) Pack(ctx context.Context, pid uint) (*fleet.Pack, error) { s.mu.Lock() s.PackFuncInvoked = true s.mu.Unlock() return s.PackFunc(ctx, pid) } func (s *DataStore) ListPacks(ctx context.Context, opt fleet.PackListOptions) ([]*fleet.Pack, error) { s.mu.Lock() s.ListPacksFuncInvoked = true s.mu.Unlock() return s.ListPacksFunc(ctx, opt) } func (s *DataStore) PackByName(ctx context.Context, name string, opts ...fleet.OptionalArg) (*fleet.Pack, bool, error) { s.mu.Lock() s.PackByNameFuncInvoked = true s.mu.Unlock() return s.PackByNameFunc(ctx, name, opts...) } func (s *DataStore) ListPacksForHost(ctx context.Context, hid uint) (packs []*fleet.Pack, err error) { s.mu.Lock() s.ListPacksForHostFuncInvoked = true s.mu.Unlock() return s.ListPacksForHostFunc(ctx, hid) } func (s *DataStore) EnsureGlobalPack(ctx context.Context) (*fleet.Pack, error) { s.mu.Lock() s.EnsureGlobalPackFuncInvoked = true s.mu.Unlock() return s.EnsureGlobalPackFunc(ctx) } func (s *DataStore) EnsureTeamPack(ctx context.Context, teamID uint) (*fleet.Pack, error) { s.mu.Lock() s.EnsureTeamPackFuncInvoked = true s.mu.Unlock() return s.EnsureTeamPackFunc(ctx, teamID) } func (s *DataStore) ApplyLabelSpecs(ctx context.Context, specs []*fleet.LabelSpec) error { s.mu.Lock() s.ApplyLabelSpecsFuncInvoked = true s.mu.Unlock() return s.ApplyLabelSpecsFunc(ctx, specs) } func (s *DataStore) GetLabelSpecs(ctx context.Context) ([]*fleet.LabelSpec, error) { s.mu.Lock() s.GetLabelSpecsFuncInvoked = true s.mu.Unlock() return s.GetLabelSpecsFunc(ctx) } func (s *DataStore) GetLabelSpec(ctx context.Context, name string) (*fleet.LabelSpec, error) { s.mu.Lock() s.GetLabelSpecFuncInvoked = true s.mu.Unlock() return s.GetLabelSpecFunc(ctx, name) } func (s *DataStore) NewLabel(ctx context.Context, Label *fleet.Label, opts ...fleet.OptionalArg) (*fleet.Label, error) { s.mu.Lock() s.NewLabelFuncInvoked = true s.mu.Unlock() return s.NewLabelFunc(ctx, Label, opts...) } func (s *DataStore) SaveLabel(ctx context.Context, label *fleet.Label) (*fleet.Label, error) { s.mu.Lock() s.SaveLabelFuncInvoked = true s.mu.Unlock() return s.SaveLabelFunc(ctx, label) } func (s *DataStore) DeleteLabel(ctx context.Context, name string) error { s.mu.Lock() s.DeleteLabelFuncInvoked = true s.mu.Unlock() return s.DeleteLabelFunc(ctx, name) } func (s *DataStore) Label(ctx context.Context, lid uint) (*fleet.Label, error) { s.mu.Lock() s.LabelFuncInvoked = true s.mu.Unlock() return s.LabelFunc(ctx, lid) } func (s *DataStore) ListLabels(ctx context.Context, filter fleet.TeamFilter, opt fleet.ListOptions) ([]*fleet.Label, error) { s.mu.Lock() s.ListLabelsFuncInvoked = true s.mu.Unlock() return s.ListLabelsFunc(ctx, filter, opt) } func (s *DataStore) LabelsSummary(ctx context.Context) ([]*fleet.LabelSummary, error) { s.mu.Lock() s.LabelsSummaryFuncInvoked = true s.mu.Unlock() return s.LabelsSummaryFunc(ctx) } func (s *DataStore) LabelQueriesForHost(ctx context.Context, host *fleet.Host) (map[string]string, error) { s.mu.Lock() s.LabelQueriesForHostFuncInvoked = true s.mu.Unlock() return s.LabelQueriesForHostFunc(ctx, host) } func (s *DataStore) ListLabelsForHost(ctx context.Context, hid uint) ([]*fleet.Label, error) { s.mu.Lock() s.ListLabelsForHostFuncInvoked = true s.mu.Unlock() return s.ListLabelsForHostFunc(ctx, hid) } func (s *DataStore) ListHostsInLabel(ctx context.Context, filter fleet.TeamFilter, lid uint, opt fleet.HostListOptions) ([]*fleet.Host, error) { s.mu.Lock() s.ListHostsInLabelFuncInvoked = true s.mu.Unlock() return s.ListHostsInLabelFunc(ctx, filter, lid, opt) } func (s *DataStore) ListUniqueHostsInLabels(ctx context.Context, filter fleet.TeamFilter, labels []uint) ([]*fleet.Host, error) { s.mu.Lock() s.ListUniqueHostsInLabelsFuncInvoked = true s.mu.Unlock() return s.ListUniqueHostsInLabelsFunc(ctx, filter, labels) } func (s *DataStore) SearchLabels(ctx context.Context, filter fleet.TeamFilter, query string, omit ...uint) ([]*fleet.Label, error) { s.mu.Lock() s.SearchLabelsFuncInvoked = true s.mu.Unlock() return s.SearchLabelsFunc(ctx, filter, query, omit...) } func (s *DataStore) LabelIDsByName(ctx context.Context, labels []string) ([]uint, error) { s.mu.Lock() s.LabelIDsByNameFuncInvoked = true s.mu.Unlock() return s.LabelIDsByNameFunc(ctx, labels) } func (s *DataStore) AsyncBatchInsertLabelMembership(ctx context.Context, batch [][2]uint) error { s.mu.Lock() s.AsyncBatchInsertLabelMembershipFuncInvoked = true s.mu.Unlock() return s.AsyncBatchInsertLabelMembershipFunc(ctx, batch) } func (s *DataStore) AsyncBatchDeleteLabelMembership(ctx context.Context, batch [][2]uint) error { s.mu.Lock() s.AsyncBatchDeleteLabelMembershipFuncInvoked = true s.mu.Unlock() return s.AsyncBatchDeleteLabelMembershipFunc(ctx, batch) } func (s *DataStore) AsyncBatchUpdateLabelTimestamp(ctx context.Context, ids []uint, ts time.Time) error { s.mu.Lock() s.AsyncBatchUpdateLabelTimestampFuncInvoked = true s.mu.Unlock() return s.AsyncBatchUpdateLabelTimestampFunc(ctx, ids, ts) } func (s *DataStore) NewHost(ctx context.Context, host *fleet.Host) (*fleet.Host, error) { s.mu.Lock() s.NewHostFuncInvoked = true s.mu.Unlock() return s.NewHostFunc(ctx, host) } func (s *DataStore) DeleteHost(ctx context.Context, hid uint) error { s.mu.Lock() s.DeleteHostFuncInvoked = true s.mu.Unlock() return s.DeleteHostFunc(ctx, hid) } func (s *DataStore) Host(ctx context.Context, id uint) (*fleet.Host, error) { s.mu.Lock() s.HostFuncInvoked = true s.mu.Unlock() return s.HostFunc(ctx, id) } func (s *DataStore) ListHosts(ctx context.Context, filter fleet.TeamFilter, opt fleet.HostListOptions) ([]*fleet.Host, error) { s.mu.Lock() s.ListHostsFuncInvoked = true s.mu.Unlock() return s.ListHostsFunc(ctx, filter, opt) } func (s *DataStore) MarkHostsSeen(ctx context.Context, hostIDs []uint, t time.Time) error { s.mu.Lock() s.MarkHostsSeenFuncInvoked = true s.mu.Unlock() return s.MarkHostsSeenFunc(ctx, hostIDs, t) } func (s *DataStore) SearchHosts(ctx context.Context, filter fleet.TeamFilter, query string, omit ...uint) ([]*fleet.Host, error) { s.mu.Lock() s.SearchHostsFuncInvoked = true s.mu.Unlock() return s.SearchHostsFunc(ctx, filter, query, omit...) } func (s *DataStore) EnrolledHostIDs(ctx context.Context) ([]uint, error) { s.mu.Lock() s.EnrolledHostIDsFuncInvoked = true s.mu.Unlock() return s.EnrolledHostIDsFunc(ctx) } func (s *DataStore) CountEnrolledHosts(ctx context.Context) (int, error) { s.mu.Lock() s.CountEnrolledHostsFuncInvoked = true s.mu.Unlock() return s.CountEnrolledHostsFunc(ctx) } func (s *DataStore) CleanupIncomingHosts(ctx context.Context, now time.Time) ([]uint, error) { s.mu.Lock() s.CleanupIncomingHostsFuncInvoked = true s.mu.Unlock() return s.CleanupIncomingHostsFunc(ctx, now) } func (s *DataStore) GenerateHostStatusStatistics(ctx context.Context, filter fleet.TeamFilter, now time.Time, platform *string, lowDiskSpace *int) (*fleet.HostSummary, error) { s.mu.Lock() s.GenerateHostStatusStatisticsFuncInvoked = true s.mu.Unlock() return s.GenerateHostStatusStatisticsFunc(ctx, filter, now, platform, lowDiskSpace) } func (s *DataStore) HostIDsByName(ctx context.Context, filter fleet.TeamFilter, hostnames []string) ([]uint, error) { s.mu.Lock() s.HostIDsByNameFuncInvoked = true s.mu.Unlock() return s.HostIDsByNameFunc(ctx, filter, hostnames) } func (s *DataStore) HostIDsByOSID(ctx context.Context, osID uint, offset int, limit int) ([]uint, error) { s.mu.Lock() s.HostIDsByOSIDFuncInvoked = true s.mu.Unlock() return s.HostIDsByOSIDFunc(ctx, osID, offset, limit) } func (s *DataStore) HostIDsByOSVersion(ctx context.Context, osVersion fleet.OSVersion, offset int, limit int) ([]uint, error) { s.mu.Lock() s.HostIDsByOSVersionFuncInvoked = true s.mu.Unlock() return s.HostIDsByOSVersionFunc(ctx, osVersion, offset, limit) } func (s *DataStore) HostByIdentifier(ctx context.Context, identifier string) (*fleet.Host, error) { s.mu.Lock() s.HostByIdentifierFuncInvoked = true s.mu.Unlock() return s.HostByIdentifierFunc(ctx, identifier) } func (s *DataStore) AddHostsToTeam(ctx context.Context, teamID *uint, hostIDs []uint) error { s.mu.Lock() s.AddHostsToTeamFuncInvoked = true s.mu.Unlock() return s.AddHostsToTeamFunc(ctx, teamID, hostIDs) } func (s *DataStore) TotalAndUnseenHostsSince(ctx context.Context, daysCount int) (total int, unseen int, err error) { s.mu.Lock() s.TotalAndUnseenHostsSinceFuncInvoked = true s.mu.Unlock() return s.TotalAndUnseenHostsSinceFunc(ctx, daysCount) } func (s *DataStore) DeleteHosts(ctx context.Context, ids []uint) error { s.mu.Lock() s.DeleteHostsFuncInvoked = true s.mu.Unlock() return s.DeleteHostsFunc(ctx, ids) } func (s *DataStore) CountHosts(ctx context.Context, filter fleet.TeamFilter, opt fleet.HostListOptions) (int, error) { s.mu.Lock() s.CountHostsFuncInvoked = true s.mu.Unlock() return s.CountHostsFunc(ctx, filter, opt) } func (s *DataStore) CountHostsInLabel(ctx context.Context, filter fleet.TeamFilter, lid uint, opt fleet.HostListOptions) (int, error) { s.mu.Lock() s.CountHostsInLabelFuncInvoked = true s.mu.Unlock() return s.CountHostsInLabelFunc(ctx, filter, lid, opt) } func (s *DataStore) ListHostDeviceMapping(ctx context.Context, id uint) ([]*fleet.HostDeviceMapping, error) { s.mu.Lock() s.ListHostDeviceMappingFuncInvoked = true s.mu.Unlock() return s.ListHostDeviceMappingFunc(ctx, id) } func (s *DataStore) ListHostBatteries(ctx context.Context, id uint) ([]*fleet.HostBattery, error) { s.mu.Lock() s.ListHostBatteriesFuncInvoked = true s.mu.Unlock() return s.ListHostBatteriesFunc(ctx, id) } func (s *DataStore) LoadHostByDeviceAuthToken(ctx context.Context, authToken string, tokenTTL time.Duration) (*fleet.Host, error) { s.mu.Lock() s.LoadHostByDeviceAuthTokenFuncInvoked = true s.mu.Unlock() return s.LoadHostByDeviceAuthTokenFunc(ctx, authToken, tokenTTL) } func (s *DataStore) SetOrUpdateDeviceAuthToken(ctx context.Context, hostID uint, authToken string) error { s.mu.Lock() s.SetOrUpdateDeviceAuthTokenFuncInvoked = true s.mu.Unlock() return s.SetOrUpdateDeviceAuthTokenFunc(ctx, hostID, authToken) } func (s *DataStore) FailingPoliciesCount(ctx context.Context, host *fleet.Host) (uint, error) { s.mu.Lock() s.FailingPoliciesCountFuncInvoked = true s.mu.Unlock() return s.FailingPoliciesCountFunc(ctx, host) } func (s *DataStore) ListPoliciesForHost(ctx context.Context, host *fleet.Host) ([]*fleet.HostPolicy, error) { s.mu.Lock() s.ListPoliciesForHostFuncInvoked = true s.mu.Unlock() return s.ListPoliciesForHostFunc(ctx, host) } func (s *DataStore) GetHostMunkiVersion(ctx context.Context, hostID uint) (string, error) { s.mu.Lock() s.GetHostMunkiVersionFuncInvoked = true s.mu.Unlock() return s.GetHostMunkiVersionFunc(ctx, hostID) } func (s *DataStore) GetHostMunkiIssues(ctx context.Context, hostID uint) ([]*fleet.HostMunkiIssue, error) { s.mu.Lock() s.GetHostMunkiIssuesFuncInvoked = true s.mu.Unlock() return s.GetHostMunkiIssuesFunc(ctx, hostID) } func (s *DataStore) GetHostMDM(ctx context.Context, hostID uint) (*fleet.HostMDM, error) { s.mu.Lock() s.GetHostMDMFuncInvoked = true s.mu.Unlock() return s.GetHostMDMFunc(ctx, hostID) } func (s *DataStore) GetHostMDMCheckinInfo(ctx context.Context, hostUUID string) (*fleet.HostMDMCheckinInfo, error) { s.mu.Lock() s.GetHostMDMCheckinInfoFuncInvoked = true s.mu.Unlock() return s.GetHostMDMCheckinInfoFunc(ctx, hostUUID) } func (s *DataStore) AggregatedMunkiVersion(ctx context.Context, teamID *uint) ([]fleet.AggregatedMunkiVersion, time.Time, error) { s.mu.Lock() s.AggregatedMunkiVersionFuncInvoked = true s.mu.Unlock() return s.AggregatedMunkiVersionFunc(ctx, teamID) } func (s *DataStore) AggregatedMunkiIssues(ctx context.Context, teamID *uint) ([]fleet.AggregatedMunkiIssue, time.Time, error) { s.mu.Lock() s.AggregatedMunkiIssuesFuncInvoked = true s.mu.Unlock() return s.AggregatedMunkiIssuesFunc(ctx, teamID) } func (s *DataStore) AggregatedMDMStatus(ctx context.Context, teamID *uint, platform string) (fleet.AggregatedMDMStatus, time.Time, error) { s.mu.Lock() s.AggregatedMDMStatusFuncInvoked = true s.mu.Unlock() return s.AggregatedMDMStatusFunc(ctx, teamID, platform) } func (s *DataStore) AggregatedMDMSolutions(ctx context.Context, teamID *uint, platform string) ([]fleet.AggregatedMDMSolutions, time.Time, error) { s.mu.Lock() s.AggregatedMDMSolutionsFuncInvoked = true s.mu.Unlock() return s.AggregatedMDMSolutionsFunc(ctx, teamID, platform) } func (s *DataStore) GenerateAggregatedMunkiAndMDM(ctx context.Context) error { s.mu.Lock() s.GenerateAggregatedMunkiAndMDMFuncInvoked = true s.mu.Unlock() return s.GenerateAggregatedMunkiAndMDMFunc(ctx) } func (s *DataStore) GetMunkiIssue(ctx context.Context, munkiIssueID uint) (*fleet.MunkiIssue, error) { s.mu.Lock() s.GetMunkiIssueFuncInvoked = true s.mu.Unlock() return s.GetMunkiIssueFunc(ctx, munkiIssueID) } func (s *DataStore) GetMDMSolution(ctx context.Context, mdmID uint) (*fleet.MDMSolution, error) { s.mu.Lock() s.GetMDMSolutionFuncInvoked = true s.mu.Unlock() return s.GetMDMSolutionFunc(ctx, mdmID) } func (s *DataStore) OSVersions(ctx context.Context, teamID *uint, platform *string, name *string, version *string) (*fleet.OSVersions, error) { s.mu.Lock() s.OSVersionsFuncInvoked = true s.mu.Unlock() return s.OSVersionsFunc(ctx, teamID, platform, name, version) } func (s *DataStore) UpdateOSVersions(ctx context.Context) error { s.mu.Lock() s.UpdateOSVersionsFuncInvoked = true s.mu.Unlock() return s.UpdateOSVersionsFunc(ctx) } func (s *DataStore) CountHostsInTargets(ctx context.Context, filter fleet.TeamFilter, targets fleet.HostTargets, now time.Time) (fleet.TargetMetrics, error) { s.mu.Lock() s.CountHostsInTargetsFuncInvoked = true s.mu.Unlock() return s.CountHostsInTargetsFunc(ctx, filter, targets, now) } func (s *DataStore) HostIDsInTargets(ctx context.Context, filter fleet.TeamFilter, targets fleet.HostTargets) ([]uint, error) { s.mu.Lock() s.HostIDsInTargetsFuncInvoked = true s.mu.Unlock() return s.HostIDsInTargetsFunc(ctx, filter, targets) } func (s *DataStore) NewPasswordResetRequest(ctx context.Context, req *fleet.PasswordResetRequest) (*fleet.PasswordResetRequest, error) { s.mu.Lock() s.NewPasswordResetRequestFuncInvoked = true s.mu.Unlock() return s.NewPasswordResetRequestFunc(ctx, req) } func (s *DataStore) DeletePasswordResetRequestsForUser(ctx context.Context, userID uint) error { s.mu.Lock() s.DeletePasswordResetRequestsForUserFuncInvoked = true s.mu.Unlock() return s.DeletePasswordResetRequestsForUserFunc(ctx, userID) } func (s *DataStore) FindPasswordResetByToken(ctx context.Context, token string) (*fleet.PasswordResetRequest, error) { s.mu.Lock() s.FindPasswordResetByTokenFuncInvoked = true s.mu.Unlock() return s.FindPasswordResetByTokenFunc(ctx, token) } func (s *DataStore) CleanupExpiredPasswordResetRequests(ctx context.Context) error { s.mu.Lock() s.CleanupExpiredPasswordResetRequestsFuncInvoked = true s.mu.Unlock() return s.CleanupExpiredPasswordResetRequestsFunc(ctx) } func (s *DataStore) SessionByKey(ctx context.Context, key string) (*fleet.Session, error) { s.mu.Lock() s.SessionByKeyFuncInvoked = true s.mu.Unlock() return s.SessionByKeyFunc(ctx, key) } func (s *DataStore) SessionByID(ctx context.Context, id uint) (*fleet.Session, error) { s.mu.Lock() s.SessionByIDFuncInvoked = true s.mu.Unlock() return s.SessionByIDFunc(ctx, id) } func (s *DataStore) ListSessionsForUser(ctx context.Context, id uint) ([]*fleet.Session, error) { s.mu.Lock() s.ListSessionsForUserFuncInvoked = true s.mu.Unlock() return s.ListSessionsForUserFunc(ctx, id) } func (s *DataStore) NewSession(ctx context.Context, userID uint, sessionKey string) (*fleet.Session, error) { s.mu.Lock() s.NewSessionFuncInvoked = true s.mu.Unlock() return s.NewSessionFunc(ctx, userID, sessionKey) } func (s *DataStore) DestroySession(ctx context.Context, session *fleet.Session) error { s.mu.Lock() s.DestroySessionFuncInvoked = true s.mu.Unlock() return s.DestroySessionFunc(ctx, session) } func (s *DataStore) DestroyAllSessionsForUser(ctx context.Context, id uint) error { s.mu.Lock() s.DestroyAllSessionsForUserFuncInvoked = true s.mu.Unlock() return s.DestroyAllSessionsForUserFunc(ctx, id) } func (s *DataStore) MarkSessionAccessed(ctx context.Context, session *fleet.Session) error { s.mu.Lock() s.MarkSessionAccessedFuncInvoked = true s.mu.Unlock() return s.MarkSessionAccessedFunc(ctx, session) } func (s *DataStore) NewAppConfig(ctx context.Context, info *fleet.AppConfig) (*fleet.AppConfig, error) { s.mu.Lock() s.NewAppConfigFuncInvoked = true s.mu.Unlock() return s.NewAppConfigFunc(ctx, info) } func (s *DataStore) AppConfig(ctx context.Context) (*fleet.AppConfig, error) { s.mu.Lock() s.AppConfigFuncInvoked = true s.mu.Unlock() return s.AppConfigFunc(ctx) } func (s *DataStore) SaveAppConfig(ctx context.Context, info *fleet.AppConfig) error { s.mu.Lock() s.SaveAppConfigFuncInvoked = true s.mu.Unlock() return s.SaveAppConfigFunc(ctx, info) } func (s *DataStore) GetEnrollSecrets(ctx context.Context, teamID *uint) ([]*fleet.EnrollSecret, error) { s.mu.Lock() s.GetEnrollSecretsFuncInvoked = true s.mu.Unlock() return s.GetEnrollSecretsFunc(ctx, teamID) } func (s *DataStore) ApplyEnrollSecrets(ctx context.Context, teamID *uint, secrets []*fleet.EnrollSecret) error { s.mu.Lock() s.ApplyEnrollSecretsFuncInvoked = true s.mu.Unlock() return s.ApplyEnrollSecretsFunc(ctx, teamID, secrets) } func (s *DataStore) NewInvite(ctx context.Context, i *fleet.Invite) (*fleet.Invite, error) { s.mu.Lock() s.NewInviteFuncInvoked = true s.mu.Unlock() return s.NewInviteFunc(ctx, i) } func (s *DataStore) ListInvites(ctx context.Context, opt fleet.ListOptions) ([]*fleet.Invite, error) { s.mu.Lock() s.ListInvitesFuncInvoked = true s.mu.Unlock() return s.ListInvitesFunc(ctx, opt) } func (s *DataStore) Invite(ctx context.Context, id uint) (*fleet.Invite, error) { s.mu.Lock() s.InviteFuncInvoked = true s.mu.Unlock() return s.InviteFunc(ctx, id) } func (s *DataStore) InviteByEmail(ctx context.Context, email string) (*fleet.Invite, error) { s.mu.Lock() s.InviteByEmailFuncInvoked = true s.mu.Unlock() return s.InviteByEmailFunc(ctx, email) } func (s *DataStore) InviteByToken(ctx context.Context, token string) (*fleet.Invite, error) { s.mu.Lock() s.InviteByTokenFuncInvoked = true s.mu.Unlock() return s.InviteByTokenFunc(ctx, token) } func (s *DataStore) DeleteInvite(ctx context.Context, id uint) error { s.mu.Lock() s.DeleteInviteFuncInvoked = true s.mu.Unlock() return s.DeleteInviteFunc(ctx, id) } func (s *DataStore) UpdateInvite(ctx context.Context, id uint, i *fleet.Invite) (*fleet.Invite, error) { s.mu.Lock() s.UpdateInviteFuncInvoked = true s.mu.Unlock() return s.UpdateInviteFunc(ctx, id, i) } func (s *DataStore) ListScheduledQueriesInPackWithStats(ctx context.Context, id uint, opts fleet.ListOptions) ([]*fleet.ScheduledQuery, error) { s.mu.Lock() s.ListScheduledQueriesInPackWithStatsFuncInvoked = true s.mu.Unlock() return s.ListScheduledQueriesInPackWithStatsFunc(ctx, id, opts) } func (s *DataStore) NewScheduledQuery(ctx context.Context, sq *fleet.ScheduledQuery, opts ...fleet.OptionalArg) (*fleet.ScheduledQuery, error) { s.mu.Lock() s.NewScheduledQueryFuncInvoked = true s.mu.Unlock() return s.NewScheduledQueryFunc(ctx, sq, opts...) } func (s *DataStore) SaveScheduledQuery(ctx context.Context, sq *fleet.ScheduledQuery) (*fleet.ScheduledQuery, error) { s.mu.Lock() s.SaveScheduledQueryFuncInvoked = true s.mu.Unlock() return s.SaveScheduledQueryFunc(ctx, sq) } func (s *DataStore) DeleteScheduledQuery(ctx context.Context, id uint) error { s.mu.Lock() s.DeleteScheduledQueryFuncInvoked = true s.mu.Unlock() return s.DeleteScheduledQueryFunc(ctx, id) } func (s *DataStore) ScheduledQuery(ctx context.Context, id uint) (*fleet.ScheduledQuery, error) { s.mu.Lock() s.ScheduledQueryFuncInvoked = true s.mu.Unlock() return s.ScheduledQueryFunc(ctx, id) } func (s *DataStore) CleanupExpiredHosts(ctx context.Context) ([]uint, error) { s.mu.Lock() s.CleanupExpiredHostsFuncInvoked = true s.mu.Unlock() return s.CleanupExpiredHostsFunc(ctx) } func (s *DataStore) ScheduledQueryIDsByName(ctx context.Context, batchSize int, packAndSchedQueryNames ...[2]string) ([]uint, error) { s.mu.Lock() s.ScheduledQueryIDsByNameFuncInvoked = true s.mu.Unlock() return s.ScheduledQueryIDsByNameFunc(ctx, batchSize, packAndSchedQueryNames...) } func (s *DataStore) NewTeam(ctx context.Context, team *fleet.Team) (*fleet.Team, error) { s.mu.Lock() s.NewTeamFuncInvoked = true s.mu.Unlock() return s.NewTeamFunc(ctx, team) } func (s *DataStore) SaveTeam(ctx context.Context, team *fleet.Team) (*fleet.Team, error) { s.mu.Lock() s.SaveTeamFuncInvoked = true s.mu.Unlock() return s.SaveTeamFunc(ctx, team) } func (s *DataStore) Team(ctx context.Context, tid uint) (*fleet.Team, error) { s.mu.Lock() s.TeamFuncInvoked = true s.mu.Unlock() return s.TeamFunc(ctx, tid) } func (s *DataStore) DeleteTeam(ctx context.Context, tid uint) error { s.mu.Lock() s.DeleteTeamFuncInvoked = true s.mu.Unlock() return s.DeleteTeamFunc(ctx, tid) } func (s *DataStore) TeamByName(ctx context.Context, name string) (*fleet.Team, error) { s.mu.Lock() s.TeamByNameFuncInvoked = true s.mu.Unlock() return s.TeamByNameFunc(ctx, name) } func (s *DataStore) ListTeams(ctx context.Context, filter fleet.TeamFilter, opt fleet.ListOptions) ([]*fleet.Team, error) { s.mu.Lock() s.ListTeamsFuncInvoked = true s.mu.Unlock() return s.ListTeamsFunc(ctx, filter, opt) } func (s *DataStore) TeamsSummary(ctx context.Context) ([]*fleet.TeamSummary, error) { s.mu.Lock() s.TeamsSummaryFuncInvoked = true s.mu.Unlock() return s.TeamsSummaryFunc(ctx) } func (s *DataStore) SearchTeams(ctx context.Context, filter fleet.TeamFilter, matchQuery string, omit ...uint) ([]*fleet.Team, error) { s.mu.Lock() s.SearchTeamsFuncInvoked = true s.mu.Unlock() return s.SearchTeamsFunc(ctx, filter, matchQuery, omit...) } func (s *DataStore) TeamEnrollSecrets(ctx context.Context, teamID uint) ([]*fleet.EnrollSecret, error) { s.mu.Lock() s.TeamEnrollSecretsFuncInvoked = true s.mu.Unlock() return s.TeamEnrollSecretsFunc(ctx, teamID) } func (s *DataStore) DeleteIntegrationsFromTeams(ctx context.Context, deletedIntgs fleet.Integrations) error { s.mu.Lock() s.DeleteIntegrationsFromTeamsFuncInvoked = true s.mu.Unlock() return s.DeleteIntegrationsFromTeamsFunc(ctx, deletedIntgs) } func (s *DataStore) ListSoftwareForVulnDetection(ctx context.Context, hostID uint) ([]fleet.Software, error) { s.mu.Lock() s.ListSoftwareForVulnDetectionFuncInvoked = true s.mu.Unlock() return s.ListSoftwareForVulnDetectionFunc(ctx, hostID) } func (s *DataStore) ListSoftwareVulnerabilitiesByHostIDsSource(ctx context.Context, hostIDs []uint, source fleet.VulnerabilitySource) (map[uint][]fleet.SoftwareVulnerability, error) { s.mu.Lock() s.ListSoftwareVulnerabilitiesByHostIDsSourceFuncInvoked = true s.mu.Unlock() return s.ListSoftwareVulnerabilitiesByHostIDsSourceFunc(ctx, hostIDs, source) } func (s *DataStore) LoadHostSoftware(ctx context.Context, host *fleet.Host, includeCVEScores bool) error { s.mu.Lock() s.LoadHostSoftwareFuncInvoked = true s.mu.Unlock() return s.LoadHostSoftwareFunc(ctx, host, includeCVEScores) } func (s *DataStore) ListSoftwareBySourceIter(ctx context.Context, sources []string) (fleet.SoftwareIterator, error) { s.mu.Lock() s.ListSoftwareBySourceIterFuncInvoked = true s.mu.Unlock() return s.ListSoftwareBySourceIterFunc(ctx, sources) } func (s *DataStore) AllSoftwareWithoutCPEIterator(ctx context.Context, excludedPlatforms []string) (fleet.SoftwareIterator, error) { s.mu.Lock() s.AllSoftwareWithoutCPEIteratorFuncInvoked = true s.mu.Unlock() return s.AllSoftwareWithoutCPEIteratorFunc(ctx, excludedPlatforms) } func (s *DataStore) AddCPEForSoftware(ctx context.Context, software fleet.Software, cpe string) error { s.mu.Lock() s.AddCPEForSoftwareFuncInvoked = true s.mu.Unlock() return s.AddCPEForSoftwareFunc(ctx, software, cpe) } func (s *DataStore) ListSoftwareCPEs(ctx context.Context) ([]fleet.SoftwareCPE, error) { s.mu.Lock() s.ListSoftwareCPEsFuncInvoked = true s.mu.Unlock() return s.ListSoftwareCPEsFunc(ctx) } func (s *DataStore) InsertSoftwareVulnerabilities(ctx context.Context, vulns []fleet.SoftwareVulnerability, source fleet.VulnerabilitySource) (int64, error) { s.mu.Lock() s.InsertSoftwareVulnerabilitiesFuncInvoked = true s.mu.Unlock() return s.InsertSoftwareVulnerabilitiesFunc(ctx, vulns, source) } func (s *DataStore) SoftwareByID(ctx context.Context, id uint, includeCVEScores bool) (*fleet.Software, error) { s.mu.Lock() s.SoftwareByIDFuncInvoked = true s.mu.Unlock() return s.SoftwareByIDFunc(ctx, id, includeCVEScores) } func (s *DataStore) ListSoftwareByHostIDShort(ctx context.Context, hostID uint) ([]fleet.Software, error) { s.mu.Lock() s.ListSoftwareByHostIDShortFuncInvoked = true s.mu.Unlock() return s.ListSoftwareByHostIDShortFunc(ctx, hostID) } func (s *DataStore) SyncHostsSoftware(ctx context.Context, updatedAt time.Time) error { s.mu.Lock() s.SyncHostsSoftwareFuncInvoked = true s.mu.Unlock() return s.SyncHostsSoftwareFunc(ctx, updatedAt) } func (s *DataStore) HostsBySoftwareIDs(ctx context.Context, softwareIDs []uint) ([]*fleet.HostShort, error) { s.mu.Lock() s.HostsBySoftwareIDsFuncInvoked = true s.mu.Unlock() return s.HostsBySoftwareIDsFunc(ctx, softwareIDs) } func (s *DataStore) HostsByCVE(ctx context.Context, cve string) ([]*fleet.HostShort, error) { s.mu.Lock() s.HostsByCVEFuncInvoked = true s.mu.Unlock() return s.HostsByCVEFunc(ctx, cve) } func (s *DataStore) InsertCVEMeta(ctx context.Context, cveMeta []fleet.CVEMeta) error { s.mu.Lock() s.InsertCVEMetaFuncInvoked = true s.mu.Unlock() return s.InsertCVEMetaFunc(ctx, cveMeta) } func (s *DataStore) ListCVEs(ctx context.Context, maxAge time.Duration) ([]fleet.CVEMeta, error) { s.mu.Lock() s.ListCVEsFuncInvoked = true s.mu.Unlock() return s.ListCVEsFunc(ctx, maxAge) } func (s *DataStore) ListOperatingSystems(ctx context.Context) ([]fleet.OperatingSystem, error) { s.mu.Lock() s.ListOperatingSystemsFuncInvoked = true s.mu.Unlock() return s.ListOperatingSystemsFunc(ctx) } func (s *DataStore) UpdateHostOperatingSystem(ctx context.Context, hostID uint, hostOS fleet.OperatingSystem) error { s.mu.Lock() s.UpdateHostOperatingSystemFuncInvoked = true s.mu.Unlock() return s.UpdateHostOperatingSystemFunc(ctx, hostID, hostOS) } func (s *DataStore) CleanupHostOperatingSystems(ctx context.Context) error { s.mu.Lock() s.CleanupHostOperatingSystemsFuncInvoked = true s.mu.Unlock() return s.CleanupHostOperatingSystemsFunc(ctx) } func (s *DataStore) UpdateHostTablesOnMDMUnenroll(ctx context.Context, uuid string) error { s.mu.Lock() s.UpdateHostTablesOnMDMUnenrollFuncInvoked = true s.mu.Unlock() return s.UpdateHostTablesOnMDMUnenrollFunc(ctx, uuid) } func (s *DataStore) NewActivity(ctx context.Context, user *fleet.User, activity fleet.ActivityDetails) error { s.mu.Lock() s.NewActivityFuncInvoked = true s.mu.Unlock() return s.NewActivityFunc(ctx, user, activity) } func (s *DataStore) ListActivities(ctx context.Context, opt fleet.ListActivitiesOptions) ([]*fleet.Activity, *fleet.PaginationMetadata, error) { s.mu.Lock() s.ListActivitiesFuncInvoked = true s.mu.Unlock() return s.ListActivitiesFunc(ctx, opt) } func (s *DataStore) MarkActivitiesAsStreamed(ctx context.Context, activityIDs []uint) error { s.mu.Lock() s.MarkActivitiesAsStreamedFuncInvoked = true s.mu.Unlock() return s.MarkActivitiesAsStreamedFunc(ctx, activityIDs) } func (s *DataStore) ShouldSendStatistics(ctx context.Context, frequency time.Duration, config config.FleetConfig) (fleet.StatisticsPayload, bool, error) { s.mu.Lock() s.ShouldSendStatisticsFuncInvoked = true s.mu.Unlock() return s.ShouldSendStatisticsFunc(ctx, frequency, config) } func (s *DataStore) RecordStatisticsSent(ctx context.Context) error { s.mu.Lock() s.RecordStatisticsSentFuncInvoked = true s.mu.Unlock() return s.RecordStatisticsSentFunc(ctx) } func (s *DataStore) CleanupStatistics(ctx context.Context) error { s.mu.Lock() s.CleanupStatisticsFuncInvoked = true s.mu.Unlock() return s.CleanupStatisticsFunc(ctx) } func (s *DataStore) ApplyPolicySpecs(ctx context.Context, authorID uint, specs []*fleet.PolicySpec) error { s.mu.Lock() s.ApplyPolicySpecsFuncInvoked = true s.mu.Unlock() return s.ApplyPolicySpecsFunc(ctx, authorID, specs) } func (s *DataStore) NewGlobalPolicy(ctx context.Context, authorID *uint, args fleet.PolicyPayload) (*fleet.Policy, error) { s.mu.Lock() s.NewGlobalPolicyFuncInvoked = true s.mu.Unlock() return s.NewGlobalPolicyFunc(ctx, authorID, args) } func (s *DataStore) Policy(ctx context.Context, id uint) (*fleet.Policy, error) { s.mu.Lock() s.PolicyFuncInvoked = true s.mu.Unlock() return s.PolicyFunc(ctx, id) } func (s *DataStore) SavePolicy(ctx context.Context, p *fleet.Policy) error { s.mu.Lock() s.SavePolicyFuncInvoked = true s.mu.Unlock() return s.SavePolicyFunc(ctx, p) } func (s *DataStore) ListGlobalPolicies(ctx context.Context) ([]*fleet.Policy, error) { s.mu.Lock() s.ListGlobalPoliciesFuncInvoked = true s.mu.Unlock() return s.ListGlobalPoliciesFunc(ctx) } func (s *DataStore) PoliciesByID(ctx context.Context, ids []uint) (map[uint]*fleet.Policy, error) { s.mu.Lock() s.PoliciesByIDFuncInvoked = true s.mu.Unlock() return s.PoliciesByIDFunc(ctx, ids) } func (s *DataStore) DeleteGlobalPolicies(ctx context.Context, ids []uint) ([]uint, error) { s.mu.Lock() s.DeleteGlobalPoliciesFuncInvoked = true s.mu.Unlock() return s.DeleteGlobalPoliciesFunc(ctx, ids) } func (s *DataStore) PolicyQueriesForHost(ctx context.Context, host *fleet.Host) (map[string]string, error) { s.mu.Lock() s.PolicyQueriesForHostFuncInvoked = true s.mu.Unlock() return s.PolicyQueriesForHostFunc(ctx, host) } func (s *DataStore) AsyncBatchInsertPolicyMembership(ctx context.Context, batch []fleet.PolicyMembershipResult) error { s.mu.Lock() s.AsyncBatchInsertPolicyMembershipFuncInvoked = true s.mu.Unlock() return s.AsyncBatchInsertPolicyMembershipFunc(ctx, batch) } func (s *DataStore) AsyncBatchUpdatePolicyTimestamp(ctx context.Context, ids []uint, ts time.Time) error { s.mu.Lock() s.AsyncBatchUpdatePolicyTimestampFuncInvoked = true s.mu.Unlock() return s.AsyncBatchUpdatePolicyTimestampFunc(ctx, ids, ts) } func (s *DataStore) MigrateTables(ctx context.Context) error { s.mu.Lock() s.MigrateTablesFuncInvoked = true s.mu.Unlock() return s.MigrateTablesFunc(ctx) } func (s *DataStore) MigrateData(ctx context.Context) error { s.mu.Lock() s.MigrateDataFuncInvoked = true s.mu.Unlock() return s.MigrateDataFunc(ctx) } func (s *DataStore) MigrationStatus(ctx context.Context) (*fleet.MigrationStatus, error) { s.mu.Lock() s.MigrationStatusFuncInvoked = true s.mu.Unlock() return s.MigrationStatusFunc(ctx) } func (s *DataStore) ListSoftware(ctx context.Context, opt fleet.SoftwareListOptions) ([]fleet.Software, error) { s.mu.Lock() s.ListSoftwareFuncInvoked = true s.mu.Unlock() return s.ListSoftwareFunc(ctx, opt) } func (s *DataStore) CountSoftware(ctx context.Context, opt fleet.SoftwareListOptions) (int, error) { s.mu.Lock() s.CountSoftwareFuncInvoked = true s.mu.Unlock() return s.CountSoftwareFunc(ctx, opt) } func (s *DataStore) DeleteSoftwareVulnerabilities(ctx context.Context, vulnerabilities []fleet.SoftwareVulnerability) error { s.mu.Lock() s.DeleteSoftwareVulnerabilitiesFuncInvoked = true s.mu.Unlock() return s.DeleteSoftwareVulnerabilitiesFunc(ctx, vulnerabilities) } func (s *DataStore) NewTeamPolicy(ctx context.Context, teamID uint, authorID *uint, args fleet.PolicyPayload) (*fleet.Policy, error) { s.mu.Lock() s.NewTeamPolicyFuncInvoked = true s.mu.Unlock() return s.NewTeamPolicyFunc(ctx, teamID, authorID, args) } func (s *DataStore) ListTeamPolicies(ctx context.Context, teamID uint) (teamPolicies []*fleet.Policy, inheritedPolicies []*fleet.Policy, err error) { s.mu.Lock() s.ListTeamPoliciesFuncInvoked = true s.mu.Unlock() return s.ListTeamPoliciesFunc(ctx, teamID) } func (s *DataStore) DeleteTeamPolicies(ctx context.Context, teamID uint, ids []uint) ([]uint, error) { s.mu.Lock() s.DeleteTeamPoliciesFuncInvoked = true s.mu.Unlock() return s.DeleteTeamPoliciesFunc(ctx, teamID, ids) } func (s *DataStore) TeamPolicy(ctx context.Context, teamID uint, policyID uint) (*fleet.Policy, error) { s.mu.Lock() s.TeamPolicyFuncInvoked = true s.mu.Unlock() return s.TeamPolicyFunc(ctx, teamID, policyID) } func (s *DataStore) CleanupPolicyMembership(ctx context.Context, now time.Time) error { s.mu.Lock() s.CleanupPolicyMembershipFuncInvoked = true s.mu.Unlock() return s.CleanupPolicyMembershipFunc(ctx, now) } func (s *DataStore) IncrementPolicyViolationDays(ctx context.Context) error { s.mu.Lock() s.IncrementPolicyViolationDaysFuncInvoked = true s.mu.Unlock() return s.IncrementPolicyViolationDaysFunc(ctx) } func (s *DataStore) InitializePolicyViolationDays(ctx context.Context) error { s.mu.Lock() s.InitializePolicyViolationDaysFuncInvoked = true s.mu.Unlock() return s.InitializePolicyViolationDaysFunc(ctx) } func (s *DataStore) Lock(ctx context.Context, name string, owner string, expiration time.Duration) (bool, error) { s.mu.Lock() s.LockFuncInvoked = true s.mu.Unlock() return s.LockFunc(ctx, name, owner, expiration) } func (s *DataStore) Unlock(ctx context.Context, name string, owner string) error { s.mu.Lock() s.UnlockFuncInvoked = true s.mu.Unlock() return s.UnlockFunc(ctx, name, owner) } func (s *DataStore) DBLocks(ctx context.Context) ([]*fleet.DBLock, error) { s.mu.Lock() s.DBLocksFuncInvoked = true s.mu.Unlock() return s.DBLocksFunc(ctx) } func (s *DataStore) GetLatestCronStats(ctx context.Context, name string) ([]fleet.CronStats, error) { s.mu.Lock() s.GetLatestCronStatsFuncInvoked = true s.mu.Unlock() return s.GetLatestCronStatsFunc(ctx, name) } func (s *DataStore) InsertCronStats(ctx context.Context, statsType fleet.CronStatsType, name string, instance string, status fleet.CronStatsStatus) (int, error) { s.mu.Lock() s.InsertCronStatsFuncInvoked = true s.mu.Unlock() return s.InsertCronStatsFunc(ctx, statsType, name, instance, status) } func (s *DataStore) UpdateCronStats(ctx context.Context, id int, status fleet.CronStatsStatus) error { s.mu.Lock() s.UpdateCronStatsFuncInvoked = true s.mu.Unlock() return s.UpdateCronStatsFunc(ctx, id, status) } func (s *DataStore) UpdateAllCronStatsForInstance(ctx context.Context, instance string, fromStatus fleet.CronStatsStatus, toStatus fleet.CronStatsStatus) error { s.mu.Lock() s.UpdateAllCronStatsForInstanceFuncInvoked = true s.mu.Unlock() return s.UpdateAllCronStatsForInstanceFunc(ctx, instance, fromStatus, toStatus) } func (s *DataStore) CleanupCronStats(ctx context.Context) error { s.mu.Lock() s.CleanupCronStatsFuncInvoked = true s.mu.Unlock() return s.CleanupCronStatsFunc(ctx) } func (s *DataStore) UpdateScheduledQueryAggregatedStats(ctx context.Context) error { s.mu.Lock() s.UpdateScheduledQueryAggregatedStatsFuncInvoked = true s.mu.Unlock() return s.UpdateScheduledQueryAggregatedStatsFunc(ctx) } func (s *DataStore) UpdateQueryAggregatedStats(ctx context.Context) error { s.mu.Lock() s.UpdateQueryAggregatedStatsFuncInvoked = true s.mu.Unlock() return s.UpdateQueryAggregatedStatsFunc(ctx) } func (s *DataStore) LoadHostByNodeKey(ctx context.Context, nodeKey string) (*fleet.Host, error) { s.mu.Lock() s.LoadHostByNodeKeyFuncInvoked = true s.mu.Unlock() return s.LoadHostByNodeKeyFunc(ctx, nodeKey) } func (s *DataStore) LoadHostByOrbitNodeKey(ctx context.Context, nodeKey string) (*fleet.Host, error) { s.mu.Lock() s.LoadHostByOrbitNodeKeyFuncInvoked = true s.mu.Unlock() return s.LoadHostByOrbitNodeKeyFunc(ctx, nodeKey) } func (s *DataStore) HostLite(ctx context.Context, hostID uint) (*fleet.Host, error) { s.mu.Lock() s.HostLiteFuncInvoked = true s.mu.Unlock() return s.HostLiteFunc(ctx, hostID) } func (s *DataStore) UpdateHostOsqueryIntervals(ctx context.Context, hostID uint, intervals fleet.HostOsqueryIntervals) error { s.mu.Lock() s.UpdateHostOsqueryIntervalsFuncInvoked = true s.mu.Unlock() return s.UpdateHostOsqueryIntervalsFunc(ctx, hostID, intervals) } func (s *DataStore) TeamAgentOptions(ctx context.Context, teamID uint) (*json.RawMessage, error) { s.mu.Lock() s.TeamAgentOptionsFuncInvoked = true s.mu.Unlock() return s.TeamAgentOptionsFunc(ctx, teamID) } func (s *DataStore) TeamFeatures(ctx context.Context, teamID uint) (*fleet.Features, error) { s.mu.Lock() s.TeamFeaturesFuncInvoked = true s.mu.Unlock() return s.TeamFeaturesFunc(ctx, teamID) } func (s *DataStore) TeamMDMConfig(ctx context.Context, teamID uint) (*fleet.TeamMDM, error) { s.mu.Lock() s.TeamMDMConfigFuncInvoked = true s.mu.Unlock() return s.TeamMDMConfigFunc(ctx, teamID) } func (s *DataStore) SaveHostPackStats(ctx context.Context, hostID uint, stats []fleet.PackStats) error { s.mu.Lock() s.SaveHostPackStatsFuncInvoked = true s.mu.Unlock() return s.SaveHostPackStatsFunc(ctx, hostID, stats) } func (s *DataStore) AsyncBatchSaveHostsScheduledQueryStats(ctx context.Context, stats map[uint][]fleet.ScheduledQueryStats, batchSize int) (int, error) { s.mu.Lock() s.AsyncBatchSaveHostsScheduledQueryStatsFuncInvoked = true s.mu.Unlock() return s.AsyncBatchSaveHostsScheduledQueryStatsFunc(ctx, stats, batchSize) } func (s *DataStore) UpdateHostSoftware(ctx context.Context, hostID uint, software []fleet.Software) error { s.mu.Lock() s.UpdateHostSoftwareFuncInvoked = true s.mu.Unlock() return s.UpdateHostSoftwareFunc(ctx, hostID, software) } func (s *DataStore) UpdateHost(ctx context.Context, host *fleet.Host) error { s.mu.Lock() s.UpdateHostFuncInvoked = true s.mu.Unlock() return s.UpdateHostFunc(ctx, host) } func (s *DataStore) ListScheduledQueriesInPack(ctx context.Context, packID uint) (fleet.ScheduledQueryList, error) { s.mu.Lock() s.ListScheduledQueriesInPackFuncInvoked = true s.mu.Unlock() return s.ListScheduledQueriesInPackFunc(ctx, packID) } func (s *DataStore) UpdateHostRefetchRequested(ctx context.Context, hostID uint, value bool) error { s.mu.Lock() s.UpdateHostRefetchRequestedFuncInvoked = true s.mu.Unlock() return s.UpdateHostRefetchRequestedFunc(ctx, hostID, value) } func (s *DataStore) FlippingPoliciesForHost(ctx context.Context, hostID uint, incomingResults map[uint]*bool) (newFailing []uint, newPassing []uint, err error) { s.mu.Lock() s.FlippingPoliciesForHostFuncInvoked = true s.mu.Unlock() return s.FlippingPoliciesForHostFunc(ctx, hostID, incomingResults) } func (s *DataStore) RecordPolicyQueryExecutions(ctx context.Context, host *fleet.Host, results map[uint]*bool, updated time.Time, deferredSaveHost bool) error { s.mu.Lock() s.RecordPolicyQueryExecutionsFuncInvoked = true s.mu.Unlock() return s.RecordPolicyQueryExecutionsFunc(ctx, host, results, updated, deferredSaveHost) } func (s *DataStore) RecordLabelQueryExecutions(ctx context.Context, host *fleet.Host, results map[uint]*bool, t time.Time, deferredSaveHost bool) error { s.mu.Lock() s.RecordLabelQueryExecutionsFuncInvoked = true s.mu.Unlock() return s.RecordLabelQueryExecutionsFunc(ctx, host, results, t, deferredSaveHost) } func (s *DataStore) SaveHostUsers(ctx context.Context, hostID uint, users []fleet.HostUser) error { s.mu.Lock() s.SaveHostUsersFuncInvoked = true s.mu.Unlock() return s.SaveHostUsersFunc(ctx, hostID, users) } func (s *DataStore) SaveHostAdditional(ctx context.Context, hostID uint, additional *json.RawMessage) error { s.mu.Lock() s.SaveHostAdditionalFuncInvoked = true s.mu.Unlock() return s.SaveHostAdditionalFunc(ctx, hostID, additional) } func (s *DataStore) SetOrUpdateMunkiInfo(ctx context.Context, hostID uint, version string, errors []string, warnings []string) error { s.mu.Lock() s.SetOrUpdateMunkiInfoFuncInvoked = true s.mu.Unlock() return s.SetOrUpdateMunkiInfoFunc(ctx, hostID, version, errors, warnings) } func (s *DataStore) SetOrUpdateMDMData(ctx context.Context, hostID uint, isServer bool, enrolled bool, serverURL string, installedFromDep bool, name string) error { s.mu.Lock() s.SetOrUpdateMDMDataFuncInvoked = true s.mu.Unlock() return s.SetOrUpdateMDMDataFunc(ctx, hostID, isServer, enrolled, serverURL, installedFromDep, name) } func (s *DataStore) SetOrUpdateHostDisksSpace(ctx context.Context, hostID uint, gigsAvailable float64, percentAvailable float64) error { s.mu.Lock() s.SetOrUpdateHostDisksSpaceFuncInvoked = true s.mu.Unlock() return s.SetOrUpdateHostDisksSpaceFunc(ctx, hostID, gigsAvailable, percentAvailable) } func (s *DataStore) SetOrUpdateHostDisksEncryption(ctx context.Context, hostID uint, encrypted bool) error { s.mu.Lock() s.SetOrUpdateHostDisksEncryptionFuncInvoked = true s.mu.Unlock() return s.SetOrUpdateHostDisksEncryptionFunc(ctx, hostID, encrypted) } func (s *DataStore) SetOrUpdateHostDiskEncryptionKey(ctx context.Context, hostID uint, encryptedBase64Key string) error { s.mu.Lock() s.SetOrUpdateHostDiskEncryptionKeyFuncInvoked = true s.mu.Unlock() return s.SetOrUpdateHostDiskEncryptionKeyFunc(ctx, hostID, encryptedBase64Key) } func (s *DataStore) GetUnverifiedDiskEncryptionKeys(ctx context.Context) ([]fleet.HostDiskEncryptionKey, error) { s.mu.Lock() s.GetUnverifiedDiskEncryptionKeysFuncInvoked = true s.mu.Unlock() return s.GetUnverifiedDiskEncryptionKeysFunc(ctx) } func (s *DataStore) SetHostsDiskEncryptionKeyStatus(ctx context.Context, hostIDs []uint, encryptable bool, threshold time.Time) error { s.mu.Lock() s.SetHostsDiskEncryptionKeyStatusFuncInvoked = true s.mu.Unlock() return s.SetHostsDiskEncryptionKeyStatusFunc(ctx, hostIDs, encryptable, threshold) } func (s *DataStore) GetHostDiskEncryptionKey(ctx context.Context, hostID uint) (*fleet.HostDiskEncryptionKey, error) { s.mu.Lock() s.GetHostDiskEncryptionKeyFuncInvoked = true s.mu.Unlock() return s.GetHostDiskEncryptionKeyFunc(ctx, hostID) } func (s *DataStore) SetOrUpdateHostOrbitInfo(ctx context.Context, hostID uint, version string) error { s.mu.Lock() s.SetOrUpdateHostOrbitInfoFuncInvoked = true s.mu.Unlock() return s.SetOrUpdateHostOrbitInfoFunc(ctx, hostID, version) } func (s *DataStore) ReplaceHostDeviceMapping(ctx context.Context, id uint, mappings []*fleet.HostDeviceMapping) error { s.mu.Lock() s.ReplaceHostDeviceMappingFuncInvoked = true s.mu.Unlock() return s.ReplaceHostDeviceMappingFunc(ctx, id, mappings) } func (s *DataStore) ReplaceHostBatteries(ctx context.Context, id uint, mappings []*fleet.HostBattery) error { s.mu.Lock() s.ReplaceHostBatteriesFuncInvoked = true s.mu.Unlock() return s.ReplaceHostBatteriesFunc(ctx, id, mappings) } func (s *DataStore) VerifyEnrollSecret(ctx context.Context, secret string) (*fleet.EnrollSecret, error) { s.mu.Lock() s.VerifyEnrollSecretFuncInvoked = true s.mu.Unlock() return s.VerifyEnrollSecretFunc(ctx, secret) } func (s *DataStore) EnrollHost(ctx context.Context, osqueryHostId string, nodeKey string, teamID *uint, cooldown time.Duration) (*fleet.Host, error) { s.mu.Lock() s.EnrollHostFuncInvoked = true s.mu.Unlock() return s.EnrollHostFunc(ctx, osqueryHostId, nodeKey, teamID, cooldown) } func (s *DataStore) EnrollOrbit(ctx context.Context, hardwareUUID string, orbitNodeKey string, teamID *uint) (*fleet.Host, error) { s.mu.Lock() s.EnrollOrbitFuncInvoked = true s.mu.Unlock() return s.EnrollOrbitFunc(ctx, hardwareUUID, orbitNodeKey, teamID) } func (s *DataStore) SerialUpdateHost(ctx context.Context, host *fleet.Host) error { s.mu.Lock() s.SerialUpdateHostFuncInvoked = true s.mu.Unlock() return s.SerialUpdateHostFunc(ctx, host) } func (s *DataStore) NewJob(ctx context.Context, job *fleet.Job) (*fleet.Job, error) { s.mu.Lock() s.NewJobFuncInvoked = true s.mu.Unlock() return s.NewJobFunc(ctx, job) } func (s *DataStore) GetQueuedJobs(ctx context.Context, maxNumJobs int) ([]*fleet.Job, error) { s.mu.Lock() s.GetQueuedJobsFuncInvoked = true s.mu.Unlock() return s.GetQueuedJobsFunc(ctx, maxNumJobs) } func (s *DataStore) UpdateJob(ctx context.Context, id uint, job *fleet.Job) (*fleet.Job, error) { s.mu.Lock() s.UpdateJobFuncInvoked = true s.mu.Unlock() return s.UpdateJobFunc(ctx, id, job) } func (s *DataStore) InnoDBStatus(ctx context.Context) (string, error) { s.mu.Lock() s.InnoDBStatusFuncInvoked = true s.mu.Unlock() return s.InnoDBStatusFunc(ctx) } func (s *DataStore) ProcessList(ctx context.Context) ([]fleet.MySQLProcess, error) { s.mu.Lock() s.ProcessListFuncInvoked = true s.mu.Unlock() return s.ProcessListFunc(ctx) } func (s *DataStore) ListWindowsUpdatesByHostID(ctx context.Context, hostID uint) ([]fleet.WindowsUpdate, error) { s.mu.Lock() s.ListWindowsUpdatesByHostIDFuncInvoked = true s.mu.Unlock() return s.ListWindowsUpdatesByHostIDFunc(ctx, hostID) } func (s *DataStore) InsertWindowsUpdates(ctx context.Context, hostID uint, updates []fleet.WindowsUpdate) error { s.mu.Lock() s.InsertWindowsUpdatesFuncInvoked = true s.mu.Unlock() return s.InsertWindowsUpdatesFunc(ctx, hostID, updates) } func (s *DataStore) ListOSVulnerabilities(ctx context.Context, hostID []uint) ([]fleet.OSVulnerability, error) { s.mu.Lock() s.ListOSVulnerabilitiesFuncInvoked = true s.mu.Unlock() return s.ListOSVulnerabilitiesFunc(ctx, hostID) } func (s *DataStore) InsertOSVulnerabilities(ctx context.Context, vulnerabilities []fleet.OSVulnerability, source fleet.VulnerabilitySource) (int64, error) { s.mu.Lock() s.InsertOSVulnerabilitiesFuncInvoked = true s.mu.Unlock() return s.InsertOSVulnerabilitiesFunc(ctx, vulnerabilities, source) } func (s *DataStore) DeleteOSVulnerabilities(ctx context.Context, vulnerabilities []fleet.OSVulnerability) error { s.mu.Lock() s.DeleteOSVulnerabilitiesFuncInvoked = true s.mu.Unlock() return s.DeleteOSVulnerabilitiesFunc(ctx, vulnerabilities) } func (s *DataStore) NewMDMAppleConfigProfile(ctx context.Context, p fleet.MDMAppleConfigProfile) (*fleet.MDMAppleConfigProfile, error) { s.mu.Lock() s.NewMDMAppleConfigProfileFuncInvoked = true s.mu.Unlock() return s.NewMDMAppleConfigProfileFunc(ctx, p) } func (s *DataStore) GetMDMAppleConfigProfile(ctx context.Context, profileID uint) (*fleet.MDMAppleConfigProfile, error) { s.mu.Lock() s.GetMDMAppleConfigProfileFuncInvoked = true s.mu.Unlock() return s.GetMDMAppleConfigProfileFunc(ctx, profileID) } func (s *DataStore) ListMDMAppleConfigProfiles(ctx context.Context, teamID *uint) ([]*fleet.MDMAppleConfigProfile, error) { s.mu.Lock() s.ListMDMAppleConfigProfilesFuncInvoked = true s.mu.Unlock() return s.ListMDMAppleConfigProfilesFunc(ctx, teamID) } func (s *DataStore) DeleteMDMAppleConfigProfile(ctx context.Context, profileID uint) error { s.mu.Lock() s.DeleteMDMAppleConfigProfileFuncInvoked = true s.mu.Unlock() return s.DeleteMDMAppleConfigProfileFunc(ctx, profileID) } func (s *DataStore) GetHostMDMProfiles(ctx context.Context, hostUUID string) ([]fleet.HostMDMAppleProfile, error) { s.mu.Lock() s.GetHostMDMProfilesFuncInvoked = true s.mu.Unlock() return s.GetHostMDMProfilesFunc(ctx, hostUUID) } func (s *DataStore) NewMDMAppleEnrollmentProfile(ctx context.Context, enrollmentPayload fleet.MDMAppleEnrollmentProfilePayload) (*fleet.MDMAppleEnrollmentProfile, error) { s.mu.Lock() s.NewMDMAppleEnrollmentProfileFuncInvoked = true s.mu.Unlock() return s.NewMDMAppleEnrollmentProfileFunc(ctx, enrollmentPayload) } func (s *DataStore) GetMDMAppleEnrollmentProfileByToken(ctx context.Context, token string) (*fleet.MDMAppleEnrollmentProfile, error) { s.mu.Lock() s.GetMDMAppleEnrollmentProfileByTokenFuncInvoked = true s.mu.Unlock() return s.GetMDMAppleEnrollmentProfileByTokenFunc(ctx, token) } func (s *DataStore) ListMDMAppleEnrollmentProfiles(ctx context.Context) ([]*fleet.MDMAppleEnrollmentProfile, error) { s.mu.Lock() s.ListMDMAppleEnrollmentProfilesFuncInvoked = true s.mu.Unlock() return s.ListMDMAppleEnrollmentProfilesFunc(ctx) } func (s *DataStore) GetMDMAppleCommandResults(ctx context.Context, commandUUID string) (map[string]*fleet.MDMAppleCommandResult, error) { s.mu.Lock() s.GetMDMAppleCommandResultsFuncInvoked = true s.mu.Unlock() return s.GetMDMAppleCommandResultsFunc(ctx, commandUUID) } func (s *DataStore) NewMDMAppleInstaller(ctx context.Context, name string, size int64, manifest string, installer []byte, urlToken string) (*fleet.MDMAppleInstaller, error) { s.mu.Lock() s.NewMDMAppleInstallerFuncInvoked = true s.mu.Unlock() return s.NewMDMAppleInstallerFunc(ctx, name, size, manifest, installer, urlToken) } func (s *DataStore) MDMAppleInstaller(ctx context.Context, token string) (*fleet.MDMAppleInstaller, error) { s.mu.Lock() s.MDMAppleInstallerFuncInvoked = true s.mu.Unlock() return s.MDMAppleInstallerFunc(ctx, token) } func (s *DataStore) MDMAppleInstallerDetailsByID(ctx context.Context, id uint) (*fleet.MDMAppleInstaller, error) { s.mu.Lock() s.MDMAppleInstallerDetailsByIDFuncInvoked = true s.mu.Unlock() return s.MDMAppleInstallerDetailsByIDFunc(ctx, id) } func (s *DataStore) DeleteMDMAppleInstaller(ctx context.Context, id uint) error { s.mu.Lock() s.DeleteMDMAppleInstallerFuncInvoked = true s.mu.Unlock() return s.DeleteMDMAppleInstallerFunc(ctx, id) } func (s *DataStore) MDMAppleInstallerDetailsByToken(ctx context.Context, token string) (*fleet.MDMAppleInstaller, error) { s.mu.Lock() s.MDMAppleInstallerDetailsByTokenFuncInvoked = true s.mu.Unlock() return s.MDMAppleInstallerDetailsByTokenFunc(ctx, token) } func (s *DataStore) ListMDMAppleInstallers(ctx context.Context) ([]fleet.MDMAppleInstaller, error) { s.mu.Lock() s.ListMDMAppleInstallersFuncInvoked = true s.mu.Unlock() return s.ListMDMAppleInstallersFunc(ctx) } func (s *DataStore) BatchSetMDMAppleProfiles(ctx context.Context, tmID *uint, profiles []*fleet.MDMAppleConfigProfile) error { s.mu.Lock() s.BatchSetMDMAppleProfilesFuncInvoked = true s.mu.Unlock() return s.BatchSetMDMAppleProfilesFunc(ctx, tmID, profiles) } func (s *DataStore) MDMAppleListDevices(ctx context.Context) ([]fleet.MDMAppleDevice, error) { s.mu.Lock() s.MDMAppleListDevicesFuncInvoked = true s.mu.Unlock() return s.MDMAppleListDevicesFunc(ctx) } func (s *DataStore) IngestMDMAppleDevicesFromDEPSync(ctx context.Context, devices []godep.Device) (int64, error) { s.mu.Lock() s.IngestMDMAppleDevicesFromDEPSyncFuncInvoked = true s.mu.Unlock() return s.IngestMDMAppleDevicesFromDEPSyncFunc(ctx, devices) } func (s *DataStore) IngestMDMAppleDeviceFromCheckin(ctx context.Context, mdmHost fleet.MDMAppleHostDetails) error { s.mu.Lock() s.IngestMDMAppleDeviceFromCheckinFuncInvoked = true s.mu.Unlock() return s.IngestMDMAppleDeviceFromCheckinFunc(ctx, mdmHost) } func (s *DataStore) GetNanoMDMEnrollmentStatus(ctx context.Context, id string) (bool, error) { s.mu.Lock() s.GetNanoMDMEnrollmentStatusFuncInvoked = true s.mu.Unlock() return s.GetNanoMDMEnrollmentStatusFunc(ctx, id) } func (s *DataStore) IncreasePolicyAutomationIteration(ctx context.Context, policyID uint) error { s.mu.Lock() s.IncreasePolicyAutomationIterationFuncInvoked = true s.mu.Unlock() return s.IncreasePolicyAutomationIterationFunc(ctx, policyID) } func (s *DataStore) OutdatedAutomationBatch(ctx context.Context) ([]fleet.PolicyFailure, error) { s.mu.Lock() s.OutdatedAutomationBatchFuncInvoked = true s.mu.Unlock() return s.OutdatedAutomationBatchFunc(ctx) } func (s *DataStore) ListMDMAppleProfilesToInstall(ctx context.Context) ([]*fleet.MDMAppleProfilePayload, error) { s.mu.Lock() s.ListMDMAppleProfilesToInstallFuncInvoked = true s.mu.Unlock() return s.ListMDMAppleProfilesToInstallFunc(ctx) } func (s *DataStore) ListMDMAppleProfilesToRemove(ctx context.Context) ([]*fleet.MDMAppleProfilePayload, error) { s.mu.Lock() s.ListMDMAppleProfilesToRemoveFuncInvoked = true s.mu.Unlock() return s.ListMDMAppleProfilesToRemoveFunc(ctx) } func (s *DataStore) BulkUpsertMDMAppleHostProfiles(ctx context.Context, payload []*fleet.MDMAppleBulkUpsertHostProfilePayload) error { s.mu.Lock() s.BulkUpsertMDMAppleHostProfilesFuncInvoked = true s.mu.Unlock() return s.BulkUpsertMDMAppleHostProfilesFunc(ctx, payload) } func (s *DataStore) GetMDMAppleProfilesContents(ctx context.Context, profileIDs []uint) (map[uint]fleet.Mobileconfig, error) { s.mu.Lock() s.GetMDMAppleProfilesContentsFuncInvoked = true s.mu.Unlock() return s.GetMDMAppleProfilesContentsFunc(ctx, profileIDs) } func (s *DataStore) UpdateHostMDMAppleProfile(ctx context.Context, profile *fleet.HostMDMAppleProfile) error { s.mu.Lock() s.UpdateHostMDMAppleProfileFuncInvoked = true s.mu.Unlock() return s.UpdateHostMDMAppleProfileFunc(ctx, profile) } func (s *DataStore) GetMDMAppleCommandRequestType(ctx context.Context, commandUUID string) (string, error) { s.mu.Lock() s.GetMDMAppleCommandRequestTypeFuncInvoked = true s.mu.Unlock() return s.GetMDMAppleCommandRequestTypeFunc(ctx, commandUUID) }