mirror of
https://github.com/prometheus/alertmanager
synced 2025-02-26 15:50:46 +00:00
01e1f5433a
The CI environment isn't as performant as local machines: the time needed to fully initialize the test environment can be significant and skew the verification. Rather than setting the "virtual" clock used to measure alert timings at the beginning of the acceptance test, it is better to wait for the test bed to be ready. Signed-off-by: Simon Pasquier <spasquie@redhat.com>
128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
// Copyright 2018 Prometheus Team
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package test
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
a "github.com/prometheus/alertmanager/test/with_api_v2"
|
|
)
|
|
|
|
// TestClusterDeduplication tests, that in an Alertmanager cluster of 3
|
|
// instances, only one should send a notification for a given alert.
|
|
func TestClusterDeduplication(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
conf := `
|
|
route:
|
|
receiver: "default"
|
|
group_by: []
|
|
group_wait: 1s
|
|
group_interval: 1s
|
|
repeat_interval: 1h
|
|
|
|
receivers:
|
|
- name: "default"
|
|
webhook_configs:
|
|
- url: 'http://%s'
|
|
`
|
|
|
|
at := a.NewAcceptanceTest(t, &a.AcceptanceOpts{
|
|
Tolerance: 1 * time.Second,
|
|
})
|
|
co := at.Collector("webhook")
|
|
wh := a.NewWebhook(t, co)
|
|
|
|
amc := at.AlertmanagerCluster(fmt.Sprintf(conf, wh.Address()), 3)
|
|
|
|
amc.Push(a.At(1), a.Alert("alertname", "test1"))
|
|
|
|
co.Want(a.Between(2, 3), a.Alert("alertname", "test1").Active(1))
|
|
|
|
at.Run()
|
|
|
|
t.Log(co.Check())
|
|
}
|
|
|
|
// TestClusterVSInstance compares notifications sent by Alertmanager cluster to
|
|
// notifications sent by single instance.
|
|
func TestClusterVSInstance(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
conf := `
|
|
route:
|
|
receiver: "default"
|
|
group_by: [ "alertname" ]
|
|
group_wait: 1s
|
|
group_interval: 1s
|
|
repeat_interval: 1h
|
|
|
|
receivers:
|
|
- name: "default"
|
|
webhook_configs:
|
|
- url: 'http://%s'
|
|
`
|
|
|
|
acceptanceOpts := func() *a.AcceptanceOpts {
|
|
return &a.AcceptanceOpts{
|
|
Tolerance: 2 * time.Second,
|
|
}
|
|
}
|
|
|
|
clusterSizes := []int{1, 3}
|
|
|
|
tests := []*a.AcceptanceTest{
|
|
a.NewAcceptanceTest(t, acceptanceOpts()),
|
|
a.NewAcceptanceTest(t, acceptanceOpts()),
|
|
}
|
|
|
|
collectors := []*a.Collector{}
|
|
amClusters := []*a.AlertmanagerCluster{}
|
|
wg := sync.WaitGroup{}
|
|
|
|
for i, tc := range tests {
|
|
collectors = append(collectors, tc.Collector("webhook"))
|
|
webhook := a.NewWebhook(t, collectors[i])
|
|
|
|
amClusters = append(amClusters, tc.AlertmanagerCluster(fmt.Sprintf(conf, webhook.Address()), clusterSizes[i]))
|
|
|
|
wg.Add(1)
|
|
}
|
|
|
|
for _, alertTime := range []float64{0, 2, 4, 6, 8} {
|
|
for i, amc := range amClusters {
|
|
alert := a.Alert("alertname", fmt.Sprintf("test1-%v", alertTime))
|
|
amc.Push(a.At(alertTime), alert)
|
|
collectors[i].Want(a.Between(alertTime, alertTime+5), alert.Active(alertTime))
|
|
}
|
|
}
|
|
|
|
for _, t := range tests {
|
|
go func(t *a.AcceptanceTest) {
|
|
t.Run()
|
|
wg.Done()
|
|
}(t)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
_, err := a.CompareCollectors(collectors[0], collectors[1], acceptanceOpts())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|