discourse/spec/models/upload_spec.rb

require 'spec_helper'
require 'digest/sha1'

describe Upload do

  it { should belong_to :user }

  it { should have_many :post_uploads }
  it { should have_many :posts }

  it { should have_many :optimized_images }

  it { should validate_presence_of :original_filename }
  it { should validate_presence_of :filesize }

  context '.create_for' do

    let(:user_id) { 1 }

    let(:logo) do
      ActionDispatch::Http::UploadedFile.new({
        filename: 'logo.png',
        content_type: 'image/png',
        tempfile: File.new("#{Rails.root}/spec/fixtures/images/logo.png")
      })
    end

    let(:upload) { Upload.create_for(user_id, logo) }

    let(:url) { "http://domain.com" }

    shared_examples_for "upload" do
      it "is valid" do
        upload.user_id.should == user_id
        upload.original_filename.should == logo.original_filename
        upload.filesize.should == File.size(logo.tempfile)
        upload.sha1.should == Digest::SHA1.file(logo.tempfile).hexdigest
        upload.width.should == 244
        upload.height.should == 66
        upload.url.should == url
      end
    end

    context "s3" do
      before(:each) do
        SiteSetting.stubs(:enable_s3_uploads?).returns(true)
        S3.stubs(:store_file).returns(url)
      end

      it_behaves_like "upload"

    end

    context "locally" do
      before(:each) { LocalStore.stubs(:store_file).returns(url) }
      it_behaves_like "upload"
    end

  end

  context 'has_been_uploaded?' do

    it "identifies internal or relatives urls" do
      Discourse.expects(:base_url_no_prefix).returns("http://discuss.site.com")
      Upload.has_been_uploaded?("http://discuss.site.com/upload/1234/42/0123456789ABCDEF.jpg").should == true
      Upload.has_been_uploaded?("/upload/42/0123456789ABCDEF.jpg").should == true
    end

    it "identifies internal urls when using a CDN" do
      ActionController::Base.expects(:asset_host).returns("http://my.cdn.com").twice
      Upload.has_been_uploaded?("http://my.cdn.com/upload/1234/42/0123456789ABCDEF.jpg").should == true
    end

    it "identifies S3 uploads" do
      SiteSetting.stubs(:enable_s3_uploads).returns(true)
      SiteSetting.stubs(:s3_upload_bucket).returns("bucket")
      Upload.has_been_uploaded?("//bucket.s3.amazonaws.com/1337.png").should == true
    end

    it "identifies external urls" do
      Upload.has_been_uploaded?("http://domain.com/upload/1234/42/0123456789ABCDEF.jpg").should == false
      Upload.has_been_uploaded?("//bucket.s3.amazonaws.com/1337.png").should == false
    end

  end

end