lib/action_view/component/test_helpers.rb in actionview-component-1.3.6 vs lib/action_view/component/test_helpers.rb in actionview-component-1.4.0

@@ -2,11 +2,11 @@
 
 module ActionView
   module Component
     module TestHelpers
       def render_inline(component, **args, &block)
-        Nokogiri::HTML(controller.view_context.render(component, args, &block))
+        Nokogiri::HTML(controller.view_context.render(component, args, &block)).css("body > *")
       end
 
       def controller
         @controller ||= ApplicationController.new.tap { |c| c.request = request }
       end
@@ -19,9 +19,17 @@
         ActiveSupport::Deprecation.warn(
           "`render_component` has been deprecated in favor of `render_inline`, and will be removed in v2.0.0."
         )
 
         render_inline(component, args, &block)
+      end
+
+      def with_variant(variant)
+        old_variants = controller.view_context.lookup_context.variants
+
+        controller.view_context.lookup_context.variants = variant
+        yield
+        controller.view_context.lookup_context.variants = old_variants
       end
     end
   end
 end