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
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
|
Code.require_file("test_helper.exs", __DIR__)
defmodule StringIOTest do
use ExUnit.Case, async: true
doctest StringIO
test "open and close" do
{:ok, pid} = StringIO.open("")
assert StringIO.close(pid) == {:ok, {"", ""}}
end
test "contents" do
{:ok, pid} = StringIO.open("abc")
IO.write(pid, "edf")
assert StringIO.contents(pid) == {"abc", "edf"}
end
test "flush" do
{:ok, pid} = StringIO.open("")
IO.write(pid, "edf")
assert StringIO.flush(pid) == "edf"
assert StringIO.contents(pid) == {"", ""}
end
## IO module
test "IO.read :line with \\n" do
{:ok, pid} = StringIO.open("abc\n")
assert IO.read(pid, :line) == "abc\n"
assert IO.read(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.read :line with \\rn" do
{:ok, pid} = StringIO.open("abc\r\n")
assert IO.read(pid, :line) == "abc\n"
assert IO.read(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.read :line without line break" do
{:ok, pid} = StringIO.open("abc")
assert IO.read(pid, :line) == "abc"
assert IO.read(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.read :line with UTF-8" do
{:ok, pid} = StringIO.open("⼊\n")
assert IO.read(pid, :line) == "⼊\n"
assert IO.read(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.read :line with invalid UTF-8" do
{:ok, pid} = StringIO.open(<<130, 227, 129, 132, 227, 129, 134>>)
assert IO.read(pid, :line) == {:error, :collect_line}
assert StringIO.contents(pid) == {<<130, 227, 129, 132, 227, 129, 134>>, ""}
end
test "IO.read count" do
{:ok, pid} = StringIO.open("abc")
assert IO.read(pid, 2) == "ab"
assert IO.read(pid, 8) == "c"
assert IO.read(pid, 1) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.read count with UTF-8" do
{:ok, pid} = StringIO.open("あいう")
assert IO.read(pid, 2) == "あい"
assert IO.read(pid, 8) == "う"
assert IO.read(pid, 1) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.read count with invalid UTF-8" do
{:ok, pid} = StringIO.open(<<130, 227, 129, 132, 227, 129, 134>>)
assert IO.read(pid, 2) == {:error, :invalid_unicode}
assert StringIO.contents(pid) == {<<130, 227, 129, 132, 227, 129, 134>>, ""}
end
test "IO.binread :line with \\n" do
{:ok, pid} = StringIO.open("abc\n")
assert IO.binread(pid, :line) == "abc\n"
assert IO.binread(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.binread :line with \\r\\n" do
{:ok, pid} = StringIO.open("abc\r\n")
assert IO.binread(pid, :line) == "abc\n"
assert IO.binread(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.binread :line without line break" do
{:ok, pid} = StringIO.open("abc")
assert IO.binread(pid, :line) == "abc"
assert IO.binread(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.binread :line with raw bytes" do
{:ok, pid} = StringIO.open(<<181, 255, 194, ?\n>>)
assert IO.binread(pid, :line) == <<181, 255, 194, ?\n>>
assert IO.binread(pid, :line) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.binread count" do
{:ok, pid} = StringIO.open("abc")
assert IO.binread(pid, 2) == "ab"
assert IO.binread(pid, 8) == "c"
assert IO.binread(pid, 1) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.binread count with UTF-8" do
{:ok, pid} = StringIO.open("あいう")
assert IO.binread(pid, 2) == <<227, 129>>
assert IO.binread(pid, 8) == <<130, 227, 129, 132, 227, 129, 134>>
assert IO.binread(pid, 1) == :eof
assert StringIO.contents(pid) == {"", ""}
end
test "IO.write" do
{:ok, pid} = StringIO.open("")
assert IO.write(pid, "foo") == :ok
assert StringIO.contents(pid) == {"", "foo"}
end
test "IO.write with UTF-8" do
{:ok, pid} = StringIO.open("")
assert IO.write(pid, "あいう") == :ok
assert StringIO.contents(pid) == {"", "あいう"}
end
test "IO.write with non-printable arguments" do
{:ok, pid} = StringIO.open("")
assert_raise ArgumentError, fn ->
IO.write(pid, [<<1::1>>])
end
assert_raise ErlangError, ~r/no_translation/, fn ->
IO.write(pid, <<222>>)
end
end
test "IO.binwrite" do
{:ok, pid} = StringIO.open("")
assert IO.binwrite(pid, "foo") == :ok
assert StringIO.contents(pid) == {"", "foo"}
end
test "IO.binwrite with UTF-8" do
{:ok, pid} = StringIO.open("")
assert IO.binwrite(pid, "あいう") == :ok
binary =
<<195, 163, 194, 129, 194, 130, 195, 163>> <>
<<194, 129, 194, 132, 195, 163, 194, 129, 194, 134>>
assert StringIO.contents(pid) == {"", binary}
end
test "IO.binwrite with bytes" do
{:ok, pid} = StringIO.open("")
assert IO.binwrite(pid, <<127, 128>>) == :ok
assert StringIO.contents(pid) == {"", <<127, 194, 128>>}
end
test "IO.binwrite with bytes and latin1 encoding" do
{:ok, pid} = StringIO.open("", encoding: :latin1)
assert IO.binwrite(pid, <<127, 128>>) == :ok
assert StringIO.contents(pid) == {"", <<127, 128>>}
end
test "IO.puts" do
{:ok, pid} = StringIO.open("")
assert IO.puts(pid, "abc") == :ok
assert StringIO.contents(pid) == {"", "abc\n"}
end
test "IO.puts with non-printable arguments" do
{:ok, pid} = StringIO.open("")
assert_raise ArgumentError, fn ->
IO.puts(pid, [<<1::1>>])
end
end
test "IO.inspect" do
{:ok, pid} = StringIO.open("")
assert IO.inspect(pid, {}, []) == {}
assert StringIO.contents(pid) == {"", "{}\n"}
end
test "IO.getn" do
{:ok, pid} = StringIO.open("abc")
assert IO.getn(pid, ">", 2) == "ab"
assert StringIO.contents(pid) == {"c", ""}
end
test "IO.getn with UTF-8" do
{:ok, pid} = StringIO.open("あいう")
assert IO.getn(pid, ">", 2) == "あい"
assert StringIO.contents(pid) == {"う", ""}
end
test "IO.getn with invalid UTF-8" do
{:ok, pid} = StringIO.open(<<130, 227, 129, 132, 227, 129, 134>>)
assert IO.getn(pid, ">", 2) == {:error, :invalid_unicode}
assert StringIO.contents(pid) == {<<130, 227, 129, 132, 227, 129, 134>>, ""}
end
test "IO.getn with capture_prompt" do
{:ok, pid} = StringIO.open("abc", capture_prompt: true)
assert IO.getn(pid, ">", 2) == "ab"
assert StringIO.contents(pid) == {"c", ">"}
end
test "IO.gets with \\n" do
{:ok, pid} = StringIO.open("abc\nd")
assert IO.gets(pid, ">") == "abc\n"
assert StringIO.contents(pid) == {"d", ""}
end
test "IO.gets with \\r\\n" do
{:ok, pid} = StringIO.open("abc\r\nd")
assert IO.gets(pid, ">") == "abc\n"
assert StringIO.contents(pid) == {"d", ""}
end
test "IO.gets without line breaks" do
{:ok, pid} = StringIO.open("abc")
assert IO.gets(pid, ">") == "abc"
assert StringIO.contents(pid) == {"", ""}
end
test "IO.gets with invalid UTF-8" do
{:ok, pid} = StringIO.open(<<130, 227, 129, 132, 227, 129, 134>>)
assert IO.gets(pid, ">") == {:error, :collect_line}
assert StringIO.contents(pid) == {<<130, 227, 129, 132, 227, 129, 134>>, ""}
end
test "IO.gets with capture_prompt" do
{:ok, pid} = StringIO.open("abc\n", capture_prompt: true)
assert IO.gets(pid, ">") == "abc\n"
assert StringIO.contents(pid) == {"", ">"}
end
test ":io.get_password" do
{:ok, pid} = StringIO.open("abc\n")
assert :io.get_password(pid) == "abc\n"
assert StringIO.contents(pid) == {"", ""}
end
test "IO.stream" do
{:ok, pid} = StringIO.open("abc")
assert IO.stream(pid, 2) |> Enum.to_list() == ["ab", "c"]
assert StringIO.contents(pid) == {"", ""}
end
test "IO.stream with invalid UTF-8" do
{:ok, pid} = StringIO.open(<<130, 227, 129, 132, 227, 129, 134>>)
assert_raise IO.StreamError, fn ->
IO.stream(pid, 2) |> Enum.to_list()
end
assert StringIO.contents(pid) == {<<130, 227, 129, 132, 227, 129, 134>>, ""}
end
test "IO.binstream" do
{:ok, pid} = StringIO.open("abc")
assert IO.stream(pid, 2) |> Enum.to_list() == ["ab", "c"]
assert StringIO.contents(pid) == {"", ""}
end
defp get_until(pid, encoding, prompt, module, function) do
:io.request(pid, {:get_until, encoding, prompt, module, function, []})
end
defmodule GetUntilCallbacks do
def until_eof(continuation, :eof) do
{:done, continuation, :eof}
end
def until_eof(continuation, content) do
{:more, continuation ++ content}
end
def until_eof_then_try_more(~c"magic-stop-prefix" ++ continuation, :eof) do
{:done, continuation, :eof}
end
def until_eof_then_try_more(continuation, :eof) do
{:more, ~c"magic-stop-prefix" ++ continuation}
end
def until_eof_then_try_more(continuation, content) do
{:more, continuation ++ content}
end
def up_to_3_bytes(continuation, :eof) do
{:done, continuation, :eof}
end
def up_to_3_bytes(continuation, content) do
case continuation ++ content do
[a, b, c | tail] -> {:done, [a, b, c], tail}
str -> {:more, str}
end
end
def up_to_3_bytes_discard_rest(continuation, :eof) do
{:done, continuation, :eof}
end
def up_to_3_bytes_discard_rest(continuation, content) do
case continuation ++ content do
[a, b, c | _tail] -> {:done, [a, b, c], :eof}
str -> {:more, str}
end
end
end
test "get_until with up_to_3_bytes" do
{:ok, pid} = StringIO.open("abcdefg")
result = get_until(pid, :unicode, "", GetUntilCallbacks, :up_to_3_bytes)
assert result == "abc"
assert IO.read(pid, :all) == "defg"
end
test "get_until with up_to_3_bytes_discard_rest" do
{:ok, pid} = StringIO.open("abcdefg")
result = get_until(pid, :unicode, "", GetUntilCallbacks, :up_to_3_bytes_discard_rest)
assert result == "abc"
assert IO.read(pid, :all) == ""
end
test "get_until with until_eof" do
{:ok, pid} = StringIO.open("abc\nd")
result = get_until(pid, :unicode, "", GetUntilCallbacks, :until_eof)
assert result == "abc\nd"
end
test "get_until with until_eof and \\r\\n" do
{:ok, pid} = StringIO.open("abc\r\nd")
result = get_until(pid, :unicode, "", GetUntilCallbacks, :until_eof)
assert result == "abc\r\nd"
end
test "get_until with until_eof capturing prompt" do
{:ok, pid} = StringIO.open("abc\nd", capture_prompt: true)
result = get_until(pid, :unicode, ">", GetUntilCallbacks, :until_eof)
assert result == "abc\nd"
assert StringIO.contents(pid) == {"", ">>>"}
end
test "get_until with until_eof_then_try_more" do
{:ok, pid} = StringIO.open("abc\nd")
result = get_until(pid, :unicode, "", GetUntilCallbacks, :until_eof_then_try_more)
assert result == "abc\nd"
end
test "get_until with invalid UTF-8" do
{:ok, pid} = StringIO.open(<<130, 227, 129, 132, 227, 129, 134>>)
result = get_until(pid, :unicode, "", GetUntilCallbacks, :until_eof)
assert result == :error
end
test "get_until with raw bytes (latin1)" do
{:ok, pid} = StringIO.open(<<181, 255, 194, ?\n>>)
result = get_until(pid, :latin1, "", GetUntilCallbacks, :until_eof)
assert result == <<181, 255, 194, ?\n>>
end
test ":io.erl_scan_form/2" do
{:ok, pid} = StringIO.open("1.")
result = :io.scan_erl_form(pid, ~c"p>")
assert result == {:ok, [{:integer, 1, 1}, {:dot, 1}], 1}
assert StringIO.contents(pid) == {"", ""}
end
test ":io.erl_scan_form/2 with capture_prompt" do
{:ok, pid} = StringIO.open("1.", capture_prompt: true)
result = :io.scan_erl_form(pid, ~c"p>")
assert result == {:ok, [{:integer, 1, 1}, {:dot, 1}], 1}
assert StringIO.contents(pid) == {"", "p>p>"}
end
end
|
