Fix prerender env cleanup scope in render options spec

cursoragent · cursoragent · commit 4b30b2362333 · 2026-03-24T00:35:32.000Z
diff --git a/react_on_rails/spec/react_on_rails/react_component/render_options_spec.rb b/react_on_rails/spec/react_on_rails/react_component/render_options_spec.rb
@@ -38,10 +38,16 @@ def with_prerender_env_override_cleared
   def build_render_options_with_clean_prerender_env(option:, attrs:)
     if option == :prerender
       with_prerender_env_override_cleared do
-        described_class.new(**attrs)
+        render_options = described_class.new(**attrs)
+        return yield(render_options) if block_given?
+
+        render_options
       end
     else
-      described_class.new(**attrs)
+      render_options = described_class.new(**attrs)
+      return yield(render_options) if block_given?
+
+      render_options
     end
   end
 
@@ -214,19 +220,19 @@ def build_render_options_with_clean_prerender_env(option:, attrs:)
           options = {}
           options[option] = false
           attrs = the_attrs(options: options)
-          opts = build_render_options_with_clean_prerender_env(option: option, attrs: attrs)
-
-          expect(opts.public_send(option)).to be false
+          build_render_options_with_clean_prerender_env(option: option, attrs: attrs) do |opts|
+            expect(opts.public_send(option)).to be false
+          end
         end
       end
 
       context "without #{option} option" do
         it "returns #{option} from config" do
           ReactOnRails.configuration.public_send(:"#{option}=", true)
           attrs = the_attrs
-          opts = build_render_options_with_clean_prerender_env(option: option, attrs: attrs)
-
-          expect(opts.public_send(option)).to be true
+          build_render_options_with_clean_prerender_env(option: option, attrs: attrs) do |opts|
+            expect(opts.public_send(option)).to be true
+          end
         end
       end
     end
