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