From 18cc81fb9f98114715f1d21dd60c3ead1011a994 Mon Sep 17 00:00:00 2001 From: Devon Hudson Date: Fri, 23 Dec 2022 17:07:51 -0700 Subject: [PATCH] Add tests for caching relay servers from db --- federationapi/statistics/statistics_test.go | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/federationapi/statistics/statistics_test.go b/federationapi/statistics/statistics_test.go index 9953377fe..24e654499 100644 --- a/federationapi/statistics/statistics_test.go +++ b/federationapi/statistics/statistics_test.go @@ -4,6 +4,10 @@ import ( "math" "testing" "time" + + "github.com/matrix-org/dendrite/federationapi/storage" + "github.com/matrix-org/gomatrixserverlib" + "github.com/stretchr/testify/assert" ) func TestBackoff(t *testing.T) { @@ -74,3 +78,14 @@ func TestBackoff(t *testing.T) { } } } + +func TestRelayServersListing(t *testing.T) { + stats := NewStatistics(storage.NewFakeFederationDatabase(), 8, 3) + server := ServerStatistics{statistics: &stats} + server.AddRelayServers([]gomatrixserverlib.ServerName{"relay1", "relay1", "relay2"}) + relayServers := server.KnownRelayServers() + assert.Equal(t, []gomatrixserverlib.ServerName{"relay1", "relay2"}, relayServers) + server.AddRelayServers([]gomatrixserverlib.ServerName{"relay1", "relay1", "relay2"}) + relayServers = server.KnownRelayServers() + assert.Equal(t, []gomatrixserverlib.ServerName{"relay1", "relay2"}, relayServers) +}