1 files changed, 23 insertions, 0 deletions
diff --git a/spec/support/helpers/rake_helpers.rb b/spec/support/helpers/rake_helpers.rb
new file mode 100644
index 00000000000..acd9cce6a67
--- /dev/null
+++ b/spec/support/helpers/rake_helpers.rb
@@ -0,0 +1,23 @@
+module RakeHelpers
+  def run_rake_task(task_name, *args)
+    Rake::Task[task_name].reenable
+    Rake.application.invoke_task("#{task_name}[#{args.join(',')}]")
+  end
+
+  def stub_warn_user_is_not_gitlab
+    allow(main_object).to receive(:warn_user_is_not_gitlab)
+  end
+
+  def silence_output
+    allow(main_object).to receive(:puts)
+    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
+end