metrics/app/vmalert/notifier/faker.go

package notifier

import (
	"context"
	"fmt"
	"sync"
	"time"

	"github.com/VictoriaMetrics/VictoriaMetrics/lib/prompb"
)

// FakeNotifier is a mock notifier
type FakeNotifier struct {
	sync.Mutex
	alerts []Alert
	// records number of received alerts in total
	counter int
}

// InitFakeNotifier initializes global notifier to FakeNotifier,
// and returns a cleanup function to restore the original getActiveNotifiers.
func InitFakeNotifier() (*FakeNotifier, func()) {
	originalGetActiveNotifiers := getActiveNotifiers
	fn := &FakeNotifier{}
	getActiveNotifiers = func() []Notifier {
		return []Notifier{fn}
	}
	return fn, func() {
		getActiveNotifiers = originalGetActiveNotifiers
	}
}

// Close does nothing
func (*FakeNotifier) Close() {}

// LastError returns last error message
func (*FakeNotifier) LastError() string {
	return ""
}

// Addr returns ""
func (*FakeNotifier) Addr() string { return "" }

// Send sets alerts and increases counter
func (fn *FakeNotifier) Send(_ context.Context, alerts []Alert, _ [][]prompb.Label, _ map[string]string) error {
	fn.Lock()
	defer fn.Unlock()
	fn.counter += len(alerts)
	fn.alerts = alerts
	return nil
}

// GetCounter returns received alerts count
func (fn *FakeNotifier) GetCounter() int {
	fn.Lock()
	defer fn.Unlock()
	return fn.counter
}

// GetAlerts returns stored alerts
func (fn *FakeNotifier) GetAlerts() []Alert {
	fn.Lock()
	defer fn.Unlock()
	return fn.alerts
}

// FaultyNotifier is a mock notifier that Send() will return failed response
type FaultyNotifier struct {
	FakeNotifier
}

// Send returns failed response
func (fn *FaultyNotifier) Send(ctx context.Context, _ []Alert, _ map[string]string) error {
	d, ok := ctx.Deadline()
	if ok {
		time.Sleep(time.Until(d))
	}
	return fmt.Errorf("send failed")
}