discourse/spec/lib/shrink_uploaded_image_spec.rb

# frozen_string_literal: true

RSpec.describe ShrinkUploadedImage do
  fab!(:user) { Fabricate(:user, refresh_auto_groups: true) }

  def create_post_with_upload
    post = Fabricate(:post, raw: "<img src='#{upload.url}'>", user: user)
    post.link_post_uploads
    post
  end

  context "when local uploads are enabled" do
    let(:upload) { Fabricate(:image_upload, width: 200, height: 200) }

    it "resizes the image" do
      create_post_with_upload
      filesize_before = upload.filesize

      result =
        ShrinkUploadedImage.new(
          upload: upload,
          path: Discourse.store.path_for(upload),
          max_pixels: 10_000,
        ).perform

      expect(result).to be(true)
      expect(upload.width).to eq(100)
      expect(upload.height).to eq(100)
      expect(upload.filesize).to be < filesize_before
    end

    it "updates HotlinkedMedia records when there is an upload for downsized image" do
      OptimizedImage.downsize(
        Discourse.store.path_for(upload),
        "/tmp/smaller.png",
        "10000@",
        filename: upload.original_filename,
      )
      smaller_sha1 = Upload.generate_digest("/tmp/smaller.png")
      smaller_upload = Fabricate(:image_upload, sha1: smaller_sha1)

      post = create_post_with_upload
      post_hotlinked_media =
        PostHotlinkedMedia.create!(
          post: post,
          url: "http://example.com/images/2/2e/Longcat1.png",
          upload: upload,
          status: :downloaded,
        )

      ShrinkUploadedImage.new(
        upload: upload,
        path: Discourse.store.path_for(upload),
        max_pixels: 10_000,
      ).perform

      expect(post_hotlinked_media.reload.upload).to eq(smaller_upload)
    end

    it "returns false if the image is not used by any models" do
      result =
        ShrinkUploadedImage.new(
          upload: upload,
          path: Discourse.store.path_for(upload),
          max_pixels: 10_000,
        ).perform

      expect(result).to be(false)
    end

    it "returns false if the image cannot be shrunk more" do
      create_post_with_upload

      ShrinkUploadedImage.new(
        upload: upload,
        path: Discourse.store.path_for(upload),
        max_pixels: 10_000,
      ).perform

      upload.reload

      result =
        ShrinkUploadedImage.new(
          upload: upload,
          path: Discourse.store.path_for(upload),
          max_pixels: 10_000,
        ).perform

      expect(result).to be(false)
    end

    it "returns false when the upload is above the size limit" do
      create_post_with_upload
      SiteSetting.max_image_size_kb = 0

      result =
        ShrinkUploadedImage.new(
          upload: upload,
          path: Discourse.store.path_for(upload),
          max_pixels: 10_000,
        ).perform

      expect(result).to be(false)
    end

    it "returns false when the upload is not used in any posts" do
      Fabricate(:user, uploaded_avatar: upload)

      result =
        ShrinkUploadedImage.new(
          upload: upload,
          path: Discourse.store.path_for(upload),
          max_pixels: 10_000,
        ).perform

      expect(result).to be(false)
    end

    it "returns false if the image is invalid" do
      post = Fabricate(:post, raw: "<img src='#{upload.url}'>")
      post.link_post_uploads
      FastImage.stubs(:size).raises(FastImage::SizeNotFound.new)

      result =
        ShrinkUploadedImage.new(
          upload: upload,
          path: Discourse.store.path_for(upload),
          max_pixels: 10_000,
        ).perform

      expect(result).to be(false)
    end
  end

  context "when S3 uploads are enabled" do
    let(:upload) { Fabricate(:s3_image_upload, width: 200, height: 200) }

    before do
      setup_s3
      stub_s3_store
    end

    it "resizes the image" do
      filesize_before = upload.filesize
      create_post_with_upload

      result =
        ShrinkUploadedImage.new(
          upload: upload,
          path: Discourse.store.download(upload).path,
          max_pixels: 10_000,
        ).perform

      expect(result).to be(true)
      expect(upload.width).to eq(100)
      expect(upload.height).to eq(100)
      expect(upload.filesize).to be < filesize_before
    end
  end
end