mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-11 14:53:49 +00:00
dc28196237
1. split package rule under /app/vmalert, expose needed objects 2. add vmalert-tool with unittest subcmd https://github.com/VictoriaMetrics/VictoriaMetrics/issues/2945
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package notifier
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// FakeNotifier is a mock notifier
|
|
type FakeNotifier struct {
|
|
sync.Mutex
|
|
alerts []Alert
|
|
// records number of received alerts in total
|
|
counter int
|
|
}
|
|
|
|
// Close does nothing
|
|
func (*FakeNotifier) Close() {}
|
|
|
|
// Addr returns ""
|
|
func (*FakeNotifier) Addr() string { return "" }
|
|
|
|
// Send sets alerts and increases counter
|
|
func (fn *FakeNotifier) Send(_ context.Context, alerts []Alert, _ 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")
|
|
}
|