lib/test_prof/rspec_dissect/collectors/let.rb



# frozen_string_literal: true

require "test_prof/rspec_dissect/collectors/base"

module TestProf
  module RSpecDissect
    module Collectors # :nodoc: all
      class Let < Base
        def initialize(params)
          super(name: :let, **params)
        end

        def populate!(data)
          super
          data[:let_calls] = RSpecDissect.meta_for(name)
        end

        def print_results
          return unless RSpecDissect.memoization_available?
          super
        end

        def print_group_result(group)
          return super unless RSpecDissect.config.let_stats_enabled?
          msgs = [super]
          group[:let_calls]
            .group_by(&:itself)
            .map { |id, calls| [id, -calls.size] }
            .sort_by(&:last)
            .take(RSpecDissect.config.let_top_count)
            .each do |(id, size)|
            msgs << " ↳ #{id}#{-size}\n"
          end
          msgs.join
        end
      end
    end
  end
end