Commit 088e8dec authored by Tiago Botelho's avatar Tiago Botelho Committed by Douwe Maan

Sends only one notification per batch review

parent 310d0c31
...@@ -390,3 +390,5 @@ module NotificationRecipientService ...@@ -390,3 +390,5 @@ module NotificationRecipientService
end end
end end
end end
NotificationRecipientService.prepend(EE::NotificationRecipientService)
...@@ -23,3 +23,5 @@ class NewNoteWorker ...@@ -23,3 +23,5 @@ class NewNoteWorker
end end
# rubocop: enable CodeReuse/ActiveRecord # rubocop: enable CodeReuse/ActiveRecord
end end
NewNoteWorker.prepend(EE::NewNoteWorker)
...@@ -1878,6 +1878,7 @@ ActiveRecord::Schema.define(version: 20181206121340) do ...@@ -1878,6 +1878,7 @@ ActiveRecord::Schema.define(version: 20181206121340) do
t.integer "cached_markdown_version" t.integer "cached_markdown_version"
t.text "change_position" t.text "change_position"
t.boolean "resolved_by_push" t.boolean "resolved_by_push"
t.bigint "review_id"
t.index ["author_id"], name: "index_notes_on_author_id", using: :btree t.index ["author_id"], name: "index_notes_on_author_id", using: :btree
t.index ["commit_id"], name: "index_notes_on_commit_id", using: :btree t.index ["commit_id"], name: "index_notes_on_commit_id", using: :btree
t.index ["created_at"], name: "index_notes_on_created_at", using: :btree t.index ["created_at"], name: "index_notes_on_created_at", using: :btree
...@@ -1887,6 +1888,7 @@ ActiveRecord::Schema.define(version: 20181206121340) do ...@@ -1887,6 +1888,7 @@ ActiveRecord::Schema.define(version: 20181206121340) do
t.index ["noteable_id", "noteable_type"], name: "index_notes_on_noteable_id_and_noteable_type", using: :btree t.index ["noteable_id", "noteable_type"], name: "index_notes_on_noteable_id_and_noteable_type", using: :btree
t.index ["noteable_type"], name: "index_notes_on_noteable_type", using: :btree t.index ["noteable_type"], name: "index_notes_on_noteable_type", using: :btree
t.index ["project_id", "noteable_type"], name: "index_notes_on_project_id_and_noteable_type", using: :btree t.index ["project_id", "noteable_type"], name: "index_notes_on_project_id_and_noteable_type", using: :btree
t.index ["review_id"], name: "index_notes_on_review_id", using: :btree
end end
create_table "notification_settings", force: :cascade do |t| create_table "notification_settings", force: :cascade do |t|
...@@ -2556,6 +2558,16 @@ ActiveRecord::Schema.define(version: 20181206121340) do ...@@ -2556,6 +2558,16 @@ ActiveRecord::Schema.define(version: 20181206121340) do
t.index ["user_id"], name: "index_resource_label_events_on_user_id", using: :btree t.index ["user_id"], name: "index_resource_label_events_on_user_id", using: :btree
end end
create_table "reviews", id: :bigserial, force: :cascade do |t|
t.integer "author_id"
t.integer "merge_request_id", null: false
t.integer "project_id", null: false
t.datetime_with_timezone "created_at", null: false
t.index ["author_id"], name: "index_reviews_on_author_id", using: :btree
t.index ["merge_request_id"], name: "index_reviews_on_merge_request_id", using: :btree
t.index ["project_id"], name: "index_reviews_on_project_id", using: :btree
end
create_table "routes", force: :cascade do |t| create_table "routes", force: :cascade do |t|
t.integer "source_id", null: false t.integer "source_id", null: false
t.string "source_type", null: false t.string "source_type", null: false
...@@ -3296,6 +3308,7 @@ ActiveRecord::Schema.define(version: 20181206121340) do ...@@ -3296,6 +3308,7 @@ ActiveRecord::Schema.define(version: 20181206121340) do
add_foreign_key "namespaces", "projects", column: "file_template_project_id", name: "fk_319256d87a", on_delete: :nullify add_foreign_key "namespaces", "projects", column: "file_template_project_id", name: "fk_319256d87a", on_delete: :nullify
add_foreign_key "note_diff_files", "notes", column: "diff_note_id", on_delete: :cascade add_foreign_key "note_diff_files", "notes", column: "diff_note_id", on_delete: :cascade
add_foreign_key "notes", "projects", name: "fk_99e097b079", on_delete: :cascade add_foreign_key "notes", "projects", name: "fk_99e097b079", on_delete: :cascade
add_foreign_key "notes", "reviews", name: "fk_2e82291620", on_delete: :nullify
add_foreign_key "notification_settings", "users", name: "fk_0c95e91db7", on_delete: :cascade add_foreign_key "notification_settings", "users", name: "fk_0c95e91db7", on_delete: :cascade
add_foreign_key "oauth_openid_requests", "oauth_access_grants", column: "access_grant_id", name: "fk_oauth_openid_requests_oauth_access_grants_access_grant_id" add_foreign_key "oauth_openid_requests", "oauth_access_grants", column: "access_grant_id", name: "fk_oauth_openid_requests_oauth_access_grants_access_grant_id"
add_foreign_key "operations_feature_flags", "projects", on_delete: :cascade add_foreign_key "operations_feature_flags", "projects", on_delete: :cascade
...@@ -3360,6 +3373,9 @@ ActiveRecord::Schema.define(version: 20181206121340) do ...@@ -3360,6 +3373,9 @@ ActiveRecord::Schema.define(version: 20181206121340) do
add_foreign_key "resource_label_events", "labels", on_delete: :nullify add_foreign_key "resource_label_events", "labels", on_delete: :nullify
add_foreign_key "resource_label_events", "merge_requests", on_delete: :cascade add_foreign_key "resource_label_events", "merge_requests", on_delete: :cascade
add_foreign_key "resource_label_events", "users", on_delete: :nullify add_foreign_key "resource_label_events", "users", on_delete: :nullify
add_foreign_key "reviews", "merge_requests", on_delete: :cascade
add_foreign_key "reviews", "projects", on_delete: :cascade
add_foreign_key "reviews", "users", column: "author_id", on_delete: :nullify
add_foreign_key "saml_providers", "namespaces", column: "group_id", on_delete: :cascade add_foreign_key "saml_providers", "namespaces", column: "group_id", on_delete: :cascade
add_foreign_key "services", "projects", name: "fk_71cce407f9", on_delete: :cascade add_foreign_key "services", "projects", name: "fk_71cce407f9", on_delete: :cascade
add_foreign_key "slack_integrations", "services", on_delete: :cascade add_foreign_key "slack_integrations", "services", on_delete: :cascade
......
...@@ -337,6 +337,12 @@ bottom of the screen with two buttons: ...@@ -337,6 +337,12 @@ bottom of the screen with two buttons:
Alternatively, every pending comment has a button to finish the entire review. Alternatively, every pending comment has a button to finish the entire review.
![Review submission](img/review_preview.png) ![Review submission](img/review_preview.png)
Submitting the review will send a single email to every notifiable user of the
merge request with all the comments associated to it.
Replying to this email will, consequentially, create a new comment on the associated merge request.
## Filtering notes ## Filtering notes
> [Introduced](https://gitlab.com/gitlab-org/gitlab-ce/issues/26723) in GitLab 11.5. > [Introduced](https://gitlab.com/gitlab-org/gitlab-ce/issues/26723) in GitLab 11.5.
......
...@@ -41,9 +41,13 @@ class Projects::MergeRequests::DraftsController < Projects::MergeRequests::Appli ...@@ -41,9 +41,13 @@ class Projects::MergeRequests::DraftsController < Projects::MergeRequests::Appli
end end
def publish def publish
DraftNotes::PublishService.new(merge_request, current_user).execute(draft_note(allow_nil: true)) result = DraftNotes::PublishService.new(merge_request, current_user).execute(draft_note(allow_nil: true))
if result[:status] == :success
head :ok head :ok
else
render json: { message: result[:message] }, status: result[:status]
end
end end
def discard def discard
......
...@@ -13,6 +13,7 @@ module EE ...@@ -13,6 +13,7 @@ module EE
include ::Emails::CsvExport include ::Emails::CsvExport
include ::Emails::ServiceDesk include ::Emails::ServiceDesk
include ::Emails::Epics include ::Emails::Epics
include ::Emails::Reviews
end end
attr_reader :group attr_reader :group
......
# frozen_string_literal: true
module Emails
module Reviews
def new_review_email(recipient_id, review_id)
setup_review_email(review_id, recipient_id)
mail_answer_thread(@merge_request, review_thread_options(recipient_id))
end
private
def review_thread_options(recipient_id)
{
from: sender(@author.id),
to: recipient(recipient_id),
subject: subject("#{@merge_request.title} (#{@merge_request.to_reference})")
}
end
def setup_review_email(review_id, recipient_id)
review = Review.find_by_id(review_id)
@notes = review.notes
@author = review.author
@author_name = review.author_name
@project = review.project
@merge_request = review.merge_request
@target_url = project_merge_request_url(@project, @merge_request)
@sent_notification = SentNotification.record(@merge_request, recipient_id, reply_key)
end
end
end
...@@ -13,6 +13,8 @@ class DraftNote < ActiveRecord::Base ...@@ -13,6 +13,8 @@ class DraftNote < ActiveRecord::Base
# Text with quick actions filtered out # Text with quick actions filtered out
attr_accessor :rendered_note attr_accessor :rendered_note
attr_accessor :review
belongs_to :author, class_name: 'User' belongs_to :author, class_name: 'User'
belongs_to :merge_request belongs_to :merge_request
...@@ -91,6 +93,7 @@ class DraftNote < ActiveRecord::Base ...@@ -91,6 +93,7 @@ class DraftNote < ActiveRecord::Base
attrs.concat(DIFF_ATTRS) if on_diff? attrs.concat(DIFF_ATTRS) if on_diff?
params = slice(*attrs) params = slice(*attrs)
params[:in_reply_to_discussion_id] = discussion_id if discussion_id.present? params[:in_reply_to_discussion_id] = discussion_id if discussion_id.present?
params[:review_id] = review.id if review.present?
params params
end end
......
...@@ -11,6 +11,7 @@ module EE ...@@ -11,6 +11,7 @@ module EE
prepended do prepended do
include Elastic::MergeRequestsSearch include Elastic::MergeRequestsSearch
has_many :reviews, inverse_of: :merge_request
has_many :approvals, dependent: :delete_all # rubocop:disable Cop/ActiveRecordDependent has_many :approvals, dependent: :delete_all # rubocop:disable Cop/ActiveRecordDependent
has_many :approved_by_users, through: :approvals, source: :user has_many :approved_by_users, through: :approvals, source: :user
has_many :approvers, as: :target, dependent: :delete_all # rubocop:disable Cop/ActiveRecordDependent has_many :approvers, as: :target, dependent: :delete_all # rubocop:disable Cop/ActiveRecordDependent
......
...@@ -9,6 +9,8 @@ module EE ...@@ -9,6 +9,8 @@ module EE
include ::ObjectStorage::BackgroundMove include ::ObjectStorage::BackgroundMove
include Elastic::NotesSearch include Elastic::NotesSearch
belongs_to :review, inverse_of: :notes
scope :searchable, -> { where(system: false) } scope :searchable, -> { where(system: false) }
end end
......
...@@ -41,6 +41,7 @@ module EE ...@@ -41,6 +41,7 @@ module EE
has_one :gitlab_slack_application_service has_one :gitlab_slack_application_service
has_one :tracing_setting, class_name: 'ProjectTracingSetting' has_one :tracing_setting, class_name: 'ProjectTracingSetting'
has_many :reviews, inverse_of: :project
has_many :approvers, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent has_many :approvers, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
has_many :approver_groups, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent has_many :approver_groups, as: :target, dependent: :destroy # rubocop:disable Cop/ActiveRecordDependent
has_many :audit_events, as: :entity has_many :audit_events, as: :entity
......
...@@ -27,6 +27,7 @@ module EE ...@@ -27,6 +27,7 @@ module EE
delegate :shared_runners_minutes_limit, :shared_runners_minutes_limit=, delegate :shared_runners_minutes_limit, :shared_runners_minutes_limit=,
to: :namespace to: :namespace
has_many :reviews, foreign_key: :author_id, inverse_of: :author
has_many :epics, foreign_key: :author_id has_many :epics, foreign_key: :author_id
has_many :assigned_epics, foreign_key: :assignee_id, class_name: "Epic" has_many :assigned_epics, foreign_key: :assignee_id, class_name: "Epic"
has_many :path_locks, dependent: :destroy # rubocop: disable Cop/ActiveRecordDependent has_many :path_locks, dependent: :destroy # rubocop: disable Cop/ActiveRecordDependent
......
# frozen_string_literal: true
class Review < ApplicationRecord
include Participable
include Mentionable
belongs_to :author, class_name: 'User', foreign_key: :author_id, inverse_of: :reviews
belongs_to :merge_request, inverse_of: :reviews
belongs_to :project, inverse_of: :reviews
has_many :notes, -> { order(:id) }, inverse_of: :review
delegate :name, to: :author, prefix: true
participant :author
def all_references(current_user = nil, extractor: nil)
ext = super
notes.each do |note|
note.all_references(current_user, extractor: ext)
end
ext
end
end
# frozen_string_literal: true # frozen_string_literal: true
module DraftNotes module DraftNotes
class BaseService class BaseService < ::BaseService
attr_accessor :merge_request, :current_user, :params attr_accessor :merge_request, :current_user, :params
def initialize(merge_request, current_user, params = nil) def initialize(merge_request, current_user, params = nil)
......
...@@ -8,6 +8,11 @@ module DraftNotes ...@@ -8,6 +8,11 @@ module DraftNotes
else else
publish_draft_notes publish_draft_notes
end end
success
rescue ActiveRecord::RecordInvalid => e
message = "Unable to save #{e.record.class.name}: #{e.record.errors.full_messages.join(", ")} "
error(message)
end end
private private
...@@ -20,7 +25,21 @@ module DraftNotes ...@@ -20,7 +25,21 @@ module DraftNotes
end end
def publish_draft_notes def publish_draft_notes
return if draft_notes.empty?
if Feature.enabled?(:batch_review_notification, project)
review = Review.create!(author: current_user, merge_request: merge_request, project: project)
draft_notes.map do |draft_note|
draft_note.review = review
create_note_from_draft(draft_note)
end
notification_service.async.new_review(review)
else
draft_notes.each(&method(:create_note_from_draft)) draft_notes.each(&method(:create_note_from_draft))
end
draft_notes.delete_all draft_notes.delete_all
MergeRequests::ResolvedDiscussionNotificationService.new(project, current_user).execute(merge_request) MergeRequests::ResolvedDiscussionNotificationService.new(project, current_user).execute(merge_request)
...@@ -33,6 +52,8 @@ module DraftNotes ...@@ -33,6 +52,8 @@ module DraftNotes
note = Notes::CreateService.new(draft.project, draft.author, draft.publish_params).execute note = Notes::CreateService.new(draft.project, draft.author, draft.publish_params).execute
set_discussion_resolve_status(note, draft) set_discussion_resolve_status(note, draft)
note
end end
def set_discussion_resolve_status(note, draft_note) def set_discussion_resolve_status(note, draft_note)
......
# frozen_string_literal: true
module EE
module NotificationRecipientService
extend ActiveSupport::Concern
class_methods do
def build_new_review_recipients(*args)
NotificationRecipientService::Builder::NewReview.new(*args).notification_recipients
end
end
prepended do
module Builder
class NewReview < ::NotificationRecipientService::Builder::Base
attr_reader :review
def initialize(review)
@review = review
end
def target
review.merge_request
end
def project
review.project
end
def group
project.group
end
def build!
add_participants(review.author)
add_mentions(review.author, target: review)
add_project_watchers
add_custom_notifications
add_subscribed_users
end
# A new review is a batch of new notes
# therefore new_note subscribers should also
# receive incoming new reviews
def custom_action
:new_note
end
def acting_user
review.author
end
end
end
end
end
end
...@@ -6,6 +6,11 @@ module EE ...@@ -6,6 +6,11 @@ module EE
module NotificationService module NotificationService
extend ::Gitlab::Utils::Override extend ::Gitlab::Utils::Override
# Notify users on new review in system
def new_review(review)
send_new_review_notification(review)
end
# When we add approvers to a merge request we should send an email to: # When we add approvers to a merge request we should send an email to:
# #
# * the new approvers # * the new approvers
...@@ -70,6 +75,14 @@ module EE ...@@ -70,6 +75,14 @@ module EE
private private
def send_new_review_notification(review)
recipients = ::NotificationRecipientService.build_new_review_recipients(review)
recipients.each do |recipient|
mailer.new_review_email(recipient.user.id, review.id).deliver_later
end
end
def add_mr_approvers_email(merge_request, approvers, current_user) def add_mr_approvers_email(merge_request, approvers, current_user)
approvers.each do |approver| approvers.each do |approver|
mailer.add_merge_request_approver_email(approver.id, merge_request.id, current_user.id).deliver_later mailer.add_merge_request_approver_email(approver.id, merge_request.id, current_user.id).deliver_later
...@@ -77,7 +90,7 @@ module EE ...@@ -77,7 +90,7 @@ module EE
end end
def approve_mr_email(merge_request, project, current_user) def approve_mr_email(merge_request, project, current_user)
recipients = NotificationRecipientService.build_recipients(merge_request, current_user, action: 'approve') recipients = ::NotificationRecipientService.build_recipients(merge_request, current_user, action: 'approve')
recipients.each do |recipient| recipients.each do |recipient|
mailer.approved_merge_request_email(recipient.user.id, merge_request.id, current_user.id).deliver_later mailer.approved_merge_request_email(recipient.user.id, merge_request.id, current_user.id).deliver_later
...@@ -85,7 +98,7 @@ module EE ...@@ -85,7 +98,7 @@ module EE
end end
def unapprove_mr_email(merge_request, project, current_user) def unapprove_mr_email(merge_request, project, current_user)
recipients = NotificationRecipientService.build_recipients(merge_request, current_user, action: 'unapprove') recipients = ::NotificationRecipientService.build_recipients(merge_request, current_user, action: 'unapprove')
recipients.each do |recipient| recipients.each do |recipient|
mailer.unapproved_merge_request_email(recipient.user.id, merge_request.id, current_user.id).deliver_later mailer.unapproved_merge_request_email(recipient.user.id, merge_request.id, current_user.id).deliver_later
...@@ -110,7 +123,7 @@ module EE ...@@ -110,7 +123,7 @@ module EE
def epic_status_change_email(target, current_user, status) def epic_status_change_email(target, current_user, status)
action = status == 'reopened' ? 'reopen' : 'close' action = status == 'reopened' ? 'reopen' : 'close'
recipients = NotificationRecipientService.build_recipients( recipients = ::NotificationRecipientService.build_recipients(
target, target,
current_user, current_user,
action: action action: action
......
...@@ -8,6 +8,7 @@ module EE ...@@ -8,6 +8,7 @@ module EE
def migrate_records def migrate_records
migrate_epics migrate_epics
migrate_vulnerabilities_feedback migrate_vulnerabilities_feedback
migrate_reviews
super super
end end
...@@ -21,6 +22,10 @@ module EE ...@@ -21,6 +22,10 @@ module EE
def migrate_vulnerabilities_feedback def migrate_vulnerabilities_feedback
user.vulnerability_feedback.update_all(author_id: ghost_user.id) user.vulnerability_feedback.update_all(author_id: ghost_user.id)
end end
def migrate_reviews
user.reviews.update_all(author_id: ghost_user.id)
end
end end
end end
end end
%table{ border: "0", cellpadding:"0", cellspacing: "0", style: "width:100%;margin:0 auto;border-collapse:separate;border-spacing:0;" }
%tbody
%tr
%td{ style: "font-family:'Helvetica Neue',Helvetica,Arial,sans-serif;background-color:#ffffff;text-align:left;overflow:hidden;" }
%table{ border: "0", cellpadding: "0", cellspacing: "0", style: "width:100%;border-collapse:separate;border-spacing:0;" }
%tbody
%tr
%td{ style: "color:#333333;border-bottom:1px solid #ededed;font-size:15px;font-weight:bold;line-height:1.4;padding: 20px 0;" }
Merge request
= link_to(@merge_request.to_reference(@project), project_merge_request_url(@project, @merge_request))
was reviewed by
= link_to(@author_name, user_url(@author))
%tr
%td{ style: "overflow:hidden;font-size:14px;line-height:1.4;display:grid;" }
- @notes.each do |note|
- target_url = project_merge_request_url(@project, @merge_request, anchor: "note_#{note.id}")
= render 'note_email', note: note, diff_limit: 3, target_url: target_url, note_style: "border-bottom:1px solid #ededed;"
<%= "Merge request #{merge_request_url(@merge_request)} was reviewed by #{@author_name}" %>
--
<% @notes.each_with_index do |note, index| %>
<%= render 'note_email', note: note, diff_limit: 3 %>
<% if index != @notes.length-1 %>
--
<% end %>
<% end %>
# frozen_string_literal: true
module EE
module NewNoteWorker
extend ActiveSupport::Concern
private
# If a note belongs to a review
# We do not want to send a standalone
# notification
def skip_notification?(note)
note.review.present?
end
end
end
# frozen_string_literal: true
class CreateReviews < ActiveRecord::Migration[5.0]
DOWNTIME = false
def up
create_table :reviews, id: :bigserial do |t|
t.references :author, index: true, references: :users
t.references :merge_request, index: true, null: false, foreign_key: { on_delete: :cascade }
t.references :project, index: true, null: false, foreign_key: { on_delete: :cascade }
t.datetime_with_timezone :created_at, null: false
end
add_foreign_key :reviews, :users, column: :author_id, on_delete: :nullify # rubocop:disable Migration/AddConcurrentForeignKey
end
def down
remove_foreign_key :reviews, column: :author_id
drop_table :reviews
end
end
# frozen_string_literal: true
class AddReviewIdToNotes < ActiveRecord::Migration[5.0]
DOWNTIME = false
def change
add_column :notes, :review_id, :bigint
end
end
# frozen_string_literal: true
class AddReviewForeignKeyToNotes < ActiveRecord::Migration[5.0]
include Gitlab::Database::MigrationHelpers
DOWNTIME = false
disable_ddl_transaction!
def up
add_concurrent_foreign_key(:notes, :reviews, column: :review_id, on_delete: :nullify)
add_concurrent_index :notes, :review_id
end
def down
remove_foreign_key :notes, column: :review_id
remove_concurrent_index(:notes, :review_id)
end
end
...@@ -193,6 +193,22 @@ describe Projects::MergeRequests::DraftsController do ...@@ -193,6 +193,22 @@ describe Projects::MergeRequests::DraftsController do
end end
end end
context 'when PublishService errors' do
it 'returns message and 500 response' do
create(:draft_note, merge_request: merge_request, author: user)
error_message = "Something went wrong"
expect_next_instance_of(DraftNotes::PublishService) do |service|
allow(service).to receive(:execute).and_return({ message: error_message, status: :error })
end
post :publish, params
expect(response).to have_gitlab_http_status(:error)
expect(json_response["message"]).to include(error_message)
end
end
it 'publishes draft notes with position' do it 'publishes draft notes with position' do
diff_refs = project.commit(RepoHelpers.sample_commit.id).try(:diff_refs) diff_refs = project.commit(RepoHelpers.sample_commit.id).try(:diff_refs)
......
# frozen_string_literal: true
FactoryBot.define do
factory :review do
merge_request
association :project, :repository
author factory: :user
end
end
...@@ -5,6 +5,7 @@ issues: ...@@ -5,6 +5,7 @@ issues:
milestone: milestone:
- boards - boards
merge_requests: merge_requests:
- reviews
- approvals - approvals
- approvers - approvers
- approver_groups - approver_groups
...@@ -69,6 +70,7 @@ project: ...@@ -69,6 +70,7 @@ project:
- packages - packages
- tracing_setting - tracing_setting
- webide_pipelines - webide_pipelines
- reviews
prometheus_metrics: prometheus_metrics:
- project - project
- prometheus_alerts - prometheus_alerts
...@@ -82,3 +84,10 @@ epic_issues: ...@@ -82,3 +84,10 @@ epic_issues:
- epic - epic
tracing_setting: tracing_setting:
- project - project
notes:
- review
reviews:
- project
- merge_request
- author
- notes
--- ---
ProjectTracingSetting: ProjectTracingSetting:
- external_url - external_url
Note:
- review_id
...@@ -227,6 +227,47 @@ describe Notify do ...@@ -227,6 +227,47 @@ describe Notify do
end end
end end
describe 'merge request reviews' do
let(:review) { create(:review, project: project, merge_request: merge_request) }
let(:notes) { create_list(:notes, 3, review: review, project: project, author: review.author, noteable: merge_request) }
subject { described_class.new_review_email(recipient.id, review.id) }
it_behaves_like 'an answer to an existing thread with reply-by-email enabled' do
let(:model) { review.merge_request }
end
it_behaves_like 'it should show Gmail Actions View Merge request link'
it_behaves_like 'an unsubscribeable thread'
it 'is sent to the given recipient as the author' do
sender = subject.header[:from].addrs[0]
aggregate_failures do
expect(sender.display_name).to eq(review.author_name)
expect(sender.address).to eq(gitlab_sender)
expect(subject).to deliver_to(recipient.notification_email)
end
end
it 'contains the message from the notes of the review' do
review.notes.each do |note|
is_expected.to have_body_text note.note
end
end
it 'contains review author name' do
is_expected.to have_body_text review.author_name
end
it 'has the correct subject and body' do
aggregate_failures do
is_expected.to have_subject "Re: #{project.name} | #{merge_request.title} (#{merge_request.to_reference})"
is_expected.to have_body_text project_merge_request_path(project, merge_request)
end
end
end
describe 'mirror was hard failed' do describe 'mirror was hard failed' do
let(:project) { create(:project, :mirror, :import_hard_failed) } let(:project) { create(:project, :mirror, :import_hard_failed) }
......
...@@ -3,6 +3,10 @@ require 'spec_helper' ...@@ -3,6 +3,10 @@ require 'spec_helper'
describe Note do describe Note do
include ::EE::GeoHelpers include ::EE::GeoHelpers
describe 'associations' do
it { is_expected.to belong_to(:review).inverse_of(:notes) }
end
context 'object storage with background upload' do context 'object storage with background upload' do
before do before do
stub_uploads_object_storage(AttachmentUploader, background_upload: true) stub_uploads_object_storage(AttachmentUploader, background_upload: true)
......
...@@ -4,6 +4,7 @@ describe EE::User do ...@@ -4,6 +4,7 @@ describe EE::User do
describe 'associations' do describe 'associations' do
subject { build(:user) } subject { build(:user) }
it { is_expected.to have_many(:reviews) }
it { is_expected.to have_many(:vulnerability_feedback) } it { is_expected.to have_many(:vulnerability_feedback) }
end end
......
...@@ -8,6 +8,7 @@ describe MergeRequest do ...@@ -8,6 +8,7 @@ describe MergeRequest do
subject(:merge_request) { create(:merge_request, source_project: project, target_project: project) } subject(:merge_request) { create(:merge_request, source_project: project, target_project: project) }
describe 'associations' do describe 'associations' do
it { is_expected.to have_many(:reviews).inverse_of(:merge_request) }
it { is_expected.to have_many(:approvals).dependent(:delete_all) } it { is_expected.to have_many(:approvals).dependent(:delete_all) }
it { is_expected.to have_many(:approvers).dependent(:delete_all) } it { is_expected.to have_many(:approvers).dependent(:delete_all) }
it { is_expected.to have_many(:approver_groups).dependent(:delete_all) } it { is_expected.to have_many(:approver_groups).dependent(:delete_all) }
......
...@@ -17,6 +17,7 @@ describe Project do ...@@ -17,6 +17,7 @@ describe Project do
it { is_expected.to have_one(:import_state).class_name('ProjectImportState') } it { is_expected.to have_one(:import_state).class_name('ProjectImportState') }
it { is_expected.to have_one(:repository_state).class_name('ProjectRepositoryState').inverse_of(:project) } it { is_expected.to have_one(:repository_state).class_name('ProjectRepositoryState').inverse_of(:project) }
it { is_expected.to have_many(:reviews).inverse_of(:project) }
it { is_expected.to have_many(:path_locks) } it { is_expected.to have_many(:path_locks) }
it { is_expected.to have_many(:vulnerability_feedback) } it { is_expected.to have_many(:vulnerability_feedback) }
it { is_expected.to have_many(:sourced_pipelines) } it { is_expected.to have_many(:sourced_pipelines) }
......
# frozen_string_literal: true
require 'spec_helper'
describe Review do
describe 'associations' do
it { is_expected.to belong_to(:author).class_name('User').with_foreign_key(:author_id).inverse_of(:reviews) }
it { is_expected.to belong_to(:merge_request).inverse_of(:reviews).touch(false) }
it { is_expected.to belong_to(:project).inverse_of(:reviews) }
it { is_expected.to have_many(:notes).order(:id).inverse_of(:review) }
end
describe 'modules' do
it { is_expected.to include_module(Participable) }
it { is_expected.to include_module(Mentionable) }
end
describe '#all_references' do
it 'returns an extractor with the correct referenced users' do
user1 = create(:user, username: "foo")
user2 = create(:user, username: "bar")
review = create(:review)
project = review.project
author = review.author
create(:note, review: review, project: project, author: author, note: "cc @foo @non_existent")
create(:note, review: review, project: project, author: author, note: "cc @bar")
expect(review.all_references(author).users).to match_array([user1, user2])
end
end
describe '#participants' do
it 'includes the review author' do
project = create(:project, :public)
merge_request = create(:merge_request, source_project: project)
review = create(:review, project: project, merge_request: merge_request)
create(:note, review: review, noteable: merge_request, project: project, author: review.author)
expect(review.participants).to include(review.author)
end
end
end
...@@ -10,20 +10,86 @@ describe DraftNotes::PublishService do ...@@ -10,20 +10,86 @@ describe DraftNotes::PublishService do
DraftNotes::PublishService.new(merge_request, user).execute(draft) DraftNotes::PublishService.new(merge_request, user).execute(draft)
end end
it 'publishes a single draft note' do context 'single draft note' do
drafts = create_list(:draft_note, 2, merge_request: merge_request, author: user) let!(:drafts) { create_list(:draft_note, 2, merge_request: merge_request, author: user) }
it 'publishes' do
expect { publish(draft: drafts.first) }.to change { DraftNote.count }.by(-1).and change { Note.count }.by(1) expect { publish(draft: drafts.first) }.to change { DraftNote.count }.by(-1).and change { Note.count }.by(1)
expect(DraftNote.count).to eq(1) expect(DraftNote.count).to eq(1)
end end
it 'publishes all draft notes for a user in a merge request' do it 'does not skip notification' do
expect(Notes::CreateService).to receive(:new).with(project, user, drafts.first.publish_params).and_call_original
expect_next_instance_of(NotificationService) do |notification_service|
expect(notification_service).to receive(:new_note)
end
result = publish(draft: drafts.first)
expect(result[:status]).to eq(:success)
end
end
context 'multiple draft notes' do
before do
create_list(:draft_note, 2, merge_request: merge_request, author: user) create_list(:draft_note, 2, merge_request: merge_request, author: user)
end
expect { publish }.to change { DraftNote.count }.by(-2).and change { Note.count }.by(2) context 'when review fails to create' do
before do
expect_next_instance_of(Review) do |review|
allow(review).to receive(:save!).and_raise(ActiveRecord::RecordInvalid.new(review))
end
end
it 'does not publish any draft note' do
expect { publish }.not_to change { DraftNote.count }
end
it 'returns an error' do
result = publish
expect(result[:status]).to eq(:error)
expect(result[:message]).to match(/Unable to save Review/)
end
end
it 'returns success' do
result = publish
expect(result[:status]).to eq(:success)
end
it 'publishes all draft notes for a user in a merge request' do
expect { publish }.to change { DraftNote.count }.by(-2).and change { Note.count }.by(2).and change { Review.count }.by(1)
expect(DraftNote.count).to eq(0) expect(DraftNote.count).to eq(0)
end end
context 'when batch_review_notification feature is enabled' do
it 'sends batch notification' do
expect_next_instance_of(NotificationService) do |notification_service|
expect(notification_service).to receive_message_chain(:async, :new_review).with(kind_of(Review))
end
publish
end
end
context 'when batch_review_notification feature is disabled' do
it 'send a notification for each note' do
stub_feature_flags(batch_review_notification: false)
2.times do
expect_next_instance_of(NotificationService) do |notification_service|
expect(notification_service).to receive(:new_note).with(kind_of(Note))
end
end
publish
end
end
end
it 'only publishes the draft notes belonging to the current user' do it 'only publishes the draft notes belonging to the current user' do
other_user = create(:user) other_user = create(:user)
project.add_maintainer(other_user) project.add_maintainer(other_user)
...@@ -54,8 +120,8 @@ describe DraftNotes::PublishService do ...@@ -54,8 +120,8 @@ describe DraftNotes::PublishService do
end end
context 'with drafts that resolve discussions' do context 'with drafts that resolve discussions' do
let(:note) { create(:discussion_note_on_merge_request, noteable: merge_request, project: project) } let!(:note) { create(:discussion_note_on_merge_request, noteable: merge_request, project: project) }
let(:draft_note) { create(:draft_note, merge_request: merge_request, author: user, resolve_discussion: true, discussion_id: note.discussion.reply_id) } let!(:draft_note) { create(:draft_note, merge_request: merge_request, author: user, resolve_discussion: true, discussion_id: note.discussion.reply_id) }
it 'resolves the discussion' do it 'resolves the discussion' do
publish(draft: draft_note) publish(draft: draft_note)
......
# frozen_string_literal: true
require 'spec_helper'
describe NotificationRecipientService do
subject(:service) { described_class }
let(:project) { create(:project, :public) }
let(:other_projects) { create_list(:project, 5, :public) }
describe '#build_new_review_recipients' do
let(:merge_request) { create(:merge_request, source_project: project, target_project: project) }
let(:review) { create(:review, merge_request: merge_request, project: project, author: merge_request.author) }
let(:notes) { create_list(:note_on_merge_request, 3, review: review, noteable: review.merge_request, project: review.project) }
shared_examples 'no N+1 queries' do
it 'avoids N+1 queries', :request_store do
create_user
service.build_new_review_recipients(review)
control_count = ActiveRecord::QueryRecorder.new do
service.build_new_review_recipients(review)
end
create_user
expect { service.build_new_review_recipients(review) }.not_to exceed_query_limit(control_count)
end
end
context 'when there are multiple watchers' do
def create_user
watcher = create(:user)
create(:notification_setting, source: project, user: watcher, level: :watch)
other_projects.each do |other_project|
create(:notification_setting, source: other_project, user: watcher, level: :watch)
end
end
include_examples 'no N+1 queries'
end
context 'when there are multiple subscribers' do
def create_user
subscriber = create(:user)
merge_request.subscriptions.create(user: subscriber, project: project, subscribed: true)
end
include_examples 'no N+1 queries'
context 'when the project is private' do
before do
project.update!(visibility_level: Gitlab::VisibilityLevel::PRIVATE)
end
include_examples 'no N+1 queries'
end
end
end
end
...@@ -121,6 +121,90 @@ describe EE::NotificationService, :mailer do ...@@ -121,6 +121,90 @@ describe EE::NotificationService, :mailer do
end end
end end
context 'new review' do
let(:project) { create(:project, :repository) }
let(:user) { create(:user) }
let(:reviewer) { create(:user) }
let(:merge_request) { create(:merge_request, source_project: project, assignee: user, author: create(:user)) }
let(:review) { create(:review, merge_request: merge_request, project: project, author: reviewer) }
let(:note) { create(:diff_note_on_merge_request, project: project, noteable: merge_request, author: reviewer, review: review) }
before do
build_team(review.project, merge_request)
project.add_maintainer(merge_request.author)
project.add_maintainer(reviewer)
project.add_maintainer(merge_request.assignee)
create(:diff_note_on_merge_request,
project: project,
noteable: merge_request,
author: reviewer,
review: review,
note: "cc @mention")
end
it 'sends emails' do
expect(Notify).not_to receive(:new_review_email).with(review.author.id, review.id)
expect(Notify).not_to receive(:new_review_email).with(@unsubscriber.id, review.id)
expect(Notify).to receive(:new_review_email).with(merge_request.assignee.id, review.id).and_call_original
expect(Notify).to receive(:new_review_email).with(merge_request.author.id, review.id).and_call_original
expect(Notify).to receive(:new_review_email).with(@u_watcher.id, review.id).and_call_original
expect(Notify).to receive(:new_review_email).with(@u_mentioned.id, review.id).and_call_original
expect(Notify).to receive(:new_review_email).with(@subscriber.id, review.id).and_call_original
expect(Notify).to receive(:new_review_email).with(@watcher_and_subscriber.id, review.id).and_call_original
expect(Notify).to receive(:new_review_email).with(@subscribed_participant.id, review.id).and_call_original
subject.new_review(review)
end
def build_team(project, merge_request)
@u_watcher = create_global_setting_for(create(:user), :watch)
@u_participating = create_global_setting_for(create(:user), :participating)
@u_participant_mentioned = create_global_setting_for(create(:user, username: 'participant'), :participating)
@u_disabled = create_global_setting_for(create(:user), :disabled)
@u_mentioned = create_global_setting_for(create(:user, username: 'mention'), :mention)
@u_committer = create(:user, username: 'committer')
@u_not_mentioned = create_global_setting_for(create(:user, username: 'regular'), :participating)
@u_outsider_mentioned = create(:user, username: 'outsider')
@u_custom_global = create_global_setting_for(create(:user, username: 'custom_global'), :custom)
# subscribers
@subscriber = create :user
@unsubscriber = create :user
@subscribed_participant = create_global_setting_for(create(:user, username: 'subscribed_participant'), :participating)
@watcher_and_subscriber = create_global_setting_for(create(:user), :watch)
# User to be participant by default
# This user does not contain any record in notification settings table
# It should be treated with a :participating notification_level
@u_lazy_participant = create(:user, username: 'lazy-participant')
@u_guest_watcher = create_user_with_notification(:watch, 'guest_watching')
@u_guest_custom = create_user_with_notification(:custom, 'guest_custom')
project.add_maintainer(@u_watcher)
project.add_maintainer(@u_participating)
project.add_maintainer(@u_participant_mentioned)
project.add_maintainer(@u_disabled)
project.add_maintainer(@u_mentioned)
project.add_maintainer(@u_committer)
project.add_maintainer(@u_not_mentioned)
project.add_maintainer(@u_lazy_participant)
project.add_maintainer(@u_custom_global)
project.add_maintainer(@subscriber)
project.add_maintainer(@unsubscriber)
project.add_maintainer(@subscribed_participant)
project.add_maintainer(@watcher_and_subscriber)
merge_request.subscriptions.create(user: @unsubscribed_mentioned, subscribed: false)
merge_request.subscriptions.create(user: @subscriber, subscribed: true)
merge_request.subscriptions.create(user: @subscribed_participant, subscribed: true)
merge_request.subscriptions.create(user: @unsubscriber, subscribed: false)
# Make the watcher a subscriber to detect dupes
merge_request.subscriptions.create(user: @watcher_and_subscriber, subscribed: true)
end
end
describe 'mirror hard failed' do describe 'mirror hard failed' do
let(:user) { create(:user) } let(:user) { create(:user) }
......
...@@ -28,4 +28,13 @@ describe Users::MigrateToGhostUserService do ...@@ -28,4 +28,13 @@ describe Users::MigrateToGhostUserService do
let(:created_record) { create(:vulnerability_feedback, author: user) } let(:created_record) { create(:vulnerability_feedback, author: user) }
end end
end end
context 'reviews' do
let!(:user) { create(:user) }
let(:service) { described_class.new(user) }
include_examples "migrating a deleted user's associated records to the ghost user", Review, [:author] do
let(:created_record) { create(:review, author: user) }
end
end
end end
# frozen_string_literal: true
require 'spec_helper'
describe NewNoteWorker do
context 'when skip_notification' do
it 'does not create a new note notification' do
note = create(:note, :with_review)
expect_any_instance_of(NotificationService).not_to receive(:new_note)
subject.perform(note.id)
end
end
end
...@@ -122,6 +122,10 @@ FactoryBot.define do ...@@ -122,6 +122,10 @@ FactoryBot.define do
system true system true
end end
trait :with_review do
review
end
trait :downvote do trait :downvote do
note "thumbsdown" note "thumbsdown"
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment