mocks

type MockAdapter struct {
	NameVal         string
	InputFormatsVal []string
	OutputFormatVal string
	CanConvertVal   bool

	ConvertFn func(ctx context.Context, input []byte, opts *core.AdapterOptions) (*ctis.Report, error)

	ConvertCalls int
}

type MockCollector struct {
	NameVal string
	TypeVal string

	CollectFn        func(ctx context.Context, opts *core.CollectOptions) (*core.CollectResult, error)
	TestConnectionFn func(ctx context.Context) error

	CollectCalls int
}

type MockConnector struct {
	NameVal      string
	TypeVal      string
	ConnectedVal bool

	ConnectFn        func(ctx context.Context) error
	TestConnectionFn func(ctx context.Context) error
	CloseFn          func() error

	ConnectCalls        int
	TestConnectionCalls int
}

type MockPusher struct {
	// PushFindingsFn is called when PushFindings is invoked
	PushFindingsFn func(ctx context.Context, report *ctis.Report) (*core.PushResult, error)

	// PushAssetsFn is called when PushAssets is invoked
	PushAssetsFn func(ctx context.Context, report *ctis.Report) (*core.PushResult, error)

	// SendHeartbeatFn is called when SendHeartbeat is invoked
	SendHeartbeatFn func(ctx context.Context, status *core.AgentStatus) error

	// TestConnectionFn is called when TestConnection is invoked
	TestConnectionFn func(ctx context.Context) error

	// Call tracking
	PushFindingsCalls   []PushFindingsCall
	PushAssetsCalls     []PushAssetsCall
	SendHeartbeatCalls  []SendHeartbeatCall
	TestConnectionCalls int
}

type MockScanner struct {
	NameVal         string
	VersionVal      string
	CapabilitiesVal []string
	InstalledVal    bool

	ScanFn        func(ctx context.Context, target string, opts *core.ScanOptions) (*core.ScanResult, error)
	IsInstalledFn func(ctx context.Context) (bool, string, error)

	ScanCalls        int
	IsInstalledCalls int
}

type PushAssetsCall struct {
	Report *ctis.Report
}

type PushFindingsCall struct {
	Report *ctis.Report
}

type SendHeartbeatCall struct {
	Status *core.AgentStatus
}