From e05649b7c064f0e95f347c701d8b6c0ec0926f08 Mon Sep 17 00:00:00 2001
From: Kegan Dougal <7190048+kegsay@users.noreply.github.com>
Date: Thu, 10 Oct 2024 11:59:48 +0100
Subject: [PATCH] Fix flake TestNewUserCannotGetKeysForOfflineServer

Servers may need to be prodded that hs2 is back online before it
will service /keys/claim requests. We do this via a typiing EDU.
---
 tests/federation_connectivity_test.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/federation_connectivity_test.go b/tests/federation_connectivity_test.go
index d961fda..bcf3259 100644
--- a/tests/federation_connectivity_test.go
+++ b/tests/federation_connectivity_test.go
@@ -72,7 +72,12 @@ func TestNewUserCannotGetKeysForOfflineServer(t *testing.T) {
 				// See https://github.com/matrix-org/matrix-rust-sdk/issues/281 for why we want to backoff.
 				// See https://github.com/matrix-org/matrix-rust-sdk/issues/2804 for discussions on what the backoff should be.
 				t.Logf("sleeping until client timeout is ready...")
-				time.Sleep(33 * time.Second)
+				time.Sleep(10 * time.Second)
+				// we may need to kick hs1 into letting it know that hs2 is back, we do this by sending a typing notif
+				// in the room, whic will send an EDU over federation which should inform hs1 that hs2 is back online.
+				tc.Bob.MustSendTyping(t, roomID, true, 1000)
+				// wait the remaining client timeout time
+				time.Sleep(23 * time.Second)
 
 				// send another message, bob should be able to decrypt it.
 				wantMsgBody = "Bob can see this because his server is now back online"