Restore access to context in render? hook (#862)

Author: willcosgrove

lib/phlex/sgml.rb

@@ -56,8 +56,6 @@ def call(buffer = +"", context: {}, fragments: nil, &)
 	end
 
 	def internal_call(parent: nil, state: nil, &block)
-		return "" unless render?
-
 		if @_state
 			raise Phlex::DoubleRenderError.new(
 				"You can't render a #{self.class.name} more than once."
@@ -66,6 +64,8 @@ def internal_call(parent: nil, state: nil, &block)
 
 		@_state = state
 
+		return "" unless render?
+
 		block ||= @_content_block
 
 		Thread.current[:__phlex_component__] = [self, Fiber.current.object_id].freeze

quickdraw/sgml/conditional_rendering.test.rb

@@ -16,3 +16,16 @@ def view_template
 	assert_equal_html Example.new(render: true).call, "<h1>Hello</h1>"
 	assert_equal_html Example.new(render: false).call, ""
 end
+
+class ExampleWithContext < Phlex::HTML
+	def render? = context[:render]
+
+	def view_template
+		h1 { "Hello" }
+	end
+end
+
+test do
+	assert_equal_html ExampleWithContext.new.call(context: { render: true }), "<h1>Hello</h1>"
+	assert_equal_html ExampleWithContext.new.call(context: { render: false }), ""
+end