diff --git a/app/models/stream_entry.rb b/app/models/stream_entry.rb index c173e62572..d451e0dded 100644 --- a/app/models/stream_entry.rb +++ b/app/models/stream_entry.rb @@ -1,4 +1,5 @@ # frozen_string_literal: true + # == Schema Information # # Table name: stream_entries @@ -26,6 +27,10 @@ class StreamEntry < ApplicationRecord default_scope { where(activity_type: 'Status') } scope :with_includes, -> { includes(:account, status: STATUS_INCLUDES) } + delegate :target, :title, :content, :thread, + to: :status, + allow_nil: true + def object_type orphaned? || targeted? ? :activity : status.object_type end @@ -38,26 +43,10 @@ class StreamEntry < ApplicationRecord [:follow, :request_friend, :authorize, :reject, :unfollow, :block, :unblock, :share, :favorite].include? verb end - def target - orphaned? ? nil : status.target - end - - def title - orphaned? ? nil : status.title - end - - def content - orphaned? ? nil : status.content - end - def threaded? (verb == :favorite || object_type == :comment) && !thread.nil? end - def thread - orphaned? ? nil : status.thread - end - def mentions orphaned? ? [] : status.mentions.map(&:account) end diff --git a/spec/models/stream_entry_spec.rb b/spec/models/stream_entry_spec.rb index 45bf26899d..3b7ff5143d 100644 --- a/spec/models/stream_entry_spec.rb +++ b/spec/models/stream_entry_spec.rb @@ -26,4 +26,52 @@ RSpec.describe StreamEntry, type: :model do expect(status.stream_entry.threaded?).to be false end end + + describe 'delegated methods' do + context 'with a nil status' do + subject { described_class.new(status: nil) } + + it 'returns nil for target' do + expect(subject.target).to be_nil + end + + it 'returns nil for title' do + expect(subject.title).to be_nil + end + + it 'returns nil for content' do + expect(subject.content).to be_nil + end + + it 'returns nil for thread' do + expect(subject.thread).to be_nil + end + end + + context 'with a real status' do + let(:original) { Fabricate(:status, text: 'Test status') } + let(:status) { Fabricate(:status, reblog: original, thread: original) } + subject { described_class.new(status: status) } + + it 'delegates target' do + expect(status.target).not_to be_nil + expect(subject.target).to eq(status.target) + end + + it 'delegates title' do + expect(status.title).not_to be_nil + expect(subject.title).to eq(status.title) + end + + it 'delegates content' do + expect(status.content).not_to be_nil + expect(subject.content).to eq(status.content) + end + + it 'delegates thread' do + expect(status.thread).not_to be_nil + expect(subject.thread).to eq(status.thread) + end + end + end end