mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-20 05:13:11 -06:00
Code Issues Packages Projects Releases Wiki Activity

Try to ensure that TestCorrectStreamWakeup doesn't block forever

Browse source
This commit is contained in:
Neil Alexander 2020-05-27 18:24:17 +01:00
parent 7a4d9e222d
commit c859164597
2 changed files with 14 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
build/scripts/build-test-lint.sh
View file

@ -13,4 +13,4 @@ go build ./cmd/...
./scripts/find-lint.sh
echo "Testing..."
go test ./...
go test -v ./...

20
syncapi/sync/notifier_test.go
View file

@ -160,14 +160,23 @@ func TestCorrectStreamWakeup(t *testing.T) {
streamone := lockedFetchUserStream(n, alice, "one")
streamtwo := lockedFetchUserStream(n, alice, "two")
go waitForBlocking(streamone, 1)
go waitForBlocking(streamtwo, 1)
wait := func(stream *UserDeviceStream) {
select {
case <-time.After(time.Second * 10):
return
default:
waitForBlocking(stream, 1)
}
}
go wait(streamone)
go wait(streamtwo)
go func() {
select {
case <-n.userDeviceStreams[alice]["one"].signalChannel:
case <-streamone.signalChannel:
awoken <- "one"
case <-n.userDeviceStreams[alice]["two"].signalChannel:
case <-streamtwo.signalChannel:
awoken <- "two"
}
}()
@ -180,9 +189,6 @@ func TestCorrectStreamWakeup(t *testing.T) {
if result := <-awoken; result != wake {
t.Fatalf("expected to wake %q, got %q", wake, result)
}
close(n.userDeviceStreams[alice]["one"].signalChannel)
close(n.userDeviceStreams[alice]["two"].signalChannel)
}
// Test that an invite unblocks the request