fleet/server/service/mock/service_osquery.go

// Automatically generated by mockimpl. DO NOT EDIT!

package mock

import (
	"context"
	"encoding/json"
	"sync"

	"github.com/fleetdm/fleet/v4/server/fleet"
)

var _ fleet.OsqueryService = (*TLSService)(nil)

type EnrollOsqueryFunc func(ctx context.Context, enrollSecret string, hostIdentifier string, hostDetails map[string](map[string]string)) (nodeKey string, err error)

type AuthenticateHostFunc func(ctx context.Context, nodeKey string) (host *fleet.Host, debug bool, err error)

type GetClientConfigFunc func(ctx context.Context) (config map[string]interface{}, err error)

type GetDistributedQueriesFunc func(ctx context.Context) (queries map[string]string, discovery map[string]string, accelerate uint, err error)

type SubmitDistributedQueryResultsFunc func(ctx context.Context, results fleet.OsqueryDistributedQueryResults, statuses map[string]fleet.OsqueryStatus, messages map[string]string, stats map[string]*fleet.Stats) (err error)

type SubmitStatusLogsFunc func(ctx context.Context, logs []json.RawMessage) (err error)

type SubmitResultLogsFunc func(ctx context.Context, logs []json.RawMessage) (err error)

type YaraRuleByNameFunc func(ctx context.Context, name string) (*fleet.YaraRule, error)

type TLSService struct {
	EnrollOsqueryFunc        EnrollOsqueryFunc
	EnrollOsqueryFuncInvoked bool

	AuthenticateHostFunc        AuthenticateHostFunc
	AuthenticateHostFuncInvoked bool

	GetClientConfigFunc        GetClientConfigFunc
	GetClientConfigFuncInvoked bool

	GetDistributedQueriesFunc        GetDistributedQueriesFunc
	GetDistributedQueriesFuncInvoked bool

	SubmitDistributedQueryResultsFunc        SubmitDistributedQueryResultsFunc
	SubmitDistributedQueryResultsFuncInvoked bool

	SubmitStatusLogsFunc        SubmitStatusLogsFunc
	SubmitStatusLogsFuncInvoked bool

	SubmitResultLogsFunc        SubmitResultLogsFunc
	SubmitResultLogsFuncInvoked bool

	YaraRuleByNameFunc        YaraRuleByNameFunc
	YaraRuleByNameFuncInvoked bool

	mu sync.Mutex
}

func (s *TLSService) EnrollOsquery(ctx context.Context, enrollSecret string, hostIdentifier string, hostDetails map[string](map[string]string)) (nodeKey string, err error) {
	s.mu.Lock()
	s.EnrollOsqueryFuncInvoked = true
	s.mu.Unlock()
	return s.EnrollOsqueryFunc(ctx, enrollSecret, hostIdentifier, hostDetails)
}

func (s *TLSService) AuthenticateHost(ctx context.Context, nodeKey string) (host *fleet.Host, debug bool, err error) {
	s.mu.Lock()
	s.AuthenticateHostFuncInvoked = true
	s.mu.Unlock()
	return s.AuthenticateHostFunc(ctx, nodeKey)
}

func (s *TLSService) GetClientConfig(ctx context.Context) (config map[string]interface{}, err error) {
	s.mu.Lock()
	s.GetClientConfigFuncInvoked = true
	s.mu.Unlock()
	return s.GetClientConfigFunc(ctx)
}

func (s *TLSService) GetDistributedQueries(ctx context.Context) (queries map[string]string, discovery map[string]string, accelerate uint, err error) {
	s.mu.Lock()
	s.GetDistributedQueriesFuncInvoked = true
	s.mu.Unlock()
	return s.GetDistributedQueriesFunc(ctx)
}

func (s *TLSService) SubmitDistributedQueryResults(ctx context.Context, results fleet.OsqueryDistributedQueryResults, statuses map[string]fleet.OsqueryStatus, messages map[string]string, stats map[string]*fleet.Stats) (err error) {
	s.mu.Lock()
	s.SubmitDistributedQueryResultsFuncInvoked = true
	s.mu.Unlock()
	return s.SubmitDistributedQueryResultsFunc(ctx, results, statuses, messages, stats)
}

func (s *TLSService) SubmitStatusLogs(ctx context.Context, logs []json.RawMessage) (err error) {
	s.mu.Lock()
	s.SubmitStatusLogsFuncInvoked = true
	s.mu.Unlock()
	return s.SubmitStatusLogsFunc(ctx, logs)
}

func (s *TLSService) SubmitResultLogs(ctx context.Context, logs []json.RawMessage) (err error) {
	s.mu.Lock()
	s.SubmitResultLogsFuncInvoked = true
	s.mu.Unlock()
	return s.SubmitResultLogsFunc(ctx, logs)
}

func (s *TLSService) YaraRuleByName(ctx context.Context, name string) (*fleet.YaraRule, error) {
	s.mu.Lock()
	s.YaraRuleByNameFuncInvoked = true
	s.mu.Unlock()
	return s.YaraRuleByNameFunc(ctx, name)
}