diff options
Diffstat (limited to 'test/remove_conn_and_wait_protocol.erl')
| -rw-r--r-- | test/remove_conn_and_wait_protocol.erl | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/test/remove_conn_and_wait_protocol.erl b/test/remove_conn_and_wait_protocol.erl index 59cedfc..caac41e 100644 --- a/test/remove_conn_and_wait_protocol.erl +++ b/test/remove_conn_and_wait_protocol.erl @@ -10,10 +10,12 @@ start_link(Ref, _, _, [{remove, MaybeRemove, Timeout}]) -> init(Ref, MaybeRemove, Timeout) -> {ok, _} = ranch:handshake(Ref), - case MaybeRemove of + _ = case MaybeRemove of true -> ranch:remove_connection(Ref); false -> - ok + ok; + N -> + [ranch:remove_connection(Ref) || _ <- lists:seq(1, N)] end, receive after Timeout -> ok end. |