discourse/spec/jobs/migrate_badge_image_to_uploads_spec.rb

# frozen_string_literal: true

describe Jobs::MigrateBadgeImageToUploads do
  let(:image_url) { "https://omg.aws.somestack/test.png" }
  let(:badge) { Fabricate(:badge) }

  before do
    @orig_logger = Rails.logger
    Rails.logger = @fake_logger = FakeLogger.new
  end

  after do
    Rails.logger = @orig_logger
  end

  it 'should migrate to the new badge `image_upload_id` column correctly' do
    stub_request(:get, image_url).to_return(
      status: 200, body: file_from_fixtures("smallest.png").read
    )
    DB.exec(<<~SQL, flair_url: image_url, id: badge.id)
      UPDATE badges SET image = :flair_url WHERE id = :id
    SQL

    expect do
      described_class.new.execute_onceoff({})
    end.to change { Upload.count }.by(1)

    badge.reload
    upload = Upload.last
    expect(badge.image_upload).to eq(upload)
    expect(badge.image_url).to eq(upload.url)
    expect(badge[:image]).to eq(nil)
  end

  it 'should skip badges with invalid flair URLs' do
    DB.exec("UPDATE badges SET image = 'abc' WHERE id = ?", badge.id)
    described_class.new.execute_onceoff({})
    expect(@fake_logger.warnings.count).to eq(0)
    expect(@fake_logger.errors.count).to eq(0)
  end

  # this case has a couple of hacks that are needed to test this behavior, so if it
  # starts failing randomly in the future, I'd just delete it and not bother with it
  it 'should not keep retrying forever if download fails' do
    stub_request(:get, image_url).to_return(status: 403)
    instance = described_class.new
    instance.expects(:sleep).times(2)

    DB.exec(<<~SQL, flair_url: image_url, id: badge.id)
      UPDATE badges SET image = :flair_url WHERE id = :id
    SQL

    expect do
      instance.execute_onceoff({})
    end.not_to change { Upload.count }

    badge.reload
    expect(badge.image_upload).to eq(nil)
    expect(badge.image_url).to eq(nil)
    expect(Badge.where(id: badge.id).select(:image).first[:image]).to eq(image_url)
    expect(@fake_logger.warnings.count).to eq(3)
  end
end