Add CLI area progress bar helper (#25208)

lib/mastodon/cli/base.rb

@@ -4,16 +4,39 @@ require_relative '../../../config/boot'
 require_relative '../../../config/environment'
 
 require 'thor'
-require_relative 'helper'
+require_relative 'progress_helper'
 
 module Mastodon
   module CLI
     class Base < Thor
-      include CLI::Helper
+      include ProgressHelper
 
       def self.exit_on_failure?
         true
       end
+
+      private
+
+      def pastel
+        @pastel ||= Pastel.new
+      end
+
+      def dry_run?
+        options[:dry_run]
+      end
+
+      def dry_run_mode_suffix
+        dry_run? ? ' (DRY RUN)' : ''
+      end
+
+      def reset_connection_pools!
+        ActiveRecord::Base.establish_connection(
+          ActiveRecord::Base.configurations.configs_for(env_name: Rails.env).first.configuration_hash
+            .dup
+            .tap { |config| config['pool'] = options[:concurrency] + 1 }
+        )
+        RedisConfiguration.establish_pool(options[:concurrency])
+      end
     end
   end
 end

lib/mastodon/cli/progress_helper.rb

@@ -9,23 +9,19 @@ HttpLog.configuration.logger = dev_null
 Paperclip.options[:log]      = false
 Chewy.logger                 = dev_null
 
-module Mastodon::CLI
+require 'ruby-progressbar/outputs/null'
-  module Helper
-    def dry_run?
-      options[:dry_run]
-    end
 
-    def dry_run_mode_suffix
+module Mastodon::CLI
-      dry_run? ? ' (DRY RUN)' : ''
+  module ProgressHelper
-    end
+    PROGRESS_FORMAT = '%c/%u |%b%i| %e'
 
     def create_progress_bar(total = nil)
-      ProgressBar.create(total: total, format: '%c/%u |%b%i| %e')
+      ProgressBar.create(
-    end
+        {
-
+          total: total,
-    def reset_connection_pools!
+          format: PROGRESS_FORMAT,
-      ActiveRecord::Base.establish_connection(ActiveRecord::Base.configurations[Rails.env].dup.tap { |config| config['pool'] = options[:concurrency] + 1 })
+        }.merge(progress_output_options)
-      RedisConfiguration.establish_pool(options[:concurrency])
+      )
     end
 
     def parallelize_with_progress(scope)
@@ -82,8 +78,10 @@ module Mastodon::CLI
       [total.value, aggregate.value]
     end
 
-    def pastel
+    private
-      @pastel ||= Pastel.new
+
+    def progress_output_options
+      Rails.env.test? ? { output: ProgressBar::Outputs::Null } : {}
     end
   end
 end