Add io tests for bufio streams

This commit is contained in:
Feoramund
2024-08-27 19:17:20 -04:00
committed by Laytan
parent 24a53c246f
commit 7fa2e56cd9
+120
View File
@@ -1,5 +1,6 @@
package test_core_io
import "core:bufio"
import "core:bytes"
import "core:io"
import "core:log"
@@ -612,3 +613,122 @@ test_os2_file_stream :: proc(t: ^testing.T) {
log.debugf("%#v", results)
}
@test
test_bufio_buffered_writer :: proc(t: ^testing.T) {
// Using a strings.Builder as the backing stream.
sb := strings.builder_make()
defer strings.builder_destroy(&sb)
buf: [32]u8
expected_buf: [64]u8
for i in 0..<u8(len(buf)) {
buf[i] = 'A' + i
expected_buf[i] = 'A' + i
strings.write_byte(&sb, 'A' + i)
}
for i in 32..<u8(len(expected_buf)) {
expected_buf[i] = ('A' + i-len(buf)) ~ 0xAA
}
writer: bufio.Writer
bufio.writer_init(&writer, strings.to_stream(&sb))
defer bufio.writer_destroy(&writer)
results, _ := _test_stream(t, bufio.writer_to_stream(&writer), buf[:],
do_destroy = false)
testing.expectf(t, bytes.compare(sb.buf[:], expected_buf[:]) == 0, "bufio buffered string builder stream failed:\nbuilder<%q>\n!=\nbuffer <%q>", sb.buf[:], expected_buf[:])
log.debugf("%#v", results)
}
@test
test_bufio_buffered_reader :: proc(t: ^testing.T) {
// Using a bytes.Reader as the backing stream.
buf: [32]u8
for i in 0..<u8(len(buf)) {
buf[i] = 'A' + i
}
results: Passed_Tests
ok: bool
for end in 0..<i64(len(buf)) {
br: bytes.Reader
bs := bytes.reader_init(&br, buf[:end])
reader: bufio.Reader
bufio.reader_init(&reader, bs)
defer bufio.reader_destroy(&reader)
results, ok = _test_stream(t, bufio.reader_to_stream(&reader), buf[:end])
if !ok {
log.debugf("buffer[:%i] := %v", end, buf[:end])
return
}
}
log.debugf("%#v", results)
}
@test
test_bufio_buffered_read_writer :: proc(t: ^testing.T) {
// Using an os2.File as the backing stream for both reader & writer.
defer if !testing.failed(t) {
testing.expect_value(t, os2.remove(TEMPORARY_FILENAME), nil)
}
buf: [32]u8
for i in 0..<u8(len(buf)) {
buf[i] = 'A' + i
}
TEMPORARY_FILENAME :: "test_core_io_bufio_read_writer_os2_file_stream"
fd, open_err := os2.open(TEMPORARY_FILENAME, {.Read, .Write, .Create, .Trunc})
if !testing.expectf(t, open_err == nil, "error on opening %q: %v", TEMPORARY_FILENAME, open_err) {
return
}
stream := os2.to_stream(fd)
bytes_written, write_err := io.write(stream, buf[:])
if !testing.expectf(t, bytes_written == len(buf) && write_err == nil,
"failed to Write initial buffer: bytes_written<%v> != len_buf<%v>, %v", bytes_written, len(buf), write_err) {
return
}
flush_err := io.flush(stream)
if !testing.expectf(t, flush_err == nil,
"failed to Flush initial buffer: %v", write_err) {
return
}
// bufio.Read_Writer isn't capable of seeking, so we have to reset the os2
// stream back to the start here.
pos, seek_err := io.seek(stream, 0, .Start)
if !testing.expectf(t, pos == 0 && seek_err == nil,
"Pre-test Seek reset failed: pos<%v>, %v", pos, seek_err) {
return
}
reader: bufio.Reader
writer: bufio.Writer
read_writer: bufio.Read_Writer
bufio.reader_init(&reader, stream)
defer bufio.reader_destroy(&reader)
bufio.writer_init(&writer, stream)
defer bufio.writer_destroy(&writer)
bufio.read_writer_init(&read_writer, &reader, &writer)
// os2 file stream proc close and destroy are the same.
results, _ := _test_stream(t, bufio.read_writer_to_stream(&read_writer), buf[:], do_destroy = false)
log.debugf("%#v", results)
}