Skip to content

Commit ba69477

Browse files
author
Grahame Bowland
committed
add tests for io::readchars and io::readchar
Additionally reformat so that 'make check' passes.
1 parent bcc2563 commit ba69477

File tree

2 files changed

+58
-7
lines changed

2 files changed

+58
-7
lines changed

src/libstd/io.rs

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -130,26 +130,29 @@ obj new_reader(rdr: buf_reader) {
130130
val += next & 63 as uint;
131131
}
132132
// See str::char_at
133-
val += (b0 << (w + 1u as u8) as uint) << (w - 1u) * 6u - w - 1u;
133+
val += (b0 << (w + 1u as u8) as uint)
134+
<< (w - 1u) * 6u - w - 1u;
134135
chars += [ val as char ];
135136
}
136137
ret (i, 0u);
137138
}
138139
let buf: [u8] = [];
139140
let chars: [char] = [];
140-
let nbread = n; // might need more bytes, but reading n will never over-read
141+
// might need more bytes, but reading n will never over-read
142+
let nbread = n;
141143
while nbread > 0u {
142-
let data = self.read_bytes(nbread);
144+
let data = self.read_bytes(nbread);
143145
if vec::len(data) == 0u {
144-
// eof - FIXME should we do something if we're split in a unicode char?
146+
// eof - FIXME should we do something if
147+
// we're split in a unicode char?
145148
break;
146149
}
147150
buf += data;
148151
let (offset, nbreq) = chars_from_buf(buf, chars);
149152
let ncreq = n - vec::len(chars);
150-
// again we either know we need a certain number of bytes to complete a
151-
// character, or we make sure we don't over-read by reading 1-byte per char
152-
// needed
153+
// again we either know we need a certain number of bytes
154+
// to complete a character, or we make sure we don't
155+
// over-read by reading 1-byte per char needed
153156
nbread = if ncreq > nbreq { ncreq } else { nbreq };
154157
if nbread > 0u {
155158
buf = vec::slice(buf, offset, vec::len(buf));

src/test/stdtest/io.rs

Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,54 @@ fn test_simple() {
2323
assert (str::eq(frood, frood2));
2424
}
2525

26+
#[test]
27+
fn test_readchars_empty() {
28+
let inp : io::reader = io::string_reader("");
29+
let res : [char] = inp.read_chars(128u);
30+
assert(vec::len(res) == 0u);
31+
}
32+
33+
#[test]
34+
fn test_readchars_wide() {
35+
let wide_test = "生锈的汤匙切肉汤hello生锈的汤匙切肉汤";
36+
let ivals : [int] = [
37+
29983, 38152, 30340, 27748,
38+
21273, 20999, 32905, 27748,
39+
104, 101, 108, 108, 111,
40+
29983, 38152, 30340, 27748,
41+
21273, 20999, 32905, 27748];
42+
fn check_read_ln(len : uint, s: str, ivals: [int]) {
43+
let inp : io::reader = io::string_reader(s);
44+
let res : [char] = inp.read_chars(len);
45+
if (len <= vec::len(ivals)) {
46+
assert(vec::len(res) == len);
47+
}
48+
assert(vec::slice(ivals, 0u, vec::len(res)) ==
49+
vec::map(res, {|x| x as int}));
50+
}
51+
let i = 0u;
52+
while i < 8u {
53+
check_read_ln(i, wide_test, ivals);
54+
i += 1u;
55+
}
56+
// check a long read for good measure
57+
check_read_ln(128u, wide_test, ivals);
58+
}
59+
60+
#[test]
61+
fn test_readchar() {
62+
let inp : io::reader = io::string_reader("生");
63+
let res : char = inp.read_char();
64+
assert(res as int == 29983);
65+
}
66+
67+
#[test]
68+
fn test_readchar_empty() {
69+
let inp : io::reader = io::string_reader("");
70+
let res : char = inp.read_char();
71+
assert(res as int == -1);
72+
}
73+
2674
#[test]
2775
fn file_reader_not_exist() {
2876
alt io::file_reader("not a file") {

0 commit comments

Comments
 (0)