Deflake leave-via-kick/ban faster joins tests (#628)

We need to allow the homeserver under test to finish making requests to
the Complement homeserver before tearing it down, otherwise the
homeserver under test may mark the hostname:port combination as down and
refuse to contact it in subsequent tests.

Signed-off-by: Sean Quah <[email protected]>

Author: squahtx

tests/federation_room_join_partial_state_test.go

@@ -3822,6 +3822,10 @@ func TestPartialStateJoin(t *testing.T) {
 
 			// Cleanup.
 			psjResult.FinishStateRequest()
+			// Dirty hack to allow the homeserver under test to finish making requests to the
+			// Complement homeserver as part of syncing the full state.
+			psjResult.AwaitStateIdsRequest(t)
+			time.Sleep(time.Second / 2)
 		})
 
 		t.Run("can be triggered by remote ban", func(t *testing.T) {
@@ -3880,6 +3884,10 @@ func TestPartialStateJoin(t *testing.T) {
 
 			// Cleanup.
 			psjResult.FinishStateRequest()
+			// Dirty hack to allow the homeserver under test to finish making requests to the
+			// Complement homeserver as part of syncing the full state.
+			psjResult.AwaitStateIdsRequest(t)
+			time.Sleep(time.Second / 2)
 		})
 	})