1 files changed, 34 insertions, 0 deletions
diff --git a/actionpack/test/dispatch/live_response_test.rb b/actionpack/test/dispatch/live_response_test.rb
new file mode 100644
index 0000000000..0e8dcc6a16
--- /dev/null
+++ b/actionpack/test/dispatch/live_response_test.rb
@@ -0,0 +1,34 @@
+require 'abstract_unit'
+require 'active_support/concurrency/latch'
+
+module ActionController
+  module Live
+    class ResponseTest < ActiveSupport::TestCase
+      def setup
+        @response = Live::Response.new
+      end
+
+      def test_parallel
+        latch = ActiveSupport::Concurrency::Latch.new
+
+        t = Thread.new {
+          @response.stream.write 'foo'
+          latch.await
+          @response.stream.close
+        }
+
+        @response.each do |part|
+          assert_equal 'foo', part
+          latch.release
+        end
+        assert t.join
+      end
+
+      def test_setting_body_populates_buffer
+        @response.body = 'omg'
+        @response.close
+        assert_equal ['omg'], @response.body_parts
+      end
+    end
+  end
+end