1 files changed, 17 insertions, 3 deletions
diff --git a/spec/features/merge_request/user_manages_subscription_spec.rb b/spec/features/merge_request/user_manages_subscription_spec.rb
index 84387965989..0853e3cb773 100644
--- a/spec/features/merge_request/user_manages_subscription_spec.rb
+++ b/spec/features/merge_request/user_manages_subscription_spec.rb
@@ -3,16 +3,15 @@
 require 'spec_helper'
 
 RSpec.describe 'User manages subscription', :js, feature_category: :code_review_workflow do
-  include CookieHelper
-
   let(:project) { create(:project, :public, :repository) }
   let(:merge_request) { create(:merge_request, source_project: project, target_project: project) }
   let(:user) { create(:user) }
   let(:moved_mr_sidebar_enabled) { false }
+  let(:notifications_todos_buttons_enabled) { false }
 
   before do
     stub_feature_flags(moved_mr_sidebar: moved_mr_sidebar_enabled)
-    set_cookie('new-actions-popover-viewed', 'true')
+    stub_feature_flags(notifications_todos_buttons: notifications_todos_buttons_enabled)
     project.add_maintainer(user)
     sign_in(user)
 
@@ -60,4 +59,19 @@ RSpec.describe 'User manages subscription', :js, feature_category: :code_review_
       expect(page).to have_selector('.gl-toggle:not(.is-checked)')
     end
   end
+
+  context 'with notifications_todos_buttons feature flag enabled' do
+    let(:moved_mr_sidebar_enabled) { true }
+    let(:notifications_todos_buttons_enabled) { true }
+
+    it 'toggles subscription' do
+      wait_for_requests
+
+      find('[data-testid="subscribe-button"]').click
+      expect(page).to have_selector('[data-testid="notifications-off-icon"]')
+
+      find('[data-testid="subscribe-button"]').click
+      expect(page).to have_selector('[data-testid="notifications-icon"]')
+    end
+  end
 end