From 619e7c8db8b2339eae8f1f51eb2c1173808039cd Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Sun, 29 Jul 2012 19:46:36 -0700
Subject: [PATCH] make sure set_response! sets the correct response object

---
 actionpack/lib/action_controller/metal/live.rb | 9 +++++++++
 actionpack/test/controller/live_stream_test.rb | 6 ++++++
 2 files changed, 15 insertions(+)

diff --git a/actionpack/lib/action_controller/metal/live.rb b/actionpack/lib/action_controller/metal/live.rb
index e98ad7e2a6..d1734f2138 100644
--- a/actionpack/lib/action_controller/metal/live.rb
+++ b/actionpack/lib/action_controller/metal/live.rb
@@ -111,5 +111,14 @@ def process(name)
 
       @_response.await_commit
     end
+
+    def set_response!(request)
+      if request.env["HTTP_VERSION"] == "HTTP/1.0"
+        super
+      else
+        @_response         = Live::Response.new
+        @_response.request = request
+      end
+    end
   end
 end
diff --git a/actionpack/test/controller/live_stream_test.rb b/actionpack/test/controller/live_stream_test.rb
index 5bb37a271d..e5fb50fca4 100644
--- a/actionpack/test/controller/live_stream_test.rb
+++ b/actionpack/test/controller/live_stream_test.rb
@@ -53,6 +53,12 @@ def build_response
       TestResponse.new
     end
 
+    def test_set_response!
+      @controller.set_response!(@request)
+      assert_kind_of(Live::Response, @controller.response)
+      assert_equal @request, @controller.response.request
+    end
+
     def test_write_to_stream
       @controller = TestController.new
       get :basic_stream
-- 
GitLab