iv-org-mirror/shard-kemal
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/shard-kemal.git synced 2024-08-15 00:53:36 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add spec for send_file binary

Browse source
This commit is contained in:
Sdogruyol 2016-08-08 21:49:47 +03:00
parent 31cc9fbff4
commit d4e96c7b44
2 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
spec/helpers_spec.cr
View file

@ -109,5 +109,17 @@ describe "Macros" do
response.headers["Content-Type"].should eq("image/jpeg") response.headers["Content-Type"].should eq("image/jpeg")
response.headers["Content-Length"].should eq("20") response.headers["Content-Length"].should eq("20")
end end
it "sends file with binary stream" do
get "/" do |env|
send_file env, "Serdar".to_slice
end
request = HTTP::Request.new("GET", "/")
response = call_request_on_app(request)
response.status_code.should eq(200)
response.headers["Content-Type"].should eq("application/octet-stream")
response.headers["Content-Length"].should eq("6")
end
end end
end end

4
src/kemal/helpers/helpers.cr
View file

@ -50,7 +50,7 @@ end
# send_file env, "./path/to/file", "image/jpeg" # send_file env, "./path/to/file", "image/jpeg"
def send_file(env, path : String, mime_type : String? = nil) def send_file(env, path : String, mime_type : String? = nil)
file_path = File.expand_path(path, Dir.current) file_path = File.expand_path(path, Dir.current)
mime_type = "application/octet-stream" unless mime_type mime_type ||= "application/octet-stream"
env.response.content_type = mime_type env.response.content_type = mime_type
env.response.content_length = File.size(file_path) env.response.content_length = File.size(file_path)
File.open(file_path) do |file| File.open(file_path) do |file|
@ -65,10 +65,8 @@ end
# Optionally you can override the mime_type # Optionally you can override the mime_type
# #
# send_file env, data_slice, "image/jpeg" # send_file env, data_slice, "image/jpeg"
def send_file(env, data : Slice(UInt8), mime_type : String? = nil) def send_file(env, data : Slice(UInt8), mime_type : String? = nil)
mime_type ||= "application/octet-stream" mime_type ||= "application/octet-stream"
env.response.content_type = mime_type env.response.content_type = mime_type
env.response.content_length = data.bytesize env.response.content_length = data.bytesize
env.response.write data env.response.write data