Use disconnect_nodes instead of restart

Signed-off-by: Kostiantyn Stepaniuk <[email protected]>
dtr-org · Mar 4, 2019 · b99f3ec · b99f3ec
1 parent 2d98432
commit b99f3ec
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/functional/feature_fork_choice_finalization.py b/test/functional/feature_fork_choice_finalization.py
@@ -89,9 +89,10 @@ def connect_sync_disconnect(node1, node2, blockhash):
         txid = validator.deposit(payto, 10000)
         wait_until(lambda: self.have_tx_in_mempool([node0, node1, node2], txid))
 
-        self.restart_node(node0.index)
-        self.restart_node(node1.index)
-        self.restart_node(node2.index)
+        disconnect_nodes(node0, node1.index)
+        disconnect_nodes(node0, node2.index)
+        disconnect_nodes(node0, validator.index)
+        assert_equal(len(node0.getpeerinfo()), 0)
 
         # 0 ... 4 ... 9 ... 14 ... 19 ... 24 ... 29 - 30
         #       F     F     F      J                  tip