1 files changed, 25 insertions, 8 deletions
diff --git a/spec/helpers/merge_requests_helper_spec.rb b/spec/helpers/merge_requests_helper_spec.rb
index 3cf855229bb..38f2efd75a8 100644
--- a/spec/helpers/merge_requests_helper_spec.rb
+++ b/spec/helpers/merge_requests_helper_spec.rb
@@ -3,7 +3,6 @@
 require 'spec_helper'
 
 RSpec.describe MergeRequestsHelper do
-  include ActionView::Helpers::UrlHelper
   include ProjectForksHelper
 
   describe '#state_name_with_icon' do
@@ -72,7 +71,8 @@ RSpec.describe MergeRequestsHelper do
     let(:user) do
       double(
         assigned_open_merge_requests_count: 1,
-        review_requested_open_merge_requests_count: 2
+        review_requested_open_merge_requests_count: 2,
+        attention_requested_open_merge_requests_count: 3
       )
     end
 
@@ -82,12 +82,29 @@ RSpec.describe MergeRequestsHelper do
       allow(helper).to receive(:current_user).and_return(user)
     end
 
-    it "returns assigned, review requested and total merge request counts" do
-      expect(subject).to eq(
-        assigned: user.assigned_open_merge_requests_count,
-        review_requested: user.review_requested_open_merge_requests_count,
-        total: user.assigned_open_merge_requests_count + user.review_requested_open_merge_requests_count
-      )
+    describe 'mr_attention_requests disabled' do
+      before do
+        stub_feature_flags(mr_attention_requests: false)
+      end
+
+      it "returns assigned, review requested and total merge request counts" do
+        expect(subject).to eq(
+          assigned: user.assigned_open_merge_requests_count,
+          review_requested: user.review_requested_open_merge_requests_count,
+          total: user.assigned_open_merge_requests_count + user.review_requested_open_merge_requests_count
+        )
+      end
+    end
+
+    describe 'mr_attention_requests enabled' do
+      it "returns assigned, review requested, attention requests and total merge request counts" do
+        expect(subject).to eq(
+          assigned: user.assigned_open_merge_requests_count,
+          review_requested: user.review_requested_open_merge_requests_count,
+          attention_requested_count: user.attention_requested_open_merge_requests_count,
+          total: user.attention_requested_open_merge_requests_count
+        )
+      end
     end
   end