Commit 4504959a authored by Bob Van Landuyt's avatar Bob Van Landuyt

Make `resolvable_types` a class method

This turns Notable::RESOLVABLE_TYPES into a
`Notable.resolvable_types`. That allows us to override it in EE.
parent 2ad5b30b
...@@ -3,14 +3,16 @@ ...@@ -3,14 +3,16 @@
module Noteable module Noteable
extend ActiveSupport::Concern extend ActiveSupport::Concern
# `Noteable` class names that support resolvable notes.
RESOLVABLE_TYPES = %w(MergeRequest).freeze
class_methods do class_methods do
# `Noteable` class names that support replying to individual notes. # `Noteable` class names that support replying to individual notes.
def replyable_types def replyable_types
%w(Issue MergeRequest) %w(Issue MergeRequest)
end end
# `Noteable` class names that support resolvable notes.
def resolvable_types
%w(MergeRequest)
end
end end
# The timestamp of the note (e.g. the :created_at or :updated_at attribute if provided via # The timestamp of the note (e.g. the :created_at or :updated_at attribute if provided via
...@@ -36,7 +38,7 @@ module Noteable ...@@ -36,7 +38,7 @@ module Noteable
end end
def supports_resolvable_notes? def supports_resolvable_notes?
RESOLVABLE_TYPES.include?(base_class_name) self.class.resolvable_types.include?(base_class_name)
end end
def supports_discussions? def supports_discussions?
...@@ -131,3 +133,5 @@ module Noteable ...@@ -131,3 +133,5 @@ module Noteable
) )
end end
end end
Noteable.extend(Noteable::ClassMethods)
...@@ -12,7 +12,7 @@ module ResolvableNote ...@@ -12,7 +12,7 @@ module ResolvableNote
validates :resolved_by, presence: true, if: :resolved? validates :resolved_by, presence: true, if: :resolved?
# Keep this scope in sync with `#potentially_resolvable?` # Keep this scope in sync with `#potentially_resolvable?`
scope :potentially_resolvable, -> { where(type: RESOLVABLE_TYPES).where(noteable_type: Noteable::RESOLVABLE_TYPES) } scope :potentially_resolvable, -> { where(type: RESOLVABLE_TYPES).where(noteable_type: Noteable.resolvable_types) }
# Keep this scope in sync with `#resolvable?` # Keep this scope in sync with `#resolvable?`
scope :resolvable, -> { potentially_resolvable.user } scope :resolvable, -> { potentially_resolvable.user }
......
...@@ -15,7 +15,9 @@ class DiffNote < Note ...@@ -15,7 +15,9 @@ class DiffNote < Note
validates :original_position, presence: true validates :original_position, presence: true
validates :position, presence: true validates :position, presence: true
validates :line_code, presence: true, line_code: true, if: :on_text? validates :line_code, presence: true, line_code: true, if: :on_text?
validates :noteable_type, inclusion: { in: noteable_types } # We need to evaluate the `noteable` types when running the validation since
# EE might have added a type when the module was prepended
validates :noteable_type, inclusion: { in: -> (_note) { noteable_types } }
validate :positions_complete validate :positions_complete
validate :verify_supported validate :verify_supported
validate :diff_refs_match_commit, if: :for_commit? validate :diff_refs_match_commit, if: :for_commit?
......
...@@ -206,7 +206,7 @@ module API ...@@ -206,7 +206,7 @@ module API
delete_note(noteable, params[:note_id]) delete_note(noteable, params[:note_id])
end end
if Noteable::RESOLVABLE_TYPES.include?(noteable_type.to_s) if Noteable.resolvable_types.include?(noteable_type.to_s)
desc "Resolve/unresolve an existing #{noteable_type.to_s.downcase} discussion" do desc "Resolve/unresolve an existing #{noteable_type.to_s.downcase} discussion" do
success Entities::Discussion success Entities::Discussion
end end
......
...@@ -260,4 +260,16 @@ describe Noteable do ...@@ -260,4 +260,16 @@ describe Noteable do
end end
end end
end end
describe '.replyable_types' do
it 'exposes the replyable types' do
expect(described_class.replyable_types).to include('Issue', 'MergeRequest')
end
end
describe '.resolvable_types' do
it 'exposes the replyable types' do
expect(described_class.resolvable_types).to include('MergeRequest')
end
end
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