1 files changed, 82 insertions, 0 deletions
diff --git a/spec/support/shared_examples/quick_actions/issuable/remove_time_spent_quick_action_shared_examples.rb b/spec/support/shared_examples/quick_actions/issuable/remove_time_spent_quick_action_shared_examples.rb
new file mode 100644
index 00000000000..6d5894b2318
--- /dev/null
+++ b/spec/support/shared_examples/quick_actions/issuable/remove_time_spent_quick_action_shared_examples.rb
@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+shared_examples 'remove_time_spent quick action' do |issuable_type|
+  before do
+    project.add_maintainer(maintainer)
+    gitlab_sign_in(maintainer)
+  end
+
+  context "new #{issuable_type}", :js do
+    before do
+      case issuable_type
+      when :merge_request
+        visit public_send('namespace_project_new_merge_request_path', project.namespace, project, new_url_opts)
+        wait_for_all_requests
+      when :issue
+        visit public_send('new_namespace_project_issue_path', project.namespace, project, new_url_opts)
+        wait_for_all_requests
+      end
+    end
+
+    it "creates the #{issuable_type} and interprets remove_time_spent quick action accordingly" do
+      fill_in "#{issuable_type}_title", with: 'bug 345'
+      fill_in "#{issuable_type}_description", with: "bug description\n/remove_time_spent"
+      click_button "Submit #{issuable_type}".humanize
+
+      issuable = project.public_send(issuable_type.to_s.pluralize).first
+
+      expect(issuable.description).to eq 'bug description'
+      expect(page).to have_content 'bug 345'
+      expect(page).to have_content 'bug description'
+      expect(issuable.total_time_spent).to eq 0
+    end
+  end
+
+  context "post note to existing #{issuable_type}" do
+    before do
+      issuable.update!(spend_time: { duration: 36180, user_id: maintainer.id })
+      visit public_send("project_#{issuable_type}_path", project, issuable)
+      wait_for_all_requests
+    end
+
+    it 'creates the note and interprets the remove_time_spent quick action accordingly' do
+      add_note("/remove_time_spent")
+
+      wait_for_requests
+      expect(page).not_to have_content '/remove_time_spent'
+      expect(page).to have_content 'Commands applied'
+      expect(issuable.reload.total_time_spent).to eq 0
+    end
+
+    context "when current user cannot set remove_time_spent time" do
+      before do
+        guest = create(:user)
+        project.add_guest(guest)
+
+        gitlab_sign_out
+        gitlab_sign_in(guest)
+        visit public_send("project_#{issuable_type}_path", project, issuable)
+        wait_for_all_requests
+      end
+
+      it 'does not set remove_time_spent time' do
+        add_note("/remove_time_spent")
+
+        wait_for_requests
+        expect(page).not_to have_content '/remove_time_spent'
+        expect(issuable.reload.total_time_spent).to eq 36180
+      end
+    end
+  end
+
+  context "preview of note on #{issuable_type}", :js do
+    it 'explains remove_time_spent quick action' do
+      visit public_send("project_#{issuable_type}_path", project, issuable)
+
+      preview_note('/remove_time_spent')
+
+      expect(page).not_to have_content '/remove_time_spent'
+      expect(page).to have_content 'Removes spent time.'
+    end
+  end
+end