mirror of
https://github.com/Ed94/Odin.git
synced 2026-06-13 01:21:38 -07:00
Add io tests for bufio streams
This commit is contained in:
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user