test_common.go

// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2016-present Datadog, Inc.

//go:build test

package aggregator

import (
	"errors"

	log "github.com/DataDog/datadog-agent/comp/core/log/def"
	secretnooptypes "github.com/DataDog/datadog-agent/comp/core/secrets/noop-impl/types"
	defaultforwarder "github.com/DataDog/datadog-agent/comp/forwarder/defaultforwarder/def"
	defaultforwarderimpl "github.com/DataDog/datadog-agent/comp/forwarder/defaultforwarder/impl"
	"github.com/DataDog/datadog-agent/pkg/aggregator/sender"
	checkid "github.com/DataDog/datadog-agent/pkg/collector/check/id"
	pkgconfigsetup "github.com/DataDog/datadog-agent/pkg/config/setup"
	"github.com/DataDog/datadog-agent/pkg/metrics"
)

// PeekSender returns a Sender with passed ID or an error if the sender is not registered
func (s *senders) PeekSender(cid checkid.ID) (sender.Sender, error) {
	return s.senderPool.getSender(cid)
}

// PeekSender returns a Sender with passed ID or an error if the sender is not registered
func (d *AgentDemultiplexer) PeekSender(cid checkid.ID) (sender.Sender, error) {
	d.m.Lock()
	defer d.m.Unlock()
	if d.senders == nil {
		return nil, errors.New("demultiplexer is stopped")
	}
	return d.senders.PeekSender(cid)
}

func NewForwarderTest(log log.Component) defaultforwarder.Forwarder {
	options, _ := defaultforwarderimpl.NewOptions(pkgconfigsetup.Datadog(), log, nil)
	options.Secrets = &secretnooptypes.SecretNoop{}
	return defaultforwarderimpl.NewDefaultForwarder(pkgconfigsetup.Datadog(), log, options)
}

// GetRecurrentSeries returns a copy of the recurrent series for testing
func GetRecurrentSeries() []*metrics.Serie {
	recurrentSeriesLock.Lock()
	defer recurrentSeriesLock.Unlock()
	result := make([]*metrics.Serie, len(recurrentSeries))
	copy(result, recurrentSeries)
	return result
}

// ClearRecurrentSeries clears the recurrent series for testing
func ClearRecurrentSeries() {
	recurrentSeriesLock.Lock()
	defer recurrentSeriesLock.Unlock()
	recurrentSeries = metrics.Series{}
}