1 files changed, 17 insertions, 0 deletions
diff --git a/spec/support/stub_dot_com_check.rb b/spec/support/stub_dot_com_check.rb
new file mode 100644
index 00000000000..178cf2577ef
--- /dev/null
+++ b/spec/support/stub_dot_com_check.rb
@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+RSpec.configure do |config|
+  config.before(:context, :saas) do
+    # Ensure Gitlab.com? returns true during context.
+    # This is needed for let_it_be which is shared across examples,
+    # therefore the value must be changed in before_all,
+    # but RSpec prevent stubbing method calls in before_all,
+    # therefore we have to resort to temporarily swap url value.
+    @_original_gitlab_url = Gitlab.config.gitlab['url']
+    Gitlab.config.gitlab['url'] = Gitlab::Saas.com_url
+  end
+  config.after(:context, :saas) do
+    # Swap back original value
+    Gitlab.config.gitlab['url'] = @_original_gitlab_url
+  end
+end