diff options
Diffstat (limited to 'spec/models/discussion_spec.rb')
-rw-r--r-- | spec/models/discussion_spec.rb | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/spec/models/discussion_spec.rb b/spec/models/discussion_spec.rb index 718b208e8f8..0221e23ced8 100644 --- a/spec/models/discussion_spec.rb +++ b/spec/models/discussion_spec.rb @@ -3,15 +3,15 @@ require 'spec_helper' describe Discussion, model: true do subject { described_class.new([first_note, second_note, third_note]) } - let(:first_note) { create(:discussion_note_on_merge_request) } + let(:first_note) { create(:diff_note_on_merge_request) } let(:merge_request) { first_note.noteable } - let(:second_note) { create(:discussion_note_on_merge_request, in_reply_to: first_note) } - let(:third_note) { create(:discussion_note_on_merge_request) } + let(:second_note) { create(:diff_note_on_merge_request, in_reply_to: first_note) } + let(:third_note) { create(:diff_note_on_merge_request) } describe '.build' do it 'returns a discussion of the right type' do discussion = described_class.build([first_note, second_note], merge_request) - expect(discussion).to be_a(SimpleDiscussion) + expect(discussion).to be_a(DiffDiscussion) expect(discussion.notes.count).to be(2) expect(discussion.first_note).to be(first_note) expect(discussion.noteable).to be(merge_request) @@ -22,8 +22,8 @@ describe Discussion, model: true do it 'returns an array of discussions of the right type' do discussions = described_class.build_collection([first_note, second_note, third_note], merge_request) expect(discussions).to eq([ - SimpleDiscussion.new([first_note, second_note], merge_request), - SimpleDiscussion.new([third_note], merge_request) + DiffDiscussion.new([first_note, second_note], merge_request), + DiffDiscussion.new([third_note], merge_request) ]) end end |