1 files changed, 106 insertions, 0 deletions
diff --git a/spec/services/work_items/widgets/current_user_todos_service/update_service_spec.rb b/spec/services/work_items/widgets/current_user_todos_service/update_service_spec.rb
new file mode 100644
index 00000000000..85b7e7a70df
--- /dev/null
+++ b/spec/services/work_items/widgets/current_user_todos_service/update_service_spec.rb
@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe WorkItems::Widgets::CurrentUserTodosService::UpdateService, feature_category: :team_planning do
+  let_it_be(:reporter) { create(:user) }
+  let_it_be(:project) { create(:project, :private) }
+  let_it_be(:current_user) { reporter }
+  let_it_be(:work_item) { create(:work_item, project: project) }
+
+  let_it_be(:pending_todo1) do
+    create(:todo, state: :pending, target: work_item, target_type: work_item.class.name, user: current_user)
+  end
+
+  let_it_be(:pending_todo2) do
+    create(:todo, state: :pending, target: work_item, target_type: work_item.class.name, user: current_user)
+  end
+
+  let_it_be(:done_todo) do
+    create(:todo, state: :done, target: work_item, target_type: work_item.class.name, user: current_user)
+  end
+
+  let_it_be(:other_work_item_todo) { create(:todo, state: :pending, target: create(:work_item), user: current_user) }
+  let_it_be(:other_user_todo) do
+    create(:todo, state: :pending, target: work_item, target_type: work_item.class.name, user: create(:user))
+  end
+
+  let(:widget) { work_item.widgets.find { |widget| widget.is_a?(WorkItems::Widgets::CurrentUserTodos) } }
+
+  before_all do
+    project.add_reporter(reporter)
+  end
+
+  describe '#before_update_in_transaction' do
+    subject do
+      described_class.new(widget: widget, current_user: current_user)
+        .before_update_in_transaction(params: params)
+    end
+
+    context 'when adding a todo' do
+      let(:params) { { action: "add" } }
+
+      context 'when user has no access' do
+        let(:current_user) { create(:user) }
+
+        it 'does add a todo' do
+          expect { subject }.not_to change { Todo.count }
+        end
+      end
+
+      context 'when user has access' do
+        let(:params) { { action: "add" } }
+
+        it 'creates a new todo for the user and the work item' do
+          expect { subject }.to change { current_user.todos.count }.by(1)
+
+          todo = current_user.todos.last
+
+          expect(todo.target).to eq(work_item)
+          expect(todo).to be_pending
+        end
+      end
+    end
+
+    context 'when marking as done' do
+      let(:params) { { action: "mark_as_done" } }
+
+      context 'when user has no access' do
+        let(:current_user) { create(:user) }
+
+        it 'does not change todo status' do
+          subject
+
+          expect(pending_todo1.reload).to be_pending
+          expect(pending_todo2.reload).to be_pending
+          expect(other_work_item_todo.reload).to be_pending
+          expect(other_user_todo.reload).to be_pending
+        end
+      end
+
+      context 'when resolving all todos of the work item', :aggregate_failures do
+        it 'resolves todos of the user for the work item' do
+          subject
+
+          expect(pending_todo1.reload).to be_done
+          expect(pending_todo2.reload).to be_done
+          expect(other_work_item_todo.reload).to be_pending
+          expect(other_user_todo.reload).to be_pending
+        end
+      end
+
+      context 'when resolving a specific todo', :aggregate_failures do
+        let(:params) { { action: "mark_as_done", todo_id: pending_todo1.id } }
+
+        it 'resolves todos of the user for the work item' do
+          subject
+
+          expect(pending_todo1.reload).to be_done
+          expect(pending_todo2.reload).to be_pending
+          expect(other_work_item_todo.reload).to be_pending
+          expect(other_user_todo.reload).to be_pending
+        end
+      end
+    end
+  end
+end