123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268 |
- # frozen_string_literal: true
- require 'rails_helper'
- RSpec.describe Notification do
- describe '#target_status' do
- let(:notification) { Fabricate(:notification, activity: activity) }
- let(:status) { Fabricate(:status) }
- let(:reblog) { Fabricate(:status, reblog: status) }
- let(:favourite) { Fabricate(:favourite, status: status) }
- let(:mention) { Fabricate(:mention, status: status) }
- context 'when Activity is reblog' do
- let(:activity) { reblog }
- it 'returns status' do
- expect(notification.target_status).to eq status
- end
- end
- context 'when Activity is favourite' do
- let(:type) { :favourite }
- let(:activity) { favourite }
- it 'returns status' do
- expect(notification.target_status).to eq status
- end
- end
- context 'when Activity is mention' do
- let(:activity) { mention }
- it 'returns status' do
- expect(notification.target_status).to eq status
- end
- end
- end
- describe '#type' do
- it 'returns :reblog for a Status' do
- notification = described_class.new(activity: Status.new)
- expect(notification.type).to eq :reblog
- end
- it 'returns :mention for a Mention' do
- notification = described_class.new(activity: Mention.new)
- expect(notification.type).to eq :mention
- end
- it 'returns :favourite for a Favourite' do
- notification = described_class.new(activity: Favourite.new)
- expect(notification.type).to eq :favourite
- end
- it 'returns :follow for a Follow' do
- notification = described_class.new(activity: Follow.new)
- expect(notification.type).to eq :follow
- end
- end
- describe 'Setting account from activity_type' do
- context 'when activity_type is a Status' do
- it 'sets the notification from_account correctly' do
- status = Fabricate(:status)
- notification = Fabricate.build(:notification, activity_type: 'Status', activity: status)
- expect(notification.from_account).to eq(status.account)
- end
- end
- context 'when activity_type is a Follow' do
- it 'sets the notification from_account correctly' do
- follow = Fabricate(:follow)
- notification = Fabricate.build(:notification, activity_type: 'Follow', activity: follow)
- expect(notification.from_account).to eq(follow.account)
- end
- end
- context 'when activity_type is a Favourite' do
- it 'sets the notification from_account correctly' do
- favourite = Fabricate(:favourite)
- notification = Fabricate.build(:notification, activity_type: 'Favourite', activity: favourite)
- expect(notification.from_account).to eq(favourite.account)
- end
- end
- context 'when activity_type is a FollowRequest' do
- it 'sets the notification from_account correctly' do
- follow_request = Fabricate(:follow_request)
- notification = Fabricate.build(:notification, activity_type: 'FollowRequest', activity: follow_request)
- expect(notification.from_account).to eq(follow_request.account)
- end
- end
- context 'when activity_type is a Poll' do
- it 'sets the notification from_account correctly' do
- poll = Fabricate(:poll)
- notification = Fabricate.build(:notification, activity_type: 'Poll', activity: poll)
- expect(notification.from_account).to eq(poll.account)
- end
- end
- context 'when activity_type is a Report' do
- it 'sets the notification from_account correctly' do
- report = Fabricate(:report)
- notification = Fabricate.build(:notification, activity_type: 'Report', activity: report)
- expect(notification.from_account).to eq(report.account)
- end
- end
- context 'when activity_type is a Mention' do
- it 'sets the notification from_account correctly' do
- mention = Fabricate(:mention)
- notification = Fabricate.build(:notification, activity_type: 'Mention', activity: mention)
- expect(notification.from_account).to eq(mention.status.account)
- end
- end
- context 'when activity_type is an Account' do
- it 'sets the notification from_account correctly' do
- account = Fabricate(:account)
- notification = Fabricate.build(:notification, activity_type: 'Account', account: account)
- expect(notification.account).to eq(account)
- end
- end
- end
- describe '.preload_cache_collection_target_statuses' do
- subject do
- described_class.preload_cache_collection_target_statuses(notifications) do |target_statuses|
- # preload account for testing instead of using cache_collection
- Status.preload(:account).where(id: target_statuses.map(&:id))
- end
- end
- context 'when notifications are empty' do
- let(:notifications) { [] }
- it 'returns []' do
- expect(subject).to eq []
- end
- end
- context 'when notifications are present' do
- before do
- notifications.each(&:reload)
- end
- let(:mention) { Fabricate(:mention) }
- let(:status) { Fabricate(:status) }
- let(:reblog) { Fabricate(:status, reblog: Fabricate(:status)) }
- let(:follow) { Fabricate(:follow) }
- let(:follow_request) { Fabricate(:follow_request) }
- let(:favourite) { Fabricate(:favourite) }
- let(:poll) { Fabricate(:poll) }
- let(:notifications) do
- [
- Fabricate(:notification, type: :mention, activity: mention),
- Fabricate(:notification, type: :status, activity: status),
- Fabricate(:notification, type: :reblog, activity: reblog),
- Fabricate(:notification, type: :follow, activity: follow),
- Fabricate(:notification, type: :follow_request, activity: follow_request),
- Fabricate(:notification, type: :favourite, activity: favourite),
- Fabricate(:notification, type: :poll, activity: poll),
- ]
- end
- context 'with a preloaded target status' do
- it 'preloads mention' do
- expect(subject[0].type).to eq :mention
- expect(subject[0].association(:mention)).to be_loaded
- expect(subject[0].mention.association(:status)).to be_loaded
- end
- it 'preloads status' do
- expect(subject[1].type).to eq :status
- expect(subject[1].association(:status)).to be_loaded
- end
- it 'preloads reblog' do
- expect(subject[2].type).to eq :reblog
- expect(subject[2].association(:status)).to be_loaded
- expect(subject[2].status.association(:reblog)).to be_loaded
- end
- it 'preloads follow as nil' do
- expect(subject[3].type).to eq :follow
- expect(subject[3].target_status).to be_nil
- end
- it 'preloads follow_request as nill' do
- expect(subject[4].type).to eq :follow_request
- expect(subject[4].target_status).to be_nil
- end
- it 'preloads favourite' do
- expect(subject[5].type).to eq :favourite
- expect(subject[5].association(:favourite)).to be_loaded
- expect(subject[5].favourite.association(:status)).to be_loaded
- end
- it 'preloads poll' do
- expect(subject[6].type).to eq :poll
- expect(subject[6].association(:poll)).to be_loaded
- expect(subject[6].poll.association(:status)).to be_loaded
- end
- end
- context 'with a cached status' do
- it 'replaces mention' do
- expect(subject[0].type).to eq :mention
- expect(subject[0].target_status.association(:account)).to be_loaded
- expect(subject[0].target_status).to eq mention.status
- end
- it 'replaces status' do
- expect(subject[1].type).to eq :status
- expect(subject[1].target_status.association(:account)).to be_loaded
- expect(subject[1].target_status).to eq status
- end
- it 'replaces reblog' do
- expect(subject[2].type).to eq :reblog
- expect(subject[2].target_status.association(:account)).to be_loaded
- expect(subject[2].target_status).to eq reblog.reblog
- end
- it 'replaces follow' do
- expect(subject[3].type).to eq :follow
- expect(subject[3].target_status).to be_nil
- end
- it 'replaces follow_request' do
- expect(subject[4].type).to eq :follow_request
- expect(subject[4].target_status).to be_nil
- end
- it 'replaces favourite' do
- expect(subject[5].type).to eq :favourite
- expect(subject[5].target_status.association(:account)).to be_loaded
- expect(subject[5].target_status).to eq favourite.status
- end
- it 'replaces poll' do
- expect(subject[6].type).to eq :poll
- expect(subject[6].target_status.association(:account)).to be_loaded
- expect(subject[6].target_status).to eq poll.status
- end
- end
- end
- end
- end
|