Contents

module Museo
  module TestIntegration
    def _museo_setup
      _add_snapshot_helper_methods
      _redefine_render_with_snapshot_layout
    end

    def _museo_teardown
      _remove_snapshot_helper_methods
    end

    def _add_snapshot_helper_methods
      @controller.class.helper do
        Museo.configuration.stubbed_methods.each do |method_name, callable|
          alias_method("_original_#{method_name}", method_name) if method_defined?(method_name)

          define_method(method_name) do |*args, &block|
            output = callable.respond_to?(:call) ? instance_exec(*args, block, &callable) : callable

            [
              "<!--",
              "Stubbed method call: #{method_name}",
              Museo.configuration.formatter.format(output),
              "-->",
              "",
            ].join("\n").html_safe
          end
        end
      end
    end

    def _remove_snapshot_helper_methods
      @controller.class.helper do
        Museo.configuration.stubbed_methods.each do |method_name, _callable|
          remove_method method_name

          if method_defined?("_original_#{method_name}")
            alias_method method_name, "_original_#{method_name}"
          end
        end
      end
    end

    def _redefine_render_with_snapshot_layout
      @controller.define_singleton_method(:render) do |*args|
        options = args.length > 1 ? args.last : {}
        super(*args[1..-2], options.merge(layout: "museo/snapshot"))
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
museo-0.3.0	lib/museo/test_integration.rb
museo-0.2.0	lib/museo/test_integration.rb
museo-0.1.0	lib/museo/test_integration.rb