Implement TodoService#mark_todos_as_done_by_id

Follow-up on 52b0c26

1 files changed, 5 insertions, 1 deletions

diff --git a/app/services/todo_service.rb b/app/services/todo_service.rb
index eb833dd82ac..5a2f7946082 100644
--- a/app/services/todo_service.rb
+++ b/app/services/todo_service.rb
@@ -142,7 +142,11 @@ class TodoService
 
   # When user marks some todos as done
   def mark_todos_as_done(todos, current_user)
-    todos = current_user.todos.where(id: todos.map(&:id)) unless todos.respond_to?(:update_all)
+    mark_todos_as_done_by_id(todos.select(&:id), current_user)
+  end
+
+  def mark_todos_as_done_by_id(ids, current_user)
+    todos = current_user.todos.where(id: ids)
 
     marked_todos = todos.update_all(state: :done)
     current_user.update_todos_count_cache