From 520b6ef2b268e263fe4a0d33fbc578343c7ebf83 Mon Sep 17 00:00:00 2001
From: Michael Klishin <mklishin@pivotal.io>
Date: Sun, 3 Jul 2016 03:54:52 +0300
Subject: [PATCH] {allow,block}_traffic_between/2 are moved to
 rabbit_ct_broker_helpers

---
 test/partitions_SUITE.erl | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/test/partitions_SUITE.erl b/test/partitions_SUITE.erl
index ff8d5f5f134a..7425bec30769 100644
--- a/test/partitions_SUITE.erl
+++ b/test/partitions_SUITE.erl
@@ -389,12 +389,10 @@ partitions(Node) ->
     end.
 
 block(X, Y) ->
-    rpc:call(X, inet_tcp_proxy, block, [Y]),
-    rpc:call(Y, inet_tcp_proxy, block, [X]).
+    rabbit_ct_broker_helpers:block_traffic_between(X, Y).
 
 allow(X, Y) ->
-    rpc:call(X, inet_tcp_proxy, allow, [Y]),
-    rpc:call(Y, inet_tcp_proxy, allow, [X]).
+    rabbit_ct_broker_helpers:allow_traffic_between(X, Y).
 
 await_running   (Node, Bool)  -> await(Node, Bool,  fun is_running/1).
 await_listening (Node, Bool)  -> await(Node, Bool,  fun is_listening/1).