Files
ww/lib/memio/memio_test.ww

648 lines
19 KiB
Plaintext

// Every @test enumerates parallel `[N]T` arrays of inputs and
// expectations, then iterates one body across them. Parallel arrays
// (rather than `[N]struct{...}`) sidestep the cstage cgen's chained
// `arr[i].field` store gap (task #6). #94 fold-eFinal: the unified
// value-return surface (`let st = memio.fixed(buf); &st.vt` into the
// io dispatchers; io.read/write/close return size/io.error).
package memio_test;
import bytes;
import errors;
import io;
import memio;
import test;
fn putstr(s: str, into: []u8, off: i32) i32 = {
let i: i32 = 0;
for (i < s.len) {
into[off + i] = s[i];
i += 1;
};
return off + s.len;
};
@test fn fixedread() void = {
let arr: [8]u8;
arr[0] = 1u8; arr[1] = 2u8; arr[2] = 3u8; arr[3] = 4u8;
arr[4] = 5u8; arr[5] = 6u8; arr[6] = 7u8; arr[7] = 8u8;
// (request, wantn, wantfirst, wantlast, weof)
// 5 rows: under-remaining, full slot, last-byte-only, post-end, 0-len-at-end.
let req: [5]i32;
let wantn: [5]i32;
let wantf: [5]u8;
let wantl: [5]u8;
let weof: [5]i32;
req[0]=3; wantn[0]=3; wantf[0]=1u8; wantl[0]=3u8; weof[0]=0;
req[1]=4; wantn[1]=4; wantf[1]=4u8; wantl[1]=7u8; weof[1]=0;
req[2]=5; wantn[2]=1; wantf[2]=8u8; wantl[2]=8u8; weof[2]=0;
req[3]=1; wantn[3]=0; wantf[3]=0u8; wantl[3]=0u8; weof[3]=1;
req[4]=0; wantn[4]=0; wantf[4]=0u8; wantl[4]=0u8; weof[4]=1;
let st: memio.stream = memio.fixed(arr[0:8]);
let s: io.stream = &st.vt;
let out: [8]u8;
let i: i32 = 0;
for (i < 5) {
// Clear out so spot checks read real data, not stale bytes.
let k: i32 = 0;
for (k < 8) { out[k] = 0u8; k += 1; };
let r: (size | io.eof | io.error) = io.read(s, out[0:req[i]]);
match (r) {
case let nz: size => {
let n: i32 = nz: i32;
assert(!(weof[i] != 0));
assert(!(n != wantn[i]));
assert(!(n > 0 && out[0] != wantf[i]));
assert(!(n > 0 && out[n - 1] != wantl[i]));
};
case io.eof => { assert(!(weof[i] == 0)); };
case let e: io.error => abort();
};
i += 1;
};
// fixed stream has no closer → io.close returns void.
let c: (void | io.error) = io.close(s);
match (c) { case void => {}; case let e: io.error => abort(); };
};
@test fn fixedreadoverlap() void = {
let arr: [4]u8;
let _: i32 = putstr("ABCD", arr[0:4], 0);
let st: memio.stream = memio.fixed(arr[0:4]);
let r: (size | io.eof | io.error) = io.read(&st.vt, arr[1:4]);
match (r) {
case let n: size => assert(n: i32 == 3);
case io.eof => abort();
case let e: io.error => abort();
};
let want: [4]u8;
let _: i32 = putstr("AABC", want[0:4], 0);
assert(bytes.equal(arr[0:4], want[0:4]));
};
@test fn fixedwritecases() void = {
let dst: [16]u8;
let st: memio.stream = memio.fixed(dst[0:16]);
let s: io.stream = &st.vt;
// One flat source; rows index into it via (off, len). Sequence:
// "hello " (6), "world!!" (7), "XXXXXXX" (7), "Y" (1).
let src: [32]u8;
let _: i32 = putstr("hello world!!XXXXXXXY", src[0:32], 0);
// (srcoff, inlen, wantn, wantnomem) — wantn diverges from inlen on
// the partial row; the overflow row writes into a now-full sink and
// returns nomem (ref/hare/memio/stream.ha:161).
let off: [4]i32;
let ln: [4]i32;
let wantn: [4]i32;
let wantnomem: [4]bool;
off[0]=0; ln[0]=6; wantn[0]=6; wantnomem[0]=false; // full fit
off[1]=6; ln[1]=7; wantn[1]=7; wantnomem[1]=false; // exact-fills cap (pos=13)
off[2]=13; ln[2]=7; wantn[2]=3; wantnomem[2]=false; // partial: 3 free
off[3]=20; ln[3]=1; wantn[3]=0; wantnomem[3]=true; // overflow: full sink → nomem
let i: i32 = 0;
for (i < 4) {
let lo: i32 = off[i];
let hi: i32 = lo + ln[i];
let r: (size | io.error) = io.write(s, src[lo:hi]);
match (r) {
case let n: size => {
assert(!wantnomem[i]);
assert(!(n: i32 != wantn[i]));
};
case let e: io.error => {
assert(wantnomem[i]);
assert(e is nomem);
};
};
i += 1;
};
let want: [16]u8;
let _: i32 = putstr("hello world!!XXX", want[0:16], 0);
assert(!(!bytes.equal(dst[0:16], want[0:16])));
// 0-byte write is always 0 with no side effects.
let r0: (size | io.error) = io.write(s, src[0:0]);
match (r0) {
case let n: size => { assert(!(n != 0: size)); };
case let e: io.error => abort();
};
let c: (void | io.error) = io.close(s);
match (c) { case void => {}; case let e: io.error => abort(); };
};
// A write that exactly fills the sink succeeds; any further non-empty
// write into the full sink returns nomem, not a 0-byte success
// (ref/hare/memio/stream.ha:161). A zero-length sink is full from the
// start, so its first non-empty write is already nomem.
@test fn fixedwritefull() void = {
let buf: [4]u8;
let st: memio.stream = memio.fixed(buf[0:4]);
let s: io.stream = &st.vt;
let src: [4]u8;
let _: i32 = putstr("abcd", src[0:4], 0);
let r4: (size | io.error) = io.write(s, src[0:4]); // exact fill
match (r4) {
case let n: size => { assert(!(n != 4: size)); };
case let e: io.error => abort();
};
let r1: (size | io.error) = io.write(s, src[0:1]); // full sink
match (r1) {
case let n: size => abort();
case let e: io.error => { assert(e is nomem); };
};
let ze: memio.stream = memio.fixed(buf[0:0]);
let zs: io.stream = &ze.vt;
let rz: (size | io.error) = io.write(zs, src[0:1]); // zero-len sink
match (rz) {
case let n: size => abort();
case let e: io.error => { assert(e is nomem); };
};
};
@test fn fixedwriteoverlap() void = {
let arr: [4]u8;
let _: i32 = putstr("ABCD", arr[0:4], 0);
let st: memio.stream = memio.fixed(arr[0:4]);
let sk: (io.off | io.error) = io.seek(&st.vt, 1, io.whence.SET);
match (sk) { case let o: io.off => {}; case let e: io.error => abort(); };
let r: (size | io.error) = io.write(&st.vt, arr[0:3]);
match (r) {
case let n: size => assert(n: i32 == 3);
case let e: io.error => abort();
};
let want: [4]u8;
let _: i32 = putstr("AABC", want[0:4], 0);
assert(bytes.equal(arr[0:4], want[0:4]));
};
// Drive grow 0 → 8 → 16 → 32 by writing sized chunks. Verify
// accumulated `pos` after each step.
@test fn dynamicgrowcases() void = {
let st: memio.stream = memio.dynamic();
let s: io.stream = &st.vt;
let src: [32]u8;
let _: i32 = putstr("abcdefghijklmnopqrstuvwxyz012345", src[0:32], 0);
// (chunkn, totaln) — cap transitions: 0→8 at row 0; 8→16 at row 2;
// 16→32 at row 4.
let chunk: [5]i32;
let total: [5]i32;
chunk[0]=3; total[0]=3;
chunk[1]=5; total[1]=8;
chunk[2]=1; total[2]=9;
chunk[3]=7; total[3]=16;
chunk[4]=10; total[4]=26;
let off: i32 = 0;
let i: i32 = 0;
for (i < 5) {
let lo: i32 = off;
let hi: i32 = lo + chunk[i];
let r: (size | io.error) = io.write(s, src[lo:hi]);
match (r) {
case let n: size => { assert(!(n: i32 != chunk[i])); };
case let e: io.error => abort();
};
assert(!(memio.buffer(&st).len != total[i]));
off += chunk[i];
i += 1;
};
assert(!(!bytes.equal(memio.buffer(&st), src[0:26])));
let c: (void | io.error) = io.close(s);
match (c) { case void => {}; case let e: io.error => abort(); };
};
@test fn dynamicgrowaliasedwrite() void = {
let seed: []u8;
seed.ptr = nil; seed.len = 0; seed.cap = 0;
append(seed, 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H');
let st: memio.stream = memio.dynamicfrom(seed);
let sk: (io.off | io.error) = io.seek(&st.vt, 0, io.whence.END);
match (sk) { case let o: io.off => {}; case let e: io.error => abort(); };
let alias: []u8 = memio.buffer(&st);
let r: (size | io.error) = io.write(&st.vt, alias);
match (r) {
case let n: size => assert(n: i32 == 8);
case let e: io.error => abort();
};
let want: [16]u8;
let _: i32 = putstr("ABCDEFGHABCDEFGH", want[0:16], 0);
assert(alias.ptr != memio.buffer(&st).ptr);
assert(bytes.equal(memio.buffer(&st), want[0:16]));
let c: (void | io.error) = io.close(&st.vt);
match (c) { case void => {}; case let e: io.error => abort(); };
};
@test fn dynamicwriteoverflow() void = {
let st: memio.stream = memio.dynamic();
let one: [1]u8;
one[0] = 1u8;
let first: (size | io.error) = io.write(&st.vt, one[0:1]);
match (first) { case let n: size => {}; case let e: io.error => abort(); };
let huge: []u8;
huge.ptr = nil;
huge.len = 2147483647;
huge.cap = huge.len;
let r: (size | io.error) = io.write(&st.vt, huge);
match (r) {
case let n: size => abort();
case let e: io.error => assert(e is nomem);
};
assert(memio.buffer(&st).len == 1);
let c: (void | io.error) = io.close(&st.vt);
match (c) { case void => {}; case let e: io.error => abort(); };
};
// op=0 writes `ln` bytes from a rolling source; op=1 resets and ignores
// ln. After each row the accumulated len must equal `want`.
@test fn dynamicreset() void = {
let st: memio.stream = memio.dynamic();
let s: io.stream = &st.vt;
let src: [16]u8;
src[0]=10u8; src[1]=20u8; src[2]=30u8; src[3]=40u8;
src[4]=50u8; src[5]=60u8; src[6]=70u8; src[7]=80u8;
let op: [6]i32;
let ln: [6]i32;
let want: [6]i32;
op[0]=0; ln[0]=5; want[0]=5; // initial write
op[1]=1; ln[1]=0; want[1]=0; // reset (after writes)
op[2]=0; ln[2]=3; want[2]=3; // refill from 0
op[3]=0; ln[3]=2; want[3]=5; // continue
op[4]=1; ln[4]=0; want[4]=0; // reset (after partial)
op[5]=0; ln[5]=8; want[5]=8; // larger refill
let off: i32 = 0;
let i: i32 = 0;
for (i < 6) {
if (op[i] == 1) {
memio.reset(&st);
off = 0;
} else {
let lo: i32 = off;
let hi: i32 = lo + ln[i];
let r: (size | io.error) = io.write(s, src[lo:hi]);
match (r) {
case let n: size => { assert(!(n: i32 != ln[i])); };
case let e: io.error => abort();
};
off += ln[i];
};
assert(!(memio.buffer(&st).len != want[i]));
i += 1;
};
assert(!(!bytes.equal(memio.buffer(&st), src[0:8])));
let c: (void | io.error) = io.close(s);
match (c) { case void => {}; case let e: io.error => abort(); };
};
@test fn borrowedreadcases() void = {
let arr: [6]u8;
arr[0]=0u8; arr[1]=1u8; arr[2]=2u8; arr[3]=3u8; arr[4]=4u8; arr[5]=5u8;
let st: memio.stream = memio.fixed(arr[0:6]);
// (amt, wantfirst, wantlast, weof)
let amt: [4]i32;
let wf: [4]u8;
let wl: [4]u8;
let weof: [4]i32;
amt[0]=4; wf[0]=0u8; wl[0]=3u8; weof[0]=0; // under remaining
amt[1]=2; wf[1]=4u8; wl[1]=5u8; weof[1]=0; // exact remaining
amt[2]=1; wf[2]=0u8; wl[2]=0u8; weof[2]=1; // past end → eof
amt[3]=0; wf[3]=0u8; wl[3]=0u8; weof[3]=0; // 0-byte view always ok
let i: i32 = 0;
for (i < 4) {
let r: ([]u8 | io.eof) = memio.borrowedread(&st, amt[i]);
match (r) {
case let v: []u8 => {
assert(!(weof[i] != 0));
assert(!(v.len != amt[i]));
assert(v.cap == v.len);
assert(!(v.len > 0 && v[0] != wf[i]));
assert(!(v.len > 0 && v[v.len - 1] != wl[i]));
};
case io.eof => { assert(!(weof[i] == 0)); };
};
i += 1;
};
};
@test fn borrowedreadnegativeaborts() void = {
test.expectabort();
let arr: [1]u8;
let st: memio.stream = memio.fixed(arr[0:1]);
let r: ([]u8 | io.eof) = memio.borrowedread(&st, -1);
};
@test fn stringview() void = {
let st: memio.stream = memio.dynamic();
let s: io.stream = &st.vt;
assert(!(memio.string(&st).len != 0));
let src: [32]u8;
let _: i32 = putstr("hello world!!", src[0:32], 0);
// (srcoff, inlen, wantacc, wantfirst, wantlast)
let off: [3]i32;
let ln: [3]i32;
let want: [3]i32;
let wf: [3]u8;
let wl: [3]u8;
off[0]=0; ln[0]=5; want[0]=5; wf[0]=104u8; wl[0]=111u8; // "hello" h..o
off[1]=5; ln[1]=6; want[1]=11; wf[1]=104u8; wl[1]=100u8; // +" world" h..d
off[2]=11; ln[2]=2; want[2]=13; wf[2]=104u8; wl[2]=33u8; // +"!!" h..!
let i: i32 = 0;
for (i < 3) {
let lo: i32 = off[i];
let hi: i32 = lo + ln[i];
let r: (size | io.error) = io.write(s, src[lo:hi]);
match (r) { case let n: size => {}; case let e: io.error => abort(); };
let v: str = memio.string(&st);
assert(!(v.len != want[i]));
assert(v.cap == v.len);
assert(memio.buffer(&st).cap == v.len);
assert(!(v[0] != wf[i]));
assert(!(v[v.len - 1] != wl[i]));
i += 1;
};
let c: (void | io.error) = io.close(s);
match (c) { case void => {}; case let e: io.error => abort(); };
};
@test fn dynamicfromseed() void = {
// Build a heap-allocated seed via append (rt_ensure path), then
// hand ownership to memio. dynamicclose then frees `seed.cap`
// bytes — the regression that motivated the `cap = buf.cap` fix.
let seed: []u8;
seed.ptr = nil; seed.len = 0; seed.cap = 0;
append(seed, 100u8, 101u8, 102u8, 103u8);
let st: memio.stream = memio.dynamicfrom(seed);
let s: io.stream = &st.vt;
// (readn, wantbyte, weof)
let rn: [5]i32;
let want: [5]u8;
let weof: [5]i32;
rn[0]=1; want[0]=100u8; weof[0]=0;
rn[1]=1; want[1]=101u8; weof[1]=0;
rn[2]=1; want[2]=102u8; weof[2]=0;
rn[3]=1; want[3]=103u8; weof[3]=0;
rn[4]=1; want[4]=0u8; weof[4]=1; // past seed end → eof
let out: [4]u8;
let i: i32 = 0;
for (i < 5) {
let r: (size | io.eof | io.error) = io.read(s, out[0:rn[i]]);
match (r) {
case let nz: size => {
let n: i32 = nz: i32;
assert(!(weof[i] != 0));
assert(!(n != rn[i]));
assert(!(out[0] != want[i]));
};
case io.eof => { assert(!(weof[i] == 0)); };
case let e: io.error => abort();
};
i += 1;
};
// Now write extends past the seed; grow path runs and close
// must free the (post-grow) cap, not the seed cap.
let extra: [4]u8;
extra[0]=200u8; extra[1]=201u8; extra[2]=202u8; extra[3]=203u8;
let w: (size | io.error) = io.write(s, extra[0:4]);
match (w) {
case let n: size => { assert(!(n: i32 != 4)); };
case let e: io.error => abort();
};
let c: (void | io.error) = io.close(s);
match (c) { case void => {}; case let e: io.error => abort(); };
};
// Sequential rows over one 8-byte fixed stream; CUR rows depend on the
// cursor left by earlier rows. want = resulting pos for valid rows,
// -1 = errors.invalid expected. After EVERY row an io.tell readback
// checks the cursor against a shadow `pos` — invalid rows must leave
// it untouched (ref/hare/memio/stream.ha:122-140 returns before the
// pos store).
@test fn seekcases() void = {
let arr: [8]u8;
arr[0]=10u8; arr[1]=11u8; arr[2]=12u8; arr[3]=13u8;
arr[4]=14u8; arr[5]=15u8; arr[6]=16u8; arr[7]=17u8;
// (whence, off, want)
let wh: [12]i32; // 0 SET / 1 CUR / 2 END
let off: [12]i64;
let want: [12]i64;
wh[0] =0; off[0] =3; want[0] =3; // SET forward
wh[1] =1; off[1] =2; want[1] =5; // CUR forward
wh[2] =1; off[2] =-5; want[2] =0; // CUR back to start
wh[3] =2; off[3] =0; want[3] =8; // END exact end
wh[4] =2; off[4] =-8; want[4] =0; // END back to start
wh[5] =0; off[5] =9; want[5] =-1; // SET past end
wh[6] =1; off[6] =-1; want[6] =-1; // CUR before start
wh[7] =2; off[7] =1; want[7] =-1; // END past end
wh[8] =2; off[8] =-9; want[8] =-1; // END before start
wh[9] =0; off[9] =8; want[9] =8; // SET end exact
wh[10]=1; off[10]=0; want[10]=8; // CUR no-op at end
wh[11]=0; off[11]=0; want[11]=0; // SET rewind
let st: memio.stream = memio.fixed(arr[0:8]);
let s: io.stream = &st.vt;
let pos: i64 = 0;
let i: i32 = 0;
for (i < 12) {
let w: io.whence = wh[i]: io.whence;
let r: (io.off | io.error) = io.seek(s, off[i], w);
match (r) {
case let o: io.off => {
assert(!(want[i] < 0));
assert(!((o: i64) != want[i]));
pos = o: i64;
};
case let e: io.error => {
assert(!(want[i] >= 0));
match (e) {
case errors.invalid => {};
case => abort();
};
};
};
let t: (io.off | io.error) = io.tell(s);
match (t) {
case let o: io.off => { assert(!((o: i64) != pos)); };
case let e: io.error => abort();
};
i += 1;
};
// Data readback through the repositioned cursor: SET 5 then read
// drains bytes 5..7, then eof.
let r: (io.off | io.error) = io.seek(s, 5, io.whence.SET);
match (r) {
case let o: io.off => { assert(!((o: i64) != 5)); };
case let e: io.error => abort();
};
let out: [8]u8;
let rd: (size | io.eof | io.error) = io.read(s, out[0:8]);
match (rd) {
case let nz: size => {
assert(!(nz: i32 != 3));
assert(!(out[0] != 15u8));
assert(!(out[2] != 17u8));
};
case io.eof => abort();
case let e: io.error => abort();
};
let rd2: (size | io.eof | io.error) = io.read(s, out[0:8]);
match (rd2) {
case io.eof => {};
case let nz: size => abort();
case let e: io.error => abort();
};
};
@test fn seekminleavescursor() void = {
let arr: [8]u8;
let st: memio.stream = memio.fixed(arr[0:8]);
let set: (io.off | io.error) = io.seek(&st.vt, 3, io.whence.SET);
match (set) { case let o: io.off => {}; case let e: io.error => abort(); };
let min: i64 = -9223372036854775807i64 - 1i64;
let r: (io.off | io.error) = io.seek(&st.vt, min, io.whence.CUR);
match (r) {
case let o: io.off => abort();
case let e: io.error => assert(e is errors.invalid);
};
let pos: (io.off | io.error) = io.tell(&st.vt);
match (pos) {
case let o: io.off => assert((o: i64) == 3);
case let e: io.error => abort();
};
};
@test fn emptyseek() void = {
let arr: [1]u8;
arr[0] = 0u8;
let st: memio.stream = memio.fixed(arr[0:0]);
let s: io.stream = &st.vt;
let wh: [5]i32;
let off: [5]i64;
let want: [5]i64;
wh[0]=0; off[0]=0; want[0]=0; // SET 0 — only valid target
wh[1]=2; off[1]=0; want[1]=0; // END 0 == start
wh[2]=0; off[2]=1; want[2]=-1;
wh[3]=1; off[3]=-1; want[3]=-1;
wh[4]=2; off[4]=-1; want[4]=-1;
let i: i32 = 0;
for (i < 5) {
let w: io.whence = wh[i]: io.whence;
let r: (io.off | io.error) = io.seek(s, off[i], w);
match (r) {
case let o: io.off => {
assert(!(want[i] < 0));
assert(!((o: i64) != want[i]));
};
case let e: io.error => {
assert(!(want[i] >= 0));
match (e) {
case errors.invalid => {};
case => abort();
};
};
};
i += 1;
};
};
// The dynamic vtable wires the same seekfn; bounds run against the
// LOGICAL length (m.len), not capacity (Hare: len(s.buf),
// ref/hare/memio/stream.ha:131,136).
@test fn dynamicseek() void = {
let st: memio.stream = memio.dynamic();
let s: io.stream = &st.vt;
let src: [5]u8;
src[0]=30u8; src[1]=31u8; src[2]=32u8; src[3]=33u8; src[4]=34u8;
let wr: (size | io.error) = io.write(s, src[0:5]);
match (wr) {
case let n: size => { assert(!(n: i32 != 5)); };
case let e: io.error => abort();
};
// END 0 → 5 (logical length); cap is 8 post-grow, so END 1 must
// still be invalid even though capacity would fit it.
let r: (io.off | io.error) = io.seek(s, 0, io.whence.END);
match (r) {
case let o: io.off => { assert(!((o: i64) != 5)); };
case let e: io.error => abort();
};
let r1: (io.off | io.error) = io.seek(s, 1, io.whence.END);
match (r1) {
case let o: io.off => abort();
case let e: io.error => {
match (e) {
case errors.invalid => {};
case => abort();
};
};
};
// CUR -2 → 3; read drains 33,34 then eof.
let r2: (io.off | io.error) = io.seek(s, -2, io.whence.CUR);
match (r2) {
case let o: io.off => { assert(!((o: i64) != 3)); };
case let e: io.error => abort();
};
let out: [4]u8;
let rd: (size | io.eof | io.error) = io.read(s, out[0:4]);
match (rd) {
case let nz: size => {
assert(!(nz: i32 != 2));
assert(!(out[0] != 33u8));
assert(!(out[1] != 34u8));
};
case io.eof => abort();
case let e: io.error => abort();
};
let rd2: (size | io.eof | io.error) = io.read(s, out[0:4]);
match (rd2) {
case io.eof => {};
case let nz: size => abort();
case let e: io.error => abort();
};
let c: (void | io.error) = io.close(s);
match (c) { case void => {}; case let e: io.error => abort(); };
};