module Museo
  module RspecIntegration
    include TestIntegration

    def self.included(base)
      base.render_views
      base.extend(ClassMethods)
    end

    module ClassMethods
      def snapshot(description)
        it "matches snapshot: #{description}" do |example|
          begin
            _museo_setup

            instance_exec(example, &Proc.new)

            expect_matching_snapshot(example)
          ensure
            _museo_teardown
          end
        end
      end
    end

    def expect_matching_snapshot(example)
      expect(Snapshot::Rspec.new(self, example.metadata).body).to(
        eq(Snapshot.sanitize_response(response.body)),
        "Snapshot did not match",
      )
    end
  end
end