mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2026-01-16 18:43:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Check assumed offline state in statistics test

Browse source
This commit is contained in:
Devon Hudson 2023-01-12 11:22:10 -07:00
parent cba0644ec2
commit f6a9a77ecc
No known key found for this signature in database
GPG key ID: CD06B18E77F6A628
1 changed files with 27 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
federationapi/statistics/statistics_test.go
View file

@ -10,8 +10,13 @@ import (
"github.com/stretchr/testify/assert"
)
const (
FailuresUntilAssumedOffline = 2
FailuresUntilBlacklist = 7
)
func TestBackoff(t *testing.T) {
stats := NewStatistics(nil, 7, 2)
stats := NewStatistics(nil, FailuresUntilBlacklist, FailuresUntilAssumedOffline)
server := ServerStatistics{
statistics: &stats,
serverName: "test.com",
@ -50,6 +55,21 @@ func TestBackoff(t *testing.T) {
}
}
// Check if we should be assumed offline by now.
if i >= stats.FailuresUntilAssumedOffline {
if !assumedOffline {
t.Fatalf("Backoff %d should have resulted in assumed offline but didn't", i)
} else {
t.Logf("Backoff %d is assumed offline as expected", i)
}
} else {
if assumedOffline {
t.Fatalf("Backoff %d should not have resulted in assumed offline but did", i)
} else {
t.Logf("Backoff %d is not assumed offline as expected", i)
}
}
// Check if we should be blacklisted by now.
if i >= stats.FailuresUntilBlacklist {
if !blacklist {
@ -60,6 +80,12 @@ func TestBackoff(t *testing.T) {
t.Logf("Backoff %d is blacklisted as expected", i)
continue
}
} else {
if blacklist {
t.Fatalf("Backoff %d should not have resulted in blacklist but did", i)
} else {
t.Logf("Backoff %d is not blacklisted as expected", i)
}
}
// Check if the duration is what we expect.