1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
#--
# Copyright (c) 2007-2013 Nick Sieger.
# See the file README.txt included with the distribution for
# software license details.
#++
require 'net/http/post/multipart'
require 'test/unit'
class Net::HTTP::Post::MultiPartTest < Test::Unit::TestCase
TEMP_FILE = "temp.txt"
HTTPPost = Struct.new("HTTPPost", :content_length, :body_stream, :content_type)
HTTPPost.module_eval do
def set_content_type(type, params = {})
self.content_type = type + params.map{|k,v|"; #{k}=#{v}"}.join('')
end
end
def teardown
File.delete(TEMP_FILE) rescue nil
end
def test_form_multipart_body
File.open(TEMP_FILE, "w") {|f| f << "1234567890"}
@io = File.open(TEMP_FILE)
@io = UploadIO.new @io, "text/plain", TEMP_FILE
assert_results Net::HTTP::Post::Multipart.new("/foo/bar", :foo => 'bar', :file => @io)
end
def test_form_multipart_body_put
File.open(TEMP_FILE, "w") {|f| f << "1234567890"}
@io = File.open(TEMP_FILE)
@io = UploadIO.new @io, "text/plain", TEMP_FILE
assert_results Net::HTTP::Put::Multipart.new("/foo/bar", :foo => 'bar', :file => @io)
end
def test_form_multipart_body_with_stringio
@io = StringIO.new("1234567890")
@io = UploadIO.new @io, "text/plain", TEMP_FILE
assert_results Net::HTTP::Post::Multipart.new("/foo/bar", :foo => 'bar', :file => @io)
end
def test_form_multiparty_body_with_parts_headers
@io = StringIO.new("1234567890")
@io = UploadIO.new @io, "text/plain", TEMP_FILE
parts = { :text => 'bar', :file => @io }
headers = {
:parts => {
:text => { "Content-Type" => "part/type" },
:file => { "Content-Transfer-Encoding" => "part-encoding" }
}
}
request = Net::HTTP::Post::Multipart.new("/foo/bar", parts, headers)
assert_results request
assert_additional_headers_added(request, headers[:parts])
end
def test_form_multipart_body_with_array_value
File.open(TEMP_FILE, "w") {|f| f << "1234567890"}
@io = File.open(TEMP_FILE)
@io = UploadIO.new @io, "text/plain", TEMP_FILE
params = {:foo => ['bar', 'quux'], :file => @io}
headers = { :parts => {
:foo => { "Content-Type" => "application/json; charset=UTF-8" } } }
post = Net::HTTP::Post::Multipart.new("/foo/bar", params, headers,
Net::HTTP::Post::Multipart::DEFAULT_BOUNDARY)
assert post.content_length && post.content_length > 0
assert post.body_stream
body = post.body_stream.read
assert_equal 2, body.lines.grep(/name="foo"/).length
assert body =~ /Content-Type: application\/json; charset=UTF-8/, body
end
def test_form_multipart_body_with_arrayparam
File.open(TEMP_FILE, "w") {|f| f << "1234567890"}
@io = File.open(TEMP_FILE)
@io = UploadIO.new @io, "text/plain", TEMP_FILE
assert_results Net::HTTP::Post::Multipart.new("/foo/bar", :multivalueParam => ['bar','bah'], :file => @io)
end
def assert_results(post)
assert post.content_length && post.content_length > 0
assert post.body_stream
assert_equal "multipart/form-data; boundary=#{Multipartable::DEFAULT_BOUNDARY}", post['content-type']
body = post.body_stream.read
boundary_regex = Regexp.quote Multipartable::DEFAULT_BOUNDARY
assert body =~ /1234567890/
# ensure there is at least one boundary
assert body =~ /^--#{boundary_regex}\r\n/
# ensure there is an epilogue
assert body =~ /^--#{boundary_regex}--\r\n/
assert body =~ /text\/plain/
if (body =~ /multivalueParam/)
assert_equal 2, body.scan(/^.*multivalueParam.*$/).size
end
end
def assert_additional_headers_added(post, parts_headers)
post.body_stream.rewind
body = post.body_stream.read
parts_headers.each do |part, headers|
headers.each do |k,v|
assert body =~ /#{k}: #{v}/
end
end
end
end
|
