Capture should use yield_content

.ruby-version

@@ -1 +1 @@
-3.2.0
+3.2.1

lib/phlex/html.rb

@@ -152,14 +152,14 @@ def unsafe_raw(content = nil)
 			@_target << content
 		end
 
-		def capture
+		def capture(&block)
 			return unless block_given?
 
 			original_buffer = @_target
 			new_buffer = +""
 			@_target = new_buffer
 
-			yield
+			yield_content(&block)
 
 			new_buffer
 		ensure

test/phlex/view/capture.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+describe Phlex::HTML do
+	extend ViewHelper
+
+	with "a return value" do
+		view do
+			attr_accessor :captured
+
+			def template
+				@captured = capture { "Hello" }
+			end
+		end
+
+		it "captures the return value" do
+			example.call
+			expect(example.captured).to be == "Hello"
+		end
+	end
+
+	with "a block" do
+		view do
+			attr_accessor :captured
+
+			def template
+				@captured = capture do
+					h1 { "Hello" }
+				end
+			end
+		end
+
+		it "captures the return value" do
+			example.call
+			expect(example.captured).to be == "<h1>Hello</h1>"
+		end
+	end
+end