From 1718ed1dcc6eb138e802e4b2304d2ea3ce22f7a7 Mon Sep 17 00:00:00 2001
From: Robert Klotzner <eskimor@users.noreply.github.com>
Date: Wed, 4 Aug 2021 14:15:44 +0200
Subject: [PATCH] Fix flaky test (#3569)

Fixes #3557
---
 polkadot/node/network/bridge/src/tests.rs | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/polkadot/node/network/bridge/src/tests.rs b/polkadot/node/network/bridge/src/tests.rs
index 5957b957f93..f22108ba3a2 100644
--- a/polkadot/node/network/bridge/src/tests.rs
+++ b/polkadot/node/network/bridge/src/tests.rs
@@ -770,13 +770,6 @@ fn peer_disconnect_from_just_one_peerset() {
 
 		let peer = PeerId::random();
 
-		network_handle
-			.connect_peer(peer.clone(), PeerSet::Validation, ObservedRole::Full)
-			.await;
-		network_handle
-			.connect_peer(peer.clone(), PeerSet::Collation, ObservedRole::Full)
-			.await;
-
 		assert_matches!(
 			virtual_overseer.recv().await,
 			AllMessages::DisputeDistribution(DisputeDistributionMessage::DisputeSendingReceiver(_))
@@ -788,6 +781,13 @@ fn peer_disconnect_from_just_one_peerset() {
 			)
 		);
 
+		network_handle
+			.connect_peer(peer.clone(), PeerSet::Validation, ObservedRole::Full)
+			.await;
+		network_handle
+			.connect_peer(peer.clone(), PeerSet::Collation, ObservedRole::Full)
+			.await;
+
 		// bridge will inform about all connected peers.
 		{
 			assert_sends_validation_event_to_all(
-- 
GitLab