client: Client#resume always waits all requests

2 files changed, 4 insertions, 11 deletions

diff --git a/lib/plum/client.rb b/lib/plum/client.rb
index 55325a8..bdbfd72 100644
--- a/lib/plum/client.rb
+++ b/lib/plum/client.rb
@@ -54,16 +54,9 @@ module Plum
       self
     end
 
-    # Resume communication with the server, until the specified (or all running) requests are complete.
-    # @param response [Response] if specified, waits only for the response
-    # @return [Response] if parameter response is specified
-    def resume(response = nil)
-      if response
-        @session.succ until response.failed? || response.finished?
-        response
-      else
-        @session.succ until @session.empty?
-      end
+    # Resume communication with the server until all running requests are complete.
+    def resume
+      @session.succ until @session.empty?
     end
 
     # Closes the connection immediately.
diff --git a/test/plum/client/test_client.rb b/test/plum/client/test_client.rb
index 2a4d8b6..c6c7af7 100644
--- a/test/plum/client/test_client.rb
+++ b/test/plum/client/test_client.rb
@@ -37,7 +37,7 @@ class ClientTest < Minitest::Test
     client = Client.start("127.0.0.1", LISTEN_PORT, https: true, verify_mode: OpenSSL::SSL::VERIFY_NONE)
     res = client.get("/error_in_data")
     assert_raises(LocalConnectionError) {
-      client.resume(res)
+      client.resume
     }
     client.close
   ensure