Add the Response#status method

Author: rsamoilov

lib/rage/controller/api.rb

@@ -451,6 +451,9 @@ def initialize(env, params)
     @__rendered = false
   end
 
+  # @private
+  attr_reader :__status, :__headers, :__body
+
   # Get the request object. See {Rage::Request}.
   # @return [Rage::Request]
   def request
@@ -460,7 +463,7 @@ def request
   # Get the response object. See {Rage::Response}.
   # @return [Rage::Response]
   def response
-    @response ||= Rage::Response.new(@__headers, @__body)
+    @response ||= Rage::Response.new(self)
   end
 
   # Get the cookie object. See {Rage::Cookies}.

lib/rage/response.rb

@@ -8,21 +8,26 @@ class Rage::Response
   LAST_MODIFIED_HEADER = "Last-Modified"
 
   # @private
-  def initialize(headers, body)
-    @headers = headers
-    @body = body
+  def initialize(controller)
+    @controller = controller
+  end
+
+  # Returns the HTTP status code of the response.
+  # @return [Integer]
+  def status
+    @controller.__status
   end
 
   # Returns the content of the response as a string. This contains the contents of any calls to `render`.
   # @return [String]
   def body
-    @body[0] || ""
+    @controller.__body[0] || ""
   end
 
   # Returns the headers for the response.
   # @return [Hash]
   def headers
-    @headers
+    @controller.__headers
   end
 
   # Returns ETag response header or +nil+ if it's empty.

spec/rage/response_spec.rb

@@ -1,10 +1,16 @@
 # frozen_string_literal: true
 
 RSpec.describe Rage::Response do
+  subject { described_class.new(controller) }
+
+  let(:controller) { RageController::API.new(nil, nil) }
   let(:headers) { {} }
-  let(:body) { [] }
 
-  subject { described_class.new(headers, body) }
+  before do
+    headers.each do |key, value|
+      controller.headers[key] = value
+    end
+  end
 
   describe "#etag" do
     let(:etag) { "1234" }
@@ -116,11 +122,35 @@
     end
 
     context "with non-empty body" do
-      let(:body) { [:test_body] }
+      before do
+        controller.render plain: "test_body"
+      end
 
       it "returns the body" do
-        expect(subject.body).to eq(:test_body)
+        expect(subject.body).to eq("test_body")
       end
     end
   end
+
+  context "#status" do
+    it "defaults to 204" do
+      expect(subject.status).to eq(204)
+    end
+
+    it "tracks status updates via head" do
+      controller.head 401
+      expect(subject.status).to eq(401)
+
+      controller.head 403
+      expect(subject.status).to eq(403)
+    end
+
+    it "tracks status updates via render" do
+      controller.render status: 401
+      expect(subject.status).to eq(401)
+
+      controller.head 403
+      expect(subject.status).to eq(403)
+    end
+  end
 end