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: require 'attachments_controller'
Chris@1296: 
Chris@1296: # Re-raise errors caught by the controller.
Chris@1296: class AttachmentsController; def rescue_action(e) raise e end; end
Chris@1296: 
Chris@1296: class AttachmentsControllerTest < ActionController::TestCase
Chris@1296:   fixtures :users, :projects, :roles, :members, :member_roles,
Chris@1296:            :enabled_modules, :issues, :trackers, :attachments,
Chris@1296:            :versions, :wiki_pages, :wikis, :documents
Chris@1296: 
Chris@1296:   def setup
Chris@1296:     @controller = AttachmentsController.new
Chris@1296:     @request    = ActionController::TestRequest.new
Chris@1296:     @response   = ActionController::TestResponse.new
Chris@1296:     User.current = nil
Chris@1296:     set_fixtures_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def teardown
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_diff
Chris@1296:     ['inline', 'sbs'].each do |dt|
Chris@1296:       # 060719210727_changeset_utf8.diff
Chris@1296:       get :show, :id => 14, :type => dt
Chris@1296:       assert_response :success
Chris@1296:       assert_template 'diff'
Chris@1296:       assert_equal 'text/html', @response.content_type
Chris@1296:       assert_tag 'th',
Chris@1296:         :attributes => {:class => /filename/},
Chris@1296:         :content => /issues_controller.rb\t\(révision 1484\)/
Chris@1296:       assert_tag 'td',
Chris@1296:         :attributes => {:class => /line-code/},
Chris@1296:         :content => /Demande créée avec succès/
Chris@1296:     end
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_diff_replcace_cannot_convert_content
Chris@1296:     with_settings :repositories_encodings => 'UTF-8' do
Chris@1296:       ['inline', 'sbs'].each do |dt|
Chris@1296:         # 060719210727_changeset_iso8859-1.diff
Chris@1296:         get :show, :id => 5, :type => dt
Chris@1296:         assert_response :success
Chris@1296:         assert_template 'diff'
Chris@1296:         assert_equal 'text/html', @response.content_type
Chris@1296:         assert_tag 'th',
Chris@1296:           :attributes => {:class => "filename"},
Chris@1296:           :content => /issues_controller.rb\t\(r\?vision 1484\)/
Chris@1296:         assert_tag 'td',
Chris@1296:           :attributes => {:class => /line-code/},
Chris@1296:           :content => /Demande cr\?\?e avec succ\?s/
Chris@1296:       end
Chris@1296:     end
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_diff_latin_1
Chris@1296:     with_settings :repositories_encodings => 'UTF-8,ISO-8859-1' do
Chris@1296:       ['inline', 'sbs'].each do |dt|
Chris@1296:         # 060719210727_changeset_iso8859-1.diff
Chris@1296:         get :show, :id => 5, :type => dt
Chris@1296:         assert_response :success
Chris@1296:         assert_template 'diff'
Chris@1296:         assert_equal 'text/html', @response.content_type
Chris@1296:         assert_tag 'th',
Chris@1296:           :attributes => {:class => "filename"},
Chris@1296:           :content => /issues_controller.rb\t\(révision 1484\)/
Chris@1296:         assert_tag 'td',
Chris@1296:           :attributes => {:class => /line-code/},
Chris@1296:           :content => /Demande créée avec succès/
Chris@1296:       end
Chris@1296:     end
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_save_diff_type
Chris@1296:     user1 = User.find(1)
Chris@1296:     user1.pref[:diff_type] = nil
Chris@1296:     user1.preference.save
Chris@1296:     user = User.find(1)
Chris@1296:     assert_nil user.pref[:diff_type]
Chris@1296: 
Chris@1296:     @request.session[:user_id] = 1 # admin
Chris@1296:     get :show, :id => 5
Chris@1296:     assert_response :success
Chris@1296:     assert_template 'diff'
Chris@1296:     user.reload
Chris@1296:     assert_equal "inline", user.pref[:diff_type]
Chris@1296:     get :show, :id => 5, :type => 'sbs'
Chris@1296:     assert_response :success
Chris@1296:     assert_template 'diff'
Chris@1296:     user.reload
Chris@1296:     assert_equal "sbs", user.pref[:diff_type]
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_diff_show_filename_in_mercurial_export
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     a = Attachment.new(:container => Issue.find(1),
Chris@1296:                        :file => uploaded_test_file("hg-export.diff", "text/plain"),
Chris@1296:                        :author => User.find(1))
Chris@1296:     assert a.save
Chris@1296:     assert_equal 'hg-export.diff', a.filename
Chris@1296: 
Chris@1296:     get :show, :id => a.id, :type => 'inline'
Chris@1296:     assert_response :success
Chris@1296:     assert_template 'diff'
Chris@1296:     assert_equal 'text/html', @response.content_type
Chris@1296:     assert_select 'th.filename', :text => 'test1.txt'
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_text_file
Chris@1296:     get :show, :id => 4
Chris@1296:     assert_response :success
Chris@1296:     assert_template 'file'
Chris@1296:     assert_equal 'text/html', @response.content_type
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_text_file_utf_8
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     a = Attachment.new(:container => Issue.find(1),
Chris@1296:                        :file => uploaded_test_file("japanese-utf-8.txt", "text/plain"),
Chris@1296:                        :author => User.find(1))
Chris@1296:     assert a.save
Chris@1296:     assert_equal 'japanese-utf-8.txt', a.filename
Chris@1296: 
Chris@1296:     str_japanese = "\xe6\x97\xa5\xe6\x9c\xac\xe8\xaa\x9e"
Chris@1296:     str_japanese.force_encoding('UTF-8') if str_japanese.respond_to?(:force_encoding)
Chris@1296: 
Chris@1296:     get :show, :id => a.id
Chris@1296:     assert_response :success
Chris@1296:     assert_template 'file'
Chris@1296:     assert_equal 'text/html', @response.content_type
Chris@1296:     assert_tag :tag => 'th',
Chris@1296:                :content => '1',
Chris@1296:                :attributes => { :class => 'line-num' },
Chris@1296:                :sibling => { :tag => 'td', :content => /#{str_japanese}/ }
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_text_file_replcace_cannot_convert_content
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     with_settings :repositories_encodings => 'UTF-8' do
Chris@1296:       a = Attachment.new(:container => Issue.find(1),
Chris@1296:                          :file => uploaded_test_file("iso8859-1.txt", "text/plain"),
Chris@1296:                          :author => User.find(1))
Chris@1296:       assert a.save
Chris@1296:       assert_equal 'iso8859-1.txt', a.filename
Chris@1296: 
Chris@1296:       get :show, :id => a.id
Chris@1296:       assert_response :success
Chris@1296:       assert_template 'file'
Chris@1296:       assert_equal 'text/html', @response.content_type
Chris@1296:       assert_tag :tag => 'th',
Chris@1296:                  :content => '7',
Chris@1296:                  :attributes => { :class => 'line-num' },
Chris@1296:                  :sibling => { :tag => 'td', :content => /Demande cr\?\?e avec succ\?s/ }
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_text_file_latin_1
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     with_settings :repositories_encodings => 'UTF-8,ISO-8859-1' do
Chris@1296:       a = Attachment.new(:container => Issue.find(1),
Chris@1296:                          :file => uploaded_test_file("iso8859-1.txt", "text/plain"),
Chris@1296:                          :author => User.find(1))
Chris@1296:       assert a.save
Chris@1296:       assert_equal 'iso8859-1.txt', a.filename
Chris@1296: 
Chris@1296:       get :show, :id => a.id
Chris@1296:       assert_response :success
Chris@1296:       assert_template 'file'
Chris@1296:       assert_equal 'text/html', @response.content_type
Chris@1296:       assert_tag :tag => 'th',
Chris@1296:                  :content => '7',
Chris@1296:                  :attributes => { :class => 'line-num' },
Chris@1296:                  :sibling => { :tag => 'td', :content => /Demande créée avec succès/ }
Chris@1296:       end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_text_file_should_send_if_too_big
Chris@1296:     Setting.file_max_size_displayed = 512
Chris@1296:     Attachment.find(4).update_attribute :filesize, 754.kilobyte
Chris@1296: 
Chris@1296:     get :show, :id => 4
Chris@1296:     assert_response :success
Chris@1296:     assert_equal 'application/x-ruby', @response.content_type
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_other
Chris@1296:     get :show, :id => 6
Chris@1296:     assert_response :success
Chris@1296:     assert_equal 'application/octet-stream', @response.content_type
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_file_from_private_issue_without_permission
Chris@1296:     get :show, :id => 15
Chris@1296:     assert_redirected_to '/login?back_url=http%3A%2F%2Ftest.host%2Fattachments%2F15'
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_file_from_private_issue_with_permission
Chris@1296:     @request.session[:user_id] = 2
Chris@1296:     get :show, :id => 15
Chris@1296:     assert_response :success
Chris@1296:     assert_tag 'h2', :content => /private.diff/
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_file_without_container_should_be_denied
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     attachment = Attachment.create!(:file => uploaded_test_file("testfile.txt", "text/plain"), :author_id => 2)
Chris@1296: 
Chris@1296:     @request.session[:user_id] = 2
Chris@1296:     get :show, :id => attachment.id
Chris@1296:     assert_response 403
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_show_invalid_should_respond_with_404
Chris@1296:     get :show, :id => 999
Chris@1296:     assert_response 404
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_download_text_file
Chris@1296:     get :download, :id => 4
Chris@1296:     assert_response :success
Chris@1296:     assert_equal 'application/x-ruby', @response.content_type
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_download_version_file_with_issue_tracking_disabled
Chris@1296:     Project.find(1).disable_module! :issue_tracking
Chris@1296:     get :download, :id => 9
Chris@1296:     assert_response :success
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_download_should_assign_content_type_if_blank
Chris@1296:     Attachment.find(4).update_attribute(:content_type, '')
Chris@1296: 
Chris@1296:     get :download, :id => 4
Chris@1296:     assert_response :success
Chris@1296:     assert_equal 'text/x-ruby', @response.content_type
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_download_missing_file
Chris@1296:     get :download, :id => 2
Chris@1296:     assert_response 404
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_download_should_be_denied_without_permission
Chris@1296:     get :download, :id => 7
Chris@1296:     assert_redirected_to '/login?back_url=http%3A%2F%2Ftest.host%2Fattachments%2Fdownload%2F7'
Chris@1296:     set_tmp_attachments_directory
Chris@1296:   end
Chris@1296: 
Chris@1296:   if convert_installed?
Chris@1296:     def test_thumbnail
Chris@1296:       Attachment.clear_thumbnails
Chris@1296:       @request.session[:user_id] = 2
Chris@1296: 
Chris@1296:       get :thumbnail, :id => 16
Chris@1296:       assert_response :success
Chris@1296:       assert_equal 'image/png', response.content_type
Chris@1296:     end
Chris@1296: 
Chris@1296:     def test_thumbnail_should_not_exceed_maximum_size
Chris@1296:       Redmine::Thumbnail.expects(:generate).with {|source, target, size| size == 800}
Chris@1296: 
Chris@1296:       @request.session[:user_id] = 2
Chris@1296:       get :thumbnail, :id => 16, :size => 2000
Chris@1296:     end
Chris@1296: 
Chris@1296:     def test_thumbnail_should_round_size
Chris@1296:       Redmine::Thumbnail.expects(:generate).with {|source, target, size| size == 250}
Chris@1296: 
Chris@1296:       @request.session[:user_id] = 2
Chris@1296:       get :thumbnail, :id => 16, :size => 260
Chris@1296:     end
Chris@1296: 
Chris@1296:     def test_thumbnail_should_return_404_for_non_image_attachment
Chris@1296:       @request.session[:user_id] = 2
Chris@1296: 
Chris@1296:       get :thumbnail, :id => 15
Chris@1296:       assert_response 404
Chris@1296:     end
Chris@1296: 
Chris@1296:     def test_thumbnail_should_return_404_if_thumbnail_generation_failed
Chris@1296:       Attachment.any_instance.stubs(:thumbnail).returns(nil)
Chris@1296:       @request.session[:user_id] = 2
Chris@1296: 
Chris@1296:       get :thumbnail, :id => 16
Chris@1296:       assert_response 404
Chris@1296:     end
Chris@1296: 
Chris@1296:     def test_thumbnail_should_be_denied_without_permission
Chris@1296:       get :thumbnail, :id => 16
Chris@1296:       assert_redirected_to '/login?back_url=http%3A%2F%2Ftest.host%2Fattachments%2Fthumbnail%2F16'
Chris@1296:     end
Chris@1296:   else
Chris@1296:     puts '(ImageMagick convert not available)'
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_destroy_issue_attachment
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     issue = Issue.find(3)
Chris@1296:     @request.session[:user_id] = 2
Chris@1296: 
Chris@1296:     assert_difference 'issue.attachments.count', -1 do
Chris@1296:       assert_difference 'Journal.count' do
Chris@1296:         delete :destroy, :id => 1
Chris@1296:         assert_redirected_to '/projects/ecookbook'
Chris@1296:       end
Chris@1296:     end
Chris@1296:     assert_nil Attachment.find_by_id(1)
Chris@1296:     j = Journal.first(:order => 'id DESC')
Chris@1296:     assert_equal issue, j.journalized
Chris@1296:     assert_equal 'attachment', j.details.first.property
Chris@1296:     assert_equal '1', j.details.first.prop_key
Chris@1296:     assert_equal 'error281.txt', j.details.first.old_value
Chris@1296:     assert_equal User.find(2), j.user
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_destroy_wiki_page_attachment
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     @request.session[:user_id] = 2
Chris@1296:     assert_difference 'Attachment.count', -1 do
Chris@1296:       delete :destroy, :id => 3
Chris@1296:       assert_response 302
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_destroy_project_attachment
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     @request.session[:user_id] = 2
Chris@1296:     assert_difference 'Attachment.count', -1 do
Chris@1296:       delete :destroy, :id => 8
Chris@1296:       assert_response 302
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_destroy_version_attachment
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     @request.session[:user_id] = 2
Chris@1296:     assert_difference 'Attachment.count', -1 do
Chris@1296:       delete :destroy, :id => 9
Chris@1296:       assert_response 302
Chris@1296:     end
Chris@1296:   end
Chris@1296: 
Chris@1296:   def test_destroy_without_permission
Chris@1296:     set_tmp_attachments_directory
Chris@1296:     assert_no_difference 'Attachment.count' do
Chris@1296:       delete :destroy, :id => 3
Chris@1296:     end
Chris@1296:     assert_response 302
Chris@1296:     assert Attachment.find_by_id(3)
Chris@1296:   end
Chris@1296: end