mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-25 07:43:10 -06:00
Code Issues Packages Projects Releases Wiki Activity

Add more graph walking tests

Browse source
This commit is contained in:
Kegan Dougal 2020-11-04 10:53:02 +00:00
parent d092f30367
commit c320261ca2
1 changed files with 113 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

114
internal/mscs/msc2836/msc2836_test.go
View file

@ -219,7 +219,7 @@ func TestMSC2836(t *testing.T) {
body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{ body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(), EventID: eventB.EventID(),
IncludeParent: &constTrue, IncludeParent: &constTrue,
Limit: 1, Limit: 2,
}) })
assertContains(t, body, []string{eventB.EventID(), eventA.EventID()}) assertContains(t, body, []string{eventB.EventID(), eventA.EventID()})
}) })
@ -239,6 +239,118 @@ func TestMSC2836(t *testing.T) {
}) })
assertContains(t, body, []string{eventD.EventID(), eventE.EventID(), eventF.EventID(), eventG.EventID()}) assertContains(t, body, []string{eventD.EventID(), eventE.EventID(), eventF.EventID(), eventG.EventID()})
}) })
t.Run("walks the graph depth first", func(t *testing.T) {
body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constFalse,
DepthFirst: &constTrue,
Limit: 6,
})
// Oldest first so:
// A
// |
// B1
// / \
// C2 D3
// /| \
// 4E 6F G
// |
// 5H
assertContains(t, body, []string{eventB.EventID(), eventC.EventID(), eventD.EventID(), eventE.EventID(), eventH.EventID(), eventF.EventID()})
body = postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constTrue,
DepthFirst: &constTrue,
Limit: 6,
})
// Recent first so:
// A
// |
// B1
// / \
// C D2
// /| \
// E5 F4 G3
// |
// H6
assertContains(t, body, []string{eventB.EventID(), eventD.EventID(), eventG.EventID(), eventF.EventID(), eventE.EventID(), eventH.EventID()})
})
t.Run("walks the graph breadth first", func(t *testing.T) {
body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constFalse,
DepthFirst: &constFalse,
Limit: 6,
})
// Oldest first so:
// A
// |
// B1
// / \
// C2 D3
// /| \
// E4 F5 G6
// |
// H
assertContains(t, body, []string{eventB.EventID(), eventC.EventID(), eventD.EventID(), eventE.EventID(), eventF.EventID(), eventG.EventID()})
body = postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constTrue,
DepthFirst: &constFalse,
Limit: 6,
})
// Recent first so:
// A
// |
// B1
// / \
// C3 D2
// /| \
// E6 F5 G4
// |
// H
assertContains(t, body, []string{eventB.EventID(), eventD.EventID(), eventC.EventID(), eventG.EventID(), eventF.EventID(), eventE.EventID()})
})
t.Run("caps via max_breadth", func(t *testing.T) {
body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constFalse,
DepthFirst: &constFalse,
MaxBreadth: 2,
Limit: 10,
})
// Event G gets omitted because of max_breadth
assertContains(t, body, []string{eventB.EventID(), eventC.EventID(), eventD.EventID(), eventE.EventID(), eventF.EventID(), eventH.EventID()})
})
t.Run("caps via max_depth", func(t *testing.T) {
body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constFalse,
DepthFirst: &constFalse,
MaxDepth: 2,
Limit: 10,
})
// Event H gets omitted because of max_depth
assertContains(t, body, []string{eventB.EventID(), eventC.EventID(), eventD.EventID(), eventE.EventID(), eventF.EventID(), eventG.EventID()})
})
t.Run("terminates when reaching the limit", func(t *testing.T) {
body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constFalse,
DepthFirst: &constFalse,
Limit: 4,
})
assertContains(t, body, []string{eventB.EventID(), eventC.EventID(), eventD.EventID(), eventE.EventID()})
})
t.Run("returns all events with a high enough limit", func(t *testing.T) {
body := postRelationships(t, 200, "alice", &msc2836.EventRelationshipRequest{
EventID: eventB.EventID(),
RecentFirst: &constFalse,
DepthFirst: &constFalse,
Limit: 400,
})
assertContains(t, body, []string{eventB.EventID(), eventC.EventID(), eventD.EventID(), eventE.EventID(), eventF.EventID(), eventG.EventID(), eventH.EventID()})
})
} }
func runServer(t *testing.T, router *mux.Router) func() { func runServer(t *testing.T, router *mux.Router) func() {