Files
c3c/test/unit7/stdlib/io/bufferstream.c3
2025-02-27 11:10:41 +01:00

95 lines
2.1 KiB
Plaintext

module std::io @test;
const DATA = "Lorem ipsum blandit.";
fn void readbuffer_large()
{
ByteReader src;
src.init(DATA);
char[DATA.len] buf;
ReadBuffer reader_buf;
reader_buf.init(&src, buf[..]);
char[DATA.len] bytes;
usz n = reader_buf.read(bytes[..])!!;
assert(n == DATA.len, "large read failed: got %d; want %d", n, DATA.len);
String got = (String)bytes[..];
assert(got == DATA, "large read failed: got %s; want %s", got, DATA);
}
fn void readbuffer()
{
ByteReader src;
src.init(DATA);
char[3] buf;
ReadBuffer reader_buf;
reader_buf.init(&src, buf[..]);
ByteWriter bw;
bw.tinit();
usz n = io::copy_to(&reader_buf, &bw)!!;
assert(n == DATA.len, "got %d; want %d", n, DATA.len);
String got = bw.str_view();
assert(got == DATA, "got %s; want %s", got, DATA);
}
fn void writebuffer_large()
{
ByteWriter out;
out.tinit();
char[16] buf;
WriteBuffer write_buf;
write_buf.init(&out, buf[..]);
usz n = write_buf.write(DATA)!!;
assert(n == DATA.len, "large write failed: got %d; want %d", n, DATA.len);
String got = out.str_view();
assert(got == DATA, "large write failed: got %s; want %s", got, DATA);
}
fn void writebuffer()
{
ByteReader br;
br.init(DATA);
ByteWriter out;
out.tinit();
char[3] buf;
WriteBuffer write_buf;
write_buf.init(&out, buf[..]);
usz n = io::copy_to(&br, &write_buf)!!;
assert(n == DATA.len, "got %d; want %d", n, DATA.len);
String got = out.str_view();
assert(got == DATA, "got %s; want %s", got, DATA);
}
fn void writebuffer_write_byte()
{
ByteWriter out;
out.tinit();
char[2] buf;
WriteBuffer write_buf;
write_buf.init(&out, buf[..]);
write_buf.write_byte('a')!!;
assert(write_buf.str_view() == "a");
assert(out.str_view() == "");
write_buf.write_byte('b')!!;
assert(write_buf.str_view() == "ab");
assert(out.str_view() == "");
write_buf.write_byte('c')!!;
assert(write_buf.str_view() == "c");
assert(out.str_view() == "ab");
write_buf.flush()!!;
assert(write_buf.str_view() == "");
assert(out.str_view() == "abc");
}