From 7fd0f5fc9d52e749a9692e7e6b42308b20e247f3 Mon Sep 17 00:00:00 2001
From: Gabriel Mazetto <brodock@gmail.com>
Date: Thu, 10 May 2018 05:53:21 +0200
Subject: [PATCH] Adding progressbar explicitly and spec extensions for it

---
 Gemfile                              | 3 +++
 Gemfile.lock                         | 1 +
 spec/support/helpers/rake_helpers.rb | 4 ++++
 3 files changed, 8 insertions(+)

diff --git a/Gemfile b/Gemfile
index 2df14c032ed..12bff1ee867 100644
--- a/Gemfile
+++ b/Gemfile
@@ -174,6 +174,9 @@ gem 'httparty', '~> 0.13.3'
 # Colored output to console
 gem 'rainbow', '~> 2.2'
 
+# Progress bar
+gem 'ruby-progressbar'
+
 # GitLab settings
 gem 'settingslogic', '~> 2.0.9'
 
diff --git a/Gemfile.lock b/Gemfile.lock
index ee28465decd..854a24ab0b4 100644
--- a/Gemfile.lock
+++ b/Gemfile.lock
@@ -1165,6 +1165,7 @@ DEPENDENCIES
   rubocop-rspec (~> 1.22.1)
   ruby-fogbugz (~> 0.2.1)
   ruby-prof (~> 0.17.0)
+  ruby-progressbar
   ruby_parser (~> 3.8)
   rufus-scheduler (~> 3.4)
   rugged (~> 0.27)
diff --git a/spec/support/helpers/rake_helpers.rb b/spec/support/helpers/rake_helpers.rb
index 86bfeed107c..acd9cce6a67 100644
--- a/spec/support/helpers/rake_helpers.rb
+++ b/spec/support/helpers/rake_helpers.rb
@@ -13,6 +13,10 @@ module RakeHelpers
     allow(main_object).to receive(:print)
   end
 
+  def silence_progress_bar
+    allow_any_instance_of(ProgressBar::Output).to receive(:stream).and_return(double().as_null_object)
+  end
+
   def main_object
     @main_object ||= TOPLEVEL_BINDING.eval('self')
   end