Mastodon/spec/presenters/status_relationships_presenter_spec.rb

# frozen_string_literal: true

require 'rails_helper'

RSpec.describe StatusRelationshipsPresenter do
  describe '.initialize' do
    before do
      allow(Status).to receive(:reblogs_map).with(match_array(status_ids), current_account_id).and_return(default_map)
      allow(Status).to receive(:favourites_map).with(status_ids, current_account_id).and_return(default_map)
      allow(Status).to receive(:bookmarks_map).with(status_ids, current_account_id).and_return(default_map)
      allow(Status).to receive(:mutes_map).with(anything, current_account_id).and_return(default_map)
      allow(Status).to receive(:pins_map).with(anything, current_account_id).and_return(default_map)
    end

    let(:presenter)          { described_class.new(statuses, current_account_id, **options) }
    let(:current_account_id) { Fabricate(:account).id }
    let(:statuses)           { [Fabricate(:status)] }
    let(:status_ids)         { statuses.map(&:id) + statuses.filter_map(&:reblog_of_id) }
    let(:default_map)        { { 1 => true } }

    context 'when options are not set' do
      let(:options) { {} }

      it 'sets default maps' do
        expect(presenter).to have_attributes(
          reblogs_map: eq(default_map),
          favourites_map: eq(default_map),
          bookmarks_map: eq(default_map),
          mutes_map: eq(default_map),
          pins_map: eq(default_map)
        )
      end
    end

    context 'when options[:reblogs_map] is set' do
      let(:options) { { reblogs_map: { 2 => true } } }

      it 'sets @reblogs_map merged with default_map and options[:reblogs_map]' do
        expect(presenter.reblogs_map).to eq default_map.merge(options[:reblogs_map])
      end
    end

    context 'when options[:favourites_map] is set' do
      let(:options) { { favourites_map: { 3 => true } } }

      it 'sets @favourites_map merged with default_map and options[:favourites_map]' do
        expect(presenter.favourites_map).to eq default_map.merge(options[:favourites_map])
      end
    end

    context 'when options[:bookmarks_map] is set' do
      let(:options) { { bookmarks_map: { 4 => true } } }

      it 'sets @bookmarks_map merged with default_map and options[:bookmarks_map]' do
        expect(presenter.bookmarks_map).to eq default_map.merge(options[:bookmarks_map])
      end
    end

    context 'when options[:mutes_map] is set' do
      let(:options) { { mutes_map: { 5 => true } } }

      it 'sets @mutes_map merged with default_map and options[:mutes_map]' do
        expect(presenter.mutes_map).to eq default_map.merge(options[:mutes_map])
      end
    end

    context 'when options[:pins_map] is set' do
      let(:options) { { pins_map: { 6 => true } } }

      it 'sets @pins_map merged with default_map and options[:pins_map]' do
        expect(presenter.pins_map).to eq default_map.merge(options[:pins_map])
      end
    end

    context 'when post includes filtered terms' do
      let(:statuses) { [Fabricate(:status, text: 'this toot is about that banned word'), Fabricate(:status, reblog: Fabricate(:status, text: 'this toot is about an irrelevant word'))] }
      let(:options) { {} }

      before do
        Account.find(current_account_id).custom_filters.create!(phrase: 'filter1', context: %w(home), action: :hide, keywords_attributes: [{ keyword: 'banned' }, { keyword: 'irrelevant' }])
      end

      it 'sets @filters_map to filter top-level status' do
        matched_filters = presenter.filters_map[statuses[0].id]

        expect(matched_filters)
          .to be_an(Array)
          .and have_attributes(size: 1)
          .and contain_exactly(
            have_attributes(
              filter: have_attributes(title: 'filter1'),
              keyword_matches: contain_exactly('banned')
            )
          )
      end

      it 'sets @filters_map to filter reblogged status' do
        matched_filters = presenter.filters_map[statuses[1].reblog_of_id]

        expect(matched_filters)
          .to be_an(Array)
          .and have_attributes(size: 1)
          .and contain_exactly(
            have_attributes(
              filter: have_attributes(title: 'filter1'),
              keyword_matches: contain_exactly('irrelevant')
            )
          )
      end
    end

    context 'when post includes filtered individual statuses' do
      let(:statuses) { [Fabricate(:status, text: 'hello world'), Fabricate(:status, reblog: Fabricate(:status, text: 'this toot is about an irrelevant word'))] }
      let(:options) { {} }

      before do
        filter = Account.find(current_account_id).custom_filters.create!(phrase: 'filter1', context: %w(home), action: :hide)
        filter.statuses.create!(status_id: statuses[0].id)
        filter.statuses.create!(status_id: statuses[1].reblog_of_id)
      end

      it 'sets @filters_map to filter top-level status' do
        matched_filters = presenter.filters_map[statuses[0].id]

        expect(matched_filters)
          .to be_an(Array)
          .and have_attributes(size: 1)
          .and contain_exactly(
            have_attributes(
              filter: have_attributes(title: 'filter1'),
              status_matches: contain_exactly(statuses.first.id)
            )
          )
      end

      it 'sets @filters_map to filter reblogged status' do
        matched_filters = presenter.filters_map[statuses[1].reblog_of_id]

        expect(matched_filters)
          .to be_an(Array)
          .and have_attributes(size: 1)
          .and contain_exactly(
            have_attributes(
              filter: have_attributes(title: 'filter1'),
              status_matches: contain_exactly(statuses.second.reblog_of_id)
            )
          )
      end
    end
  end
end