Add HTTP::Response#connection

Author: janko

lib/http/response.rb

@@ -71,6 +71,10 @@ def initialize(opts)
     #   (see HTTP::Response::Body#readpartial)
     def_delegator :body, :readpartial
 
+    # @!method connection
+    #   (see HTTP::Response::Body#connection)
+    def_delegator :body, :connection
+
     # Returns an Array ala Rack: `[status, headers, body]`
     #
     # @return [Array(Fixnum, Hash, String)]

lib/http/response/body.rb

@@ -10,17 +10,22 @@ class Body
       include Enumerable
       def_delegator :to_s, :empty?
 
-      def initialize(client, encoding = Encoding::BINARY)
-        @client    = client
-        @streaming = nil
-        @contents  = nil
-        @encoding  = encoding
+      # The connection object used to make the corresponding request.
+      #
+      # @return [HTTP::Connection]
+      attr_reader :connection
+
+      def initialize(connection, encoding = Encoding::BINARY)
+        @connection = connection
+        @streaming  = nil
+        @contents   = nil
+        @encoding   = encoding
       end
 
       # (see HTTP::Client#readpartial)
       def readpartial(*args)
         stream!
-        @client.readpartial(*args)
+        @connection.readpartial(*args)
       end
 
       # Iterate over the body, allowing it to be enumerable
@@ -47,7 +52,7 @@ def to_s
           @streaming  = false
           @contents   = String.new("").force_encoding(encoding)
 
-          while (chunk = @client.readpartial)
+          while (chunk = @connection.readpartial)
             @contents << chunk.force_encoding(encoding)
           end
         rescue

spec/lib/http/response/body_spec.rb

@@ -1,11 +1,11 @@
 # frozen_string_literal: true
 RSpec.describe HTTP::Response::Body do
-  let(:client)   { double(:sequence_id => 0) }
-  let(:chunks)   { [String.new("Hello, "), String.new("World!")] }
+  let(:connection) { double(:sequence_id => 0) }
+  let(:chunks)     { [String.new("Hello, "), String.new("World!")] }
 
-  before         { allow(client).to receive(:readpartial) { chunks.shift } }
+  before           { allow(connection).to receive(:readpartial) { chunks.shift } }
 
-  subject(:body) { described_class.new(client, Encoding::UTF_8) }
+  subject(:body)   { described_class.new(connection, Encoding::UTF_8) }
 
   it "streams bodies from responses" do
     expect(subject.to_s).to eq("Hello, World!")
@@ -21,8 +21,8 @@
 
   describe "#readpartial" do
     context "with size given" do
-      it "passes value to underlying client" do
-        expect(client).to receive(:readpartial).with(42)
+      it "passes value to underlying connection" do
+        expect(connection).to receive(:readpartial).with(42)
         body.readpartial 42
       end
     end
@@ -32,8 +32,8 @@
         expect { body.readpartial }.to_not raise_error
       end
 
-      it "calls underlying client readpartial without specific size" do
-        expect(client).to receive(:readpartial).with no_args
+      it "calls underlying connection readpartial without specific size" do
+        expect(connection).to receive(:readpartial).with no_args
         body.readpartial
       end
     end

spec/lib/http/response_spec.rb

@@ -157,4 +157,20 @@
       expect(jar.count { |c| "c" == c.name }).to eq 0
     end
   end
+
+  describe "#connection" do
+    let(:connection) { double }
+
+    subject(:response) do
+      HTTP::Response.new(
+        :version    => "1.1",
+        :status     => 200,
+        :connection => connection
+      )
+    end
+
+    it "returns the connection object used to instantiate the response" do
+      expect(response.connection).to eq connection
+    end
+  end
 end