# frozen_string_literal: true

require 'set'

module RuboCop
  module Cop
    # Helper methods for different explorations against test files and test cases.
    # @api private
    module MinitestExplorationHelpers
      include DefNode
      extend NodePattern::Macros

      ASSERTION_PREFIXES = %w[assert refute].freeze

      LIFECYCLE_HOOK_METHODS_IN_ORDER = %i[
        before_setup
        setup
        after_setup
        before_teardown
        teardown
        after_teardown
      ].freeze

      LIFECYCLE_HOOK_METHODS = LIFECYCLE_HOOK_METHODS_IN_ORDER.to_set.freeze

      private

      def test_class?(class_node)
        class_node.parent_class && class_node.identifier.source.end_with?('Test')
      end

      def test_case?(node)
        return false unless (node&.def_type? && test_method?(node)) ||
                            (node&.block_type? && test_block?(node))

        class_ancestor = node.each_ancestor(:class).first
        class_ancestor && test_class?(class_ancestor)
      end

      def test_cases(class_node, visibility_check: true)
        test_methods = class_def_nodes(class_node).select do |def_node|
          test_method?(def_node, visibility_check: visibility_check)
        end

        # Support Active Support's `test 'example' { ... }` method.
        # https://api.rubyonrails.org/classes/ActiveSupport/Testing/Declarative.html
        test_blocks = class_node.each_descendant(:block).select { |block_node| test_block?(block_node) }

        test_methods + test_blocks
      end

      def test_method?(def_node, visibility_check: true)
        return false if visibility_check && non_public?(def_node)

        test_case_name?(def_node.method_name) && !def_node.arguments?
      end

      def test_block?(block_node)
        block_node.method?(:test) || block_node.method?(:it)
      end

      def lifecycle_hooks(class_node)
        class_def_nodes(class_node)
          .select { |def_node| lifecycle_hook_method?(def_node) }
      end

      def class_def_nodes(class_node)
        class_def = class_node.body
        return [] unless class_def

        if class_def.def_type?
          [class_def]
        else
          class_def.each_child_node(:def).to_a
        end
      end

      def test_case_name?(name)
        name.to_s.start_with?('test_')
      end

      def assertions(def_node)
        method_def = def_node.body
        return [] unless method_def

        send_nodes =
          if method_def.send_type?
            [method_def]
          else
            method_def.each_child_node(:send)
          end

        send_nodes.select { |send_node| assertion_method?(send_node) }
      end

      def assertions_count(node)
        node.each_descendant(:send).count do |send_node|
          assertion_method?(send_node)
        end
      end

      def assertion_method?(node)
        return false if !node.send_type? && !node.block_type? && !node.numblock_type?

        ASSERTION_PREFIXES.any? do |prefix|
          method_name = node.method_name

          # TODO: Remove the fllowing `to_s` since Ruby 2.7 that supports `Symbol#start_with?`.
          method_name.to_s.start_with?(prefix) || node.method?(:flunk)
        end
      end

      def lifecycle_hook_method?(node)
        node.def_type? && LIFECYCLE_HOOK_METHODS.include?(node.method_name)
      end
    end
  end
end