mastodon/spec/lib/mastodon/cli/media_spec.rb

# frozen_string_literal: true

require 'rails_helper'
require 'mastodon/cli/media'

describe Mastodon::CLI::Media do
  let(:cli) { described_class.new }

  it_behaves_like 'CLI Command'

  describe '#remove' do
    context 'with --prune-profiles and --remove-headers' do
      let(:options) { { prune_profiles: true, remove_headers: true } }

      it 'warns about usage and exits' do
        expect { cli.invoke(:remove, [], options) }.to output(
          a_string_including('--prune-profiles and --remove-headers should not be specified simultaneously')
        ).to_stdout.and raise_error(SystemExit)
      end
    end

    context 'with --include-follows but not including --prune-profiles and --remove-headers' do
      let(:options) { { include_follows: true } }

      it 'warns about usage and exits' do
        expect { cli.invoke(:remove, [], options) }.to output(
          a_string_including('--include-follows can only be used with --prune-profiles or --remove-headers')
        ).to_stdout.and raise_error(SystemExit)
      end
    end

    context 'with a relevant account' do
      let!(:account) do
        Fabricate(:account, domain: 'example.com', updated_at: 1.month.ago, last_webfingered_at: 1.month.ago, avatar: attachment_fixture('attachment.jpg'), header: attachment_fixture('attachment.jpg'))
      end

      context 'with --prune-profiles' do
        let(:options) { { prune_profiles: true } }

        it 'removes account avatars' do
          expect { cli.invoke(:remove, [], options) }.to output(
            a_string_including('Visited 1')
          ).to_stdout

          expect(account.reload.avatar).to be_blank
        end
      end

      context 'with --remove-headers' do
        let(:options) { { remove_headers: true } }

        it 'removes account header' do
          expect { cli.invoke(:remove, [], options) }.to output(
            a_string_including('Visited 1')
          ).to_stdout

          expect(account.reload.header).to be_blank
        end
      end
    end

    context 'with a relevant media attachment' do
      let!(:media_attachment) { Fabricate(:media_attachment, remote_url: 'https://example.com/image.jpg', created_at: 1.month.ago) }

      context 'without options' do
        it 'removes account avatars' do
          expect { cli.invoke(:remove) }.to output(
            a_string_including('Removed 1')
          ).to_stdout

          expect(media_attachment.reload.file).to be_blank
          expect(media_attachment.reload.thumbnail).to be_blank
        end
      end
    end
  end

  describe '#usage' do
    context 'without options' do
      let(:options) { {} }

      it 'reports about storage size' do
        expect { cli.invoke(:usage, [], options) }.to output(
          a_string_including('0 Bytes')
        ).to_stdout
      end
    end
  end

  describe '#refresh' do
    context 'without any options' do
      let(:options) { {} }

      it 'warns about usage and exits' do
        expect { cli.invoke(:refresh, [], options) }.to output(
          a_string_including('Specify the source')
        ).to_stdout.and raise_error(SystemExit)
      end
    end

    context 'with --status option' do
      before do
        media_attachment.update(file_file_name: nil)
      end

      let(:media_attachment) { Fabricate(:media_attachment, status: status, remote_url: 'https://host.example/asset.jpg') }
      let(:options) { { status: status.id } }
      let(:status) { Fabricate(:status) }

      it 'redownloads the attachment file' do
        expect { cli.invoke(:refresh, [], options) }.to output(
          a_string_including('Downloaded 1 media')
        ).to_stdout
      end
    end

    context 'with --account option' do
      context 'when the account does not exist' do
        let(:options) { { account: 'not-real-user@example.host' } }

        it 'warns about usage and exits' do
          expect { cli.invoke(:refresh, [], options) }.to output(
            a_string_including('No such account')
          ).to_stdout.and raise_error(SystemExit)
        end
      end

      context 'when the account exists' do
        before do
          media_attachment.update(file_file_name: nil)
        end

        let(:media_attachment) { Fabricate(:media_attachment, account: account) }
        let(:options) { { account: account.acct } }
        let(:account) { Fabricate(:account) }

        it 'redownloads the attachment file' do
          expect { cli.invoke(:refresh, [], options) }.to output(
            a_string_including('Downloaded 1 media')
          ).to_stdout
        end
      end
    end

    context 'with --domain option' do
      before do
        media_attachment.update(file_file_name: nil)
      end

      let(:domain) { 'example.host' }
      let(:media_attachment) { Fabricate(:media_attachment, account: account) }
      let(:options) { { domain: domain } }
      let(:account) { Fabricate(:account, domain: domain) }

      it 'redownloads the attachment file' do
        expect { cli.invoke(:refresh, [], options) }.to output(
          a_string_including('Downloaded 1 media')
        ).to_stdout
      end
    end
  end
end