test: Fix intermittent race in p2p_unrequested_blocks.py

test/functional/p2p_unrequested_blocks.py

@@ -257,16 +257,11 @@ class AcceptBlockTest(BitcoinTestFramework):
         test_node.send_message(msg_block(block_291))
 
         # At this point we've sent an obviously-bogus block, wait for full processing
-        # without assuming whether we will be disconnected or not
+        # and assume disconnection
-        try:
+        test_node.wait_for_disconnect()
-            # Only wait a short while so the test doesn't take forever if we do get
+
-            # disconnected
+        self.nodes[0].disconnect_p2ps()
-            test_node.sync_with_ping(timeout=1)
+        test_node = self.nodes[0].add_p2p_connection(P2PInterface())
-        except AssertionError:
-            test_node.wait_for_disconnect()
-
-            self.nodes[0].disconnect_p2ps()
-            test_node = self.nodes[0].add_p2p_connection(P2PInterface())
 
         # We should have failed reorg and switched back to 290 (but have block 291)
         assert_equal(self.nodes[0].getblockcount(), 290)