mirror of
https://github.com/hanzoai/o11y.git
synced 2026-10-15 12:58:33 +00:00
zeekay
b36a8788a5
Full rebrand of the SigNoz observability platform: - Go module: github.com/SigNoz/signoz → github.com/hanzoai/o11y - Frontend: @signozhq/* → @hanzo/o11y-*, posthog-js → @hanzo/insights - Docker: signoz/* → ghcr.io/hanzoai/o11y* - ClickHouse → Hanzo Datastore (hanzoai/datastore) - Env vars: SIGNOZ_* → HANZO_* - DB tables: signoz_* → o11y_* - All branding, URLs, docs updated - 2172 files changed
64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
package alertmanagerbatcher
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"testing"
|
|
|
|
"github.com/hanzoai/o11y/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))
|
|
}
|
|
}
|