Chris@1296: # encoding: utf-8
Chris@1296: #
Chris@1296: # Redmine - project management software
Chris@1296: # Copyright (C) 2006-2012  Jean-Philippe Lang
Chris@1296: #
Chris@1296: # This program is free software; you can redistribute it and/or
Chris@1296: # modify it under the terms of the GNU General Public License
Chris@1296: # as published by the Free Software Foundation; either version 2
Chris@1296: # of the License, or (at your option) any later version.
Chris@1296: #
Chris@1296: # This program is distributed in the hope that it will be useful,
Chris@1296: # but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@1296: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@1296: # GNU General Public License for more details.
Chris@1296: #
Chris@1296: # You should have received a copy of the GNU General Public License
Chris@1296: # along with this program; if not, write to the Free Software
Chris@1296: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Chris@1296: 
Chris@1296: require File.expand_path('../../test_helper', __FILE__)
Chris@1296: 
Chris@1296: class AttachmentTest < ActiveSupport::TestCase
Chris@1296:   fixtures :users, :projects, :roles, :members, :member_roles,
Chris@1296:            :enabled_modules, :issues, :trackers, :attachments
Chris@1296:   
Chris@1296:   class MockFile
Chris@1296:     attr_reader :original_filename, :content_type, :content, :size
Chris@1296:     
Chris@1296:     def initialize(attributes)
Chris@1296:       @original_filename = attributes[:original_filename]
Chris@1296:       @content_type = attributes[:content_type]
Chris@1296:       @content = attributes[:content] || "Content"
Chris@1296:       @size = content.size
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def setup
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_container_for_new_attachment_should_be_nil
Chris@1296:     assert_nil Attachment.new.container
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_create
Chris@1296:     a = Attachment.new(:container => Issue.find(1),
Chris@1296:                        :file => uploaded_test_file("testfile.txt", "text/plain"),
Chris@1296:                        :author => User.find(1))
Chris@1296:     assert a.save
Chris@1296:     assert_equal 'testfile.txt', a.filename
Chris@1296:     assert_equal 59, a.filesize
Chris@1296:     assert_equal 'text/plain', a.content_type
Chris@1296:     assert_equal 0, a.downloads
Chris@1296:     assert_equal '1478adae0d4eb06d35897518540e25d6', a.digest
Chris@1296:     assert File.exist?(a.diskfile)
Chris@1296:     assert_equal 59, File.size(a.diskfile)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_size_should_be_validated_for_new_file
Chris@1296:     with_settings :attachment_max_size => 0 do
Chris@1296:       a = Attachment.new(:container => Issue.find(1),
Chris@1296:                          :file => uploaded_test_file("testfile.txt", "text/plain"),
Chris@1296:                          :author => User.find(1))
Chris@1296:       assert !a.save
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_size_should_not_be_validated_when_copying
Chris@1296:     a = Attachment.create!(:container => Issue.find(1),
Chris@1296:                            :file => uploaded_test_file("testfile.txt", "text/plain"),
Chris@1296:                            :author => User.find(1))
Chris@1296:     with_settings :attachment_max_size => 0 do
Chris@1296:       copy = a.copy
Chris@1296:       assert copy.save
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_description_length_should_be_validated
Chris@1296:     a = Attachment.new(:description => 'a' * 300)
Chris@1296:     assert !a.save
Chris@1296:     assert_not_nil a.errors[:description]
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_destroy
Chris@1296:     a = Attachment.new(:container => Issue.find(1),
Chris@1296:                        :file => uploaded_test_file("testfile.txt", "text/plain"),
Chris@1296:                        :author => User.find(1))
Chris@1296:     assert a.save
Chris@1296:     assert_equal 'testfile.txt', a.filename
Chris@1296:     assert_equal 59, a.filesize
Chris@1296:     assert_equal 'text/plain', a.content_type
Chris@1296:     assert_equal 0, a.downloads
Chris@1296:     assert_equal '1478adae0d4eb06d35897518540e25d6', a.digest
Chris@1296:     diskfile = a.diskfile
Chris@1296:     assert File.exist?(diskfile)
Chris@1296:     assert_equal 59, File.size(a.diskfile)
Chris@1296:     assert a.destroy
Chris@1296:     assert !File.exist?(diskfile)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_destroy_should_not_delete_file_referenced_by_other_attachment
Chris@1296:     a = Attachment.create!(:container => Issue.find(1),
Chris@1296:                            :file => uploaded_test_file("testfile.txt", "text/plain"),
Chris@1296:                            :author => User.find(1))
Chris@1296:     diskfile = a.diskfile
Chris@1296: 
Chris@1296:     copy = a.copy
Chris@1296:     copy.save!
Chris@1296: 
Chris@1296:     assert File.exists?(diskfile)
Chris@1296:     a.destroy
Chris@1296:     assert File.exists?(diskfile)
Chris@1296:     copy.destroy
Chris@1296:     assert !File.exists?(diskfile)
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_create_should_auto_assign_content_type
Chris@1296:     a = Attachment.new(:container => Issue.find(1),
Chris@1296:                        :file => uploaded_test_file("testfile.txt", ""),
Chris@1296:                        :author => User.find(1))
Chris@1296:     assert a.save
Chris@1296:     assert_equal 'text/plain', a.content_type
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_identical_attachments_at_the_same_time_should_not_overwrite
Chris@1296:     a1 = Attachment.create!(:container => Issue.find(1),
Chris@1296:                             :file => uploaded_test_file("testfile.txt", ""),
Chris@1296:                             :author => User.find(1))
Chris@1296:     a2 = Attachment.create!(:container => Issue.find(1),
Chris@1296:                             :file => uploaded_test_file("testfile.txt", ""),
Chris@1296:                             :author => User.find(1))
Chris@1296:     assert a1.disk_filename != a2.disk_filename
Chris@1296:   end
Chris@1296:   
Chris@1296:   def test_filename_should_be_basenamed
Chris@1296:     a = Attachment.new(:file => MockFile.new(:original_filename => "path/to/the/file"))
Chris@1296:     assert_equal 'file', a.filename
Chris@1296:   end
Chris@1296:   
Chris@1296:   def test_filename_should_be_sanitized
Chris@1296:     a = Attachment.new(:file => MockFile.new(:original_filename => "valid:[] invalid:?%*|\"'<>chars"))
Chris@1296:     assert_equal 'valid_[] invalid_chars', a.filename
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_diskfilename
Chris@1296:     assert Attachment.disk_filename("test_file.txt") =~ /^\d{12}_test_file.txt$/
Chris@1296:     assert_equal 'test_file.txt', Attachment.disk_filename("test_file.txt")[13..-1]
Chris@1296:     assert_equal '770c509475505f37c2b8fb6030434d6b.txt', Attachment.disk_filename("test_accentué.txt")[13..-1]
Chris@1296:     assert_equal 'f8139524ebb8f32e51976982cd20a85d', Attachment.disk_filename("test_accentué")[13..-1]
Chris@1296:     assert_equal 'cbb5b0f30978ba03731d61f9f6d10011', Attachment.disk_filename("test_accentué.ça")[13..-1]
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_title
Chris@1296:     a = Attachment.new(:filename => "test.png")
Chris@1296:     assert_equal "test.png", a.title
Chris@1296: 
Chris@1296:     a = Attachment.new(:filename => "test.png", :description => "Cool image")
Chris@1296:     assert_equal "test.png (Cool image)", a.title
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_prune_should_destroy_old_unattached_attachments
Chris@1296:     Attachment.create!(:file => uploaded_test_file("testfile.txt", ""), :author_id => 1, :created_on => 2.days.ago)
Chris@1296:     Attachment.create!(:file => uploaded_test_file("testfile.txt", ""), :author_id => 1, :created_on => 2.days.ago)
Chris@1296:     Attachment.create!(:file => uploaded_test_file("testfile.txt", ""), :author_id => 1)
Chris@1296: 
Chris@1296:     assert_difference 'Attachment.count', -2 do
Chris@1296:       Attachment.prune
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   context "Attachmnet.attach_files" do
Chris@1296:     should "attach the file" do
Chris@1296:       issue = Issue.first
Chris@1296:       assert_difference 'Attachment.count' do
Chris@1296:         Attachment.attach_files(issue,
Chris@1296:           '1' => {
Chris@1296:             'file' => uploaded_test_file('testfile.txt', 'text/plain'),
Chris@1296:             'description' => 'test'
Chris@1296:           })
Chris@1296:       end
Chris@1296: 
Chris@1296:       attachment = Attachment.first(:order => 'id DESC')
Chris@1296:       assert_equal issue, attachment.container
Chris@1296:       assert_equal 'testfile.txt', attachment.filename
Chris@1296:       assert_equal 59, attachment.filesize
Chris@1296:       assert_equal 'test', attachment.description
Chris@1296:       assert_equal 'text/plain', attachment.content_type
Chris@1296:       assert File.exists?(attachment.diskfile)
Chris@1296:       assert_equal 59, File.size(attachment.diskfile)
Chris@1296:     end
Chris@1296: 
Chris@1296:     should "add unsaved files to the object as unsaved attachments" do
Chris@1296:       # Max size of 0 to force Attachment creation failures
Chris@1296:       with_settings(:attachment_max_size => 0) do
Chris@1296:         @project = Project.find(1)
Chris@1296:         response = Attachment.attach_files(@project, {
Chris@1296:                                              '1' => {'file' => mock_file, 'description' => 'test'},
Chris@1296:                                              '2' => {'file' => mock_file, 'description' => 'test'}
Chris@1296:                                            })
Chris@1296: 
Chris@1296:         assert response[:unsaved].present?
Chris@1296:         assert_equal 2, response[:unsaved].length
Chris@1296:         assert response[:unsaved].first.new_record?
Chris@1296:         assert response[:unsaved].second.new_record?
Chris@1296:         assert_equal response[:unsaved], @project.unsaved_attachments
Chris@1296:       end
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_latest_attach
Chris@1296:     set_fixtures_attachments_directory
Chris@1296:     a1 = Attachment.find(16)
Chris@1296:     assert_equal "testfile.png", a1.filename
Chris@1296:     assert a1.readable?
Chris@1296:     assert (! a1.visible?(User.anonymous))
Chris@1296:     assert a1.visible?(User.find(2))
Chris@1296:     a2 = Attachment.find(17)
Chris@1296:     assert_equal "testfile.PNG", a2.filename
Chris@1296:     assert a2.readable?
Chris@1296:     assert (! a2.visible?(User.anonymous))
Chris@1296:     assert a2.visible?(User.find(2))
Chris@1296:     assert a1.created_on < a2.created_on
Chris@1296: 
Chris@1296:     la1 = Attachment.latest_attach([a1, a2], "testfile.png")
Chris@1296:     assert_equal 17, la1.id
Chris@1296:     la2 = Attachment.latest_attach([a1, a2], "Testfile.PNG")
Chris@1296:     assert_equal 17, la2.id
Chris@1296: 
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_thumbnailable_should_be_true_for_images
Chris@1296:     assert_equal true, Attachment.new(:filename => 'test.jpg').thumbnailable?
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_thumbnailable_should_be_true_for_non_images
Chris@1296:     assert_equal false, Attachment.new(:filename => 'test.txt').thumbnailable?
Chris@1296:   end
Chris@1296: 
Chris@1296:   if convert_installed?
Chris@1296:     def test_thumbnail_should_generate_the_thumbnail
Chris@1296:       set_fixtures_attachments_directory
Chris@1296:       attachment = Attachment.find(16)
Chris@1296:       Attachment.clear_thumbnails
Chris@1296: 
Chris@1296:       assert_difference "Dir.glob(File.join(Attachment.thumbnails_storage_path, '*.thumb')).size" do
Chris@1296:         thumbnail = attachment.thumbnail
Chris@1296:         assert_equal "16_8e0294de2441577c529f170b6fb8f638_100.thumb", File.basename(thumbnail)
Chris@1296:         assert File.exists?(thumbnail)
Chris@1296:       end
Chris@1296:     end
Chris@1296:   else
Chris@1296:     puts '(ImageMagick convert not available)'
Chris@1296:   end
Chris@1296: end