Colorized FactoryBot Linter

colorized_factory_bot.rake

# frozen_string_literal: true

require 'colorize'
require 'English'

namespace :factory_bot do
  desc 'Verify that all FactoryBot factories are valid'
  task lint: :environment do
    return unless Rails.env.development?

    FactoryBot.lint factories_to_lint, traits: true
    fail ActiveRecord::Rollback
  end
end

def factories_to_lint
  FactoryBot.factories.reject do |factory|
    factory.name =~ /^old_/
  end
end

module FactoryBot
  class Linter
    class FactoryError
      def message
        message = @wrapped_error.message
        '* '.colorize(:light_red) +
          location.to_s.colorize(:light_yellow) +
          ' - '.colorize(:light_red) +
          message.to_s.colorize(:light_blue) +
          " (#{@wrapped_error.class.name})".colorize(:light_red)
      end
    end
  end
end