69 lines
1.6 KiB
Ruby
69 lines
1.6 KiB
Ruby
require File.join(File.dirname(__FILE__), '..', 'abstract_unit')
|
|
|
|
|
|
module TestFileUtils
|
|
def file_name() File.basename(__FILE__) end
|
|
def file_path() File.expand_path(__FILE__) end
|
|
def file_data() File.open(file_path, 'rb') { |f| f.read } end
|
|
end
|
|
|
|
|
|
class SendFileController < ActionController::Base
|
|
include TestFileUtils
|
|
|
|
attr_writer :options
|
|
def options() @options ||= {} end
|
|
|
|
def file() send_file(file_path, options) end
|
|
def data() send_data(file_data, options) end
|
|
|
|
def rescue_action(e) raise end
|
|
end
|
|
|
|
|
|
class SendFileTest < Test::Unit::TestCase
|
|
include TestFileUtils
|
|
|
|
def setup
|
|
@controller = SendFileController.new
|
|
@request = ActionController::TestRequest.new
|
|
@response = ActionController::TestResponse.new
|
|
end
|
|
|
|
def test_file_nostream
|
|
@controller.options = { :stream => false }
|
|
response = nil
|
|
assert_nothing_raised { response = process('file') }
|
|
assert_not_nil response
|
|
assert_kind_of String, response.body
|
|
assert_equal file_data, response.body
|
|
end
|
|
|
|
def test_file_stream
|
|
response = nil
|
|
assert_nothing_raised { response = process('file') }
|
|
assert_not_nil response
|
|
assert_kind_of Proc, response.body
|
|
|
|
old_stdout = $stdout
|
|
begin
|
|
require 'stringio'
|
|
$stdout = StringIO.new
|
|
$stdout.binmode
|
|
assert_nothing_raised { response.body.call }
|
|
assert_equal file_data, $stdout.string
|
|
ensure
|
|
$stdout = old_stdout
|
|
end
|
|
end
|
|
|
|
def test_data
|
|
response = nil
|
|
assert_nothing_raised { response = process('data') }
|
|
assert_not_nil response
|
|
|
|
assert_kind_of String, response.body
|
|
assert_equal file_data, response.body
|
|
end
|
|
end
|