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

Expand monolith tests to cover edge cases

Browse source
This commit is contained in:
Devon Hudson 2023-01-28 16:03:01 -07:00
parent e56f777e50
commit e49ab5e321
No known key found for this signature in database
GPG key ID: CD06B18E77F6A628
1 changed files with 29 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
build/gobind-pinecone/monolith_test.go
View file

@ -198,32 +198,53 @@ func TestMonolithStarts(t *testing.T) {
}
func TestMonolithSetRelayServers(t *testing.T) {
// "valid,invalid,,valid"
// ensure both valids and only them exist
// test with both valid & invalid nodeID input
// test with both self & remote nodeID
testCases := []struct {
name string
nodeID string
relays string
expectedRelays string
expectSelf bool
}{
{
name: "assorted valid, invalid, empty & self keys",
nodeID: "@valid:abcdef123456abcdef123456abcdef123456abcdef123456abcdef123456abcd",
relays: "@valid:123456123456abcdef123456abcdef123456abcdef123456abcdef123456abcd,@invalid:notakey,,",
expectedRelays: "123456123456abcdef123456abcdef123456abcdef123456abcdef123456abcd",
expectSelf: true,
},
{
name: "invalid node key",
nodeID: "@invalid:notakey",
relays: "@valid:123456123456abcdef123456abcdef123456abcdef123456abcdef123456abcd,@invalid:notakey,,",
expectedRelays: "",
expectSelf: false,
},
{
name: "node is self",
nodeID: "self",
relays: "@valid:123456123456abcdef123456abcdef123456abcdef123456abcdef123456abcd,@invalid:notakey,,",
expectedRelays: "123456123456abcdef123456abcdef123456abcdef123456abcdef123456abcd",
expectSelf: false,
},
}
for _, tc := range testCases {
monolith := DendriteMonolith{}
monolith.Start()
inputRelays := tc.relays + "," + monolith.PublicKey()
expectedRelays := tc.expectedRelays + "," + monolith.PublicKey()
monolith.SetRelayServers(tc.nodeID, inputRelays)
relays := monolith.GetRelayServers(tc.nodeID)
inputRelays := tc.relays
expectedRelays := tc.expectedRelays
if tc.expectSelf {
inputRelays += "," + monolith.PublicKey()
expectedRelays += "," + monolith.PublicKey()
}
nodeID := tc.nodeID
if nodeID == "self" {
nodeID = monolith.PublicKey()
}
monolith.SetRelayServers(nodeID, inputRelays)
relays := monolith.GetRelayServers(nodeID)
monolith.Stop()
if relays != expectedRelays {