turbo_stream tag builder: support :partial with block

app/models/turbo/streams/tag_builder.rb

@@ -267,6 +267,8 @@ def render_template(target, content = nil, allow_inferred_rendering: true, **ren
         content.render_in(@view_context, &block)
       when content
         allow_inferred_rendering ? (render_record(content) || content) : content
+      when block_given? && (rendering.key?(:partial) || rendering.key?(:layout))
+        @view_context.render(formats: [ :html ], layout: rendering[:partial], **rendering, &block)
       when block_given?
         @view_context.capture(&block)
       when rendering.any?

bug_report_template.rb

@@ -6,7 +6,7 @@
   gem "rails"
   gem "propshaft"
   gem "puma"
-  gem "sqlite3", "~> 1.4"
+  gem "sqlite3"
   gem "turbo-rails"
 
   gem "capybara"

test/dummy/app/views/application/_partial_with_block.html.erb

@@ -0,0 +1 @@
+<p><%= yield %></p>

test/streams/streams_helper_test.rb

@@ -19,6 +19,48 @@ def to_key
 
   attr_accessor :formats
 
+  test "turbo_stream builder captures block when called without :partial keyword" do
+    rendered = turbo_stream.update "target_id" do
+      tag.span "Hello, world"
+    end
+
+    assert_dom_equal <<~HTML.strip, rendered
+      <turbo-stream action="update" target="target_id">
+        <template>
+          <span>Hello, world</span>
+        </template>
+      </turbo-stream>
+    HTML
+  end
+
+  test "turbo_stream builder forwards block to partial when called with :partial keyword" do
+    rendered = turbo_stream.update "target_id", partial: "application/partial_with_block" do
+      "Hello, from application/partial_with_block partial"
+    end
+
+    assert_dom_equal <<~HTML.strip, rendered
+      <turbo-stream action="update" target="target_id">
+        <template>
+          <p>Hello, from application/partial_with_block partial</p>
+        </template>
+      </turbo-stream>
+    HTML
+  end
+
+  test "turbo_stream builder forwards block to partial when called with :layout keyword" do
+    rendered = turbo_stream.update "target_id", layout: "application/partial_with_block" do
+      "Hello, from application/partial_with_block partial"
+    end
+
+    assert_dom_equal <<~HTML.strip, rendered
+      <turbo-stream action="update" target="target_id">
+        <template>
+          <p>Hello, from application/partial_with_block partial</p>
+        </template>
+      </turbo-stream>
+    HTML
+  end
+
   test "supports valid :renderable option object with nil content" do
     component = Component.new(id: 1, content: "Hello, world")