lib/lookbook/entities/preview_entity.rb in lookbook-2.0.0.beta.8 vs lib/lookbook/entities/preview_entity.rb in lookbook-2.0.0.beta.9

@@ -5,11 +5,11 @@
   class PreviewEntity < Entity
     include AnnotatableEntity
     include LocatableEntity
     include NavigableEntity
 
-    delegate :render_args, to: :preview_class
+    delegate :after_render, :render_args, to: :preview_class
 
     # @api private
     attr_reader :preview_class
 
     # @api private
@@ -154,9 +154,14 @@
 
     # @api private
     def display_options
       global_options = Lookbook.config.preview_display_options
       global_options.deep_merge(fetch_config(:display_options, {}))
+    end
+
+    # @api private
+    def after_render_method
+      fetch_config(:after_render)
     end
 
     # @api private
     def layout
       preview_class.instance_variable_get(:@layout)