discourse/spec/lib/file_helper_spec.rb

# frozen_string_literal: true

require 'file_helper'

describe FileHelper do

  let(:url) { "https://eviltrout.com/trout.png" }
  let(:png) { File.read("#{Rails.root}/spec/fixtures/images/cropped.png") }

  before do
    stub_request(:any, /https:\/\/eviltrout.com/)
    stub_request(:get, url).to_return(body: png)
  end

  describe "download" do
    it "correctly raises an OpenURI HTTP error if it gets a 404 even with redirect" do
      url = "http://fourohfour.com/404"
      stub_request(:get, url).to_return(status: 404, body: "404")

      expect do
        begin
          FileHelper.download(
            url,
            max_file_size: 10000,
            tmp_file_name: 'trouttmp',
            follow_redirect: true
          )
        rescue => e
          expect(e.io.status[0]).to eq("404")
          raise
        end
      end.to raise_error(OpenURI::HTTPError, "404 Error")
    end

    it "does not follow redirects if instructed not to" do
      url2 = "https://test.com/image.png"
      stub_request(:get, url).to_return(status: 302, body: "", headers: { location: url2 })

      missing = FileHelper.download(
        url,
        max_file_size: 10000,
        tmp_file_name: 'trouttmp',
        follow_redirect: false
      )

      expect(missing).to eq(nil)
    end

    it "does follow redirects if instructed to" do
      url2 = "https://test.com/image.png"
      stub_request(:get, url).to_return(status: 302, body: "", headers: { location: url2 })
      stub_request(:get, url2).to_return(status: 200, body: "i am the body")

      begin
        found = FileHelper.download(
          url,
          max_file_size: 10000,
          tmp_file_name: 'trouttmp',
          follow_redirect: true
        )

        expect(found.read).to eq("i am the body")
      ensure
        found&.close
        found&.unlink
      end
    end

    it "correctly raises an OpenURI HTTP error if it gets a 404" do
      url = "http://fourohfour.com/404"

      stub_request(:get, url).to_return(status: 404, body: "404")

      expect do
        begin
          FileHelper.download(
            url,
            max_file_size: 10000,
            tmp_file_name: 'trouttmp',
            follow_redirect: false
          )
        rescue => e
          expect(e.io.status[0]).to eq("404")
          raise
        end
      end.to raise_error(OpenURI::HTTPError)
    end

    it "returns a file with the image" do
      begin
        tmpfile = FileHelper.download(
          url,
          max_file_size: 10000,
          tmp_file_name: 'trouttmp'
        )

        expect(Base64.encode64(tmpfile.read)).to eq(Base64.encode64(png))
      ensure
        tmpfile&.close
        tmpfile&.unlink
      end
    end

    it "works with a protocol relative url" do
      begin
        tmpfile = FileHelper.download(
          "//eviltrout.com/trout.png",
          max_file_size: 10000,
          tmp_file_name: 'trouttmp'
        )

        expect(Base64.encode64(tmpfile.read)).to eq(Base64.encode64(png))
      ensure
        tmpfile&.close
        tmpfile&.unlink
      end
    end

    describe 'when max_file_size is exceeded' do
      it 'should return nil' do
        tmpfile = FileHelper.download(
          "//eviltrout.com/trout.png",
          max_file_size: 1,
          tmp_file_name: 'trouttmp'
        )

        expect(tmpfile).to eq(nil)
      end

      it 'is able to retain the tmpfile' do
        begin
          tmpfile = FileHelper.download(
            "//eviltrout.com/trout.png",
            max_file_size: 1,
            tmp_file_name: 'trouttmp',
            retain_on_max_file_size_exceeded: true
          )

          expect(tmpfile.closed?).to eq(false)
        ensure
          tmpfile&.close
          tmpfile&.unlink
        end
      end
    end

    describe 'when url is a jpeg' do
      let(:url) { "https://eviltrout.com/trout.jpg" }

      it "should prioritize the content type returned by the response" do
        begin
          stub_request(:get, url).to_return(body: png, headers: {
            "content-type": "image/png"
          })

          tmpfile = FileHelper.download(
            url,
            max_file_size: 10000,
            tmp_file_name: 'trouttmp'
          )

          expect(File.extname(tmpfile)).to eq('.png')
        ensure
          tmpfile&.close
          tmpfile&.unlink
        end
      end
    end
  end

end