64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
package alertmanagerbatcher
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"testing"
|
|
|
|
"github.com/SigNoz/signoz/pkg/types/alertmanagertypes"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBatcherWithOneAlertAndDefaultConfigs(t *testing.T) {
|
|
batcher := New(slog.New(slog.DiscardHandler), NewConfig())
|
|
_ = batcher.Start(context.Background())
|
|
|
|
batcher.Add(context.Background(), &alertmanagertypes.PostableAlert{Alert: alertmanagertypes.AlertModel{
|
|
Labels: map[string]string{"alertname": "test"},
|
|
}})
|
|
|
|
alerts := <-batcher.C
|
|
assert.Equal(t, 1, len(alerts))
|
|
|
|
batcher.Stop(context.Background())
|
|
}
|
|
|
|
func TestBatcherWithBatchSize(t *testing.T) {
|
|
batcher := New(slog.New(slog.DiscardHandler), Config{Size: 2, Capacity: 4})
|
|
_ = batcher.Start(context.Background())
|
|
|
|
var alerts alertmanagertypes.PostableAlerts
|
|
for i := 0; i < 4; i++ {
|
|
alerts = append(alerts, &alertmanagertypes.PostableAlert{Alert: alertmanagertypes.AlertModel{
|
|
Labels: map[string]string{"alertname": "test"},
|
|
}})
|
|
}
|
|
batcher.Add(context.Background(), alerts...)
|
|
|
|
for i := 0; i < 2; i++ {
|
|
alerts := <-batcher.C
|
|
assert.Equal(t, 2, len(alerts))
|
|
}
|
|
|
|
batcher.Stop(context.Background())
|
|
}
|
|
|
|
func TestBatcherWithCClosed(t *testing.T) {
|
|
batcher := New(slog.New(slog.DiscardHandler), Config{Size: 2, Capacity: 4})
|
|
_ = batcher.Start(context.Background())
|
|
|
|
var alerts alertmanagertypes.PostableAlerts
|
|
for i := 0; i < 4; i++ {
|
|
alerts = append(alerts, &alertmanagertypes.PostableAlert{Alert: alertmanagertypes.AlertModel{
|
|
Labels: map[string]string{"alertname": "test"},
|
|
}})
|
|
}
|
|
batcher.Add(context.Background(), alerts...)
|
|
|
|
batcher.Stop(context.Background())
|
|
|
|
for alerts := range batcher.C {
|
|
assert.Equal(t, 2, len(alerts))
|
|
}
|
|
}
|