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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
|
# frozen_string_literal: true
require_relative 'helper'
separate_testing do
require_relative '../lib/rack/rewindable_input'
end
module RewindableTest
extend Minitest::Spec::DSL
def setup
@rio = Rack::RewindableInput.new(@io)
end
it "be able to handle to read()" do
@rio.read.must_equal "hello world"
end
it "be able to handle to read(nil)" do
@rio.read(nil).must_equal "hello world"
end
it "be able to handle to read(length)" do
@rio.read(1).must_equal "h"
end
it "be able to handle to read(length, buffer)" do
buffer = "".dup
result = @rio.read(1, buffer)
result.must_equal "h"
result.object_id.must_equal buffer.object_id
end
it "be able to handle to read(nil, buffer)" do
buffer = "".dup
result = @rio.read(nil, buffer)
result.must_equal "hello world"
result.object_id.must_equal buffer.object_id
end
it "rewind to the beginning when #rewind is called" do
@rio.rewind
@rio.read(1).must_equal 'h'
@rio.rewind
@rio.read.must_equal "hello world"
end
it "be able to handle gets" do
@rio.gets.must_equal "hello world"
@rio.rewind
@rio.gets.must_equal "hello world"
end
it "be able to handle size" do
@rio.size.must_equal "hello world".size
@rio.size.must_equal "hello world".size
@rio.rewind
@rio.gets.must_equal "hello world"
end
it "be able to handle each" do
array = []
@rio.each do |data|
array << data
end
array.must_equal ["hello world"]
@rio.rewind
array = []
@rio.each do |data|
array << data
end
array.must_equal ["hello world"]
end
it "not buffer into a Tempfile if no data has been read yet" do
@rio.instance_variable_get(:@rewindable_io).must_be_nil
end
it "buffer into a Tempfile when data has been consumed for the first time" do
@rio.read(1)
tempfile = @rio.instance_variable_get(:@rewindable_io)
tempfile.wont_be :nil?
@rio.read(1)
tempfile2 = @rio.instance_variable_get(:@rewindable_io)
tempfile2.path.must_equal tempfile.path
end
it "close the underlying tempfile upon calling #close" do
@rio.read(1)
tempfile = @rio.instance_variable_get(:@rewindable_io)
@rio.close
tempfile.must_be :closed?
end
it "handle partial writes to tempfile" do
def @rio.filesystem_has_posix_semantics?
def @rewindable_io.write(buffer)
super(buffer[0..1])
end
super
end
@rio.read(1)
tempfile = @rio.instance_variable_get(:@rewindable_io)
@rio.close
tempfile.must_be :closed?
end
it "close the underlying tempfile upon calling #close when not using posix semantics" do
def @rio.filesystem_has_posix_semantics?; false end
@rio.read(1)
tempfile = @rio.instance_variable_get(:@rewindable_io)
@rio.close
tempfile.must_be :closed?
end
it "be possible to call #close when no data has been buffered yet" do
@rio.close.must_be_nil
end
it "be possible to call #close multiple times" do
@rio.close.must_be_nil
@rio.close.must_be_nil
end
after do
@rio.close
@rio = nil
end
end
describe Rack::RewindableInput do
describe "given an IO object that is already rewindable" do
def setup
@io = StringIO.new("hello world".dup)
super
end
include RewindableTest
end
describe "given an IO object that is not rewindable" do
def setup
@io = StringIO.new("hello world".dup)
@io.instance_eval do
undef :rewind
end
super
end
include RewindableTest
end
describe "given an IO object whose rewind method raises Errno::ESPIPE" do
def setup
@io = StringIO.new("hello world".dup)
def @io.rewind
raise Errno::ESPIPE, "You can't rewind this!"
end
super
end
include RewindableTest
end
end
describe Rack::RewindableInput::Middleware do
it "wraps rack.input in RewindableInput" do
app = proc{|env| [200, {}, [env['rack.input'].class.to_s]]}
app.call('rack.input'=>StringIO.new(''))[2].must_equal ['StringIO']
app = Rack::RewindableInput::Middleware.new(app)
app.call('rack.input'=>StringIO.new(''))[2].must_equal ['Rack::RewindableInput']
end
it "preserves a nil rack.input" do
app = proc{|env| [200, {}, [env['rack.input'].class.to_s]]}
app.call('rack.input'=>nil)[2].must_equal ['NilClass']
app = Rack::RewindableInput::Middleware.new(app)
app.call('rack.input'=>nil)[2].must_equal ['NilClass']
end
end
|
