tests/ports/unix: Add coverage test for readinto1 stream method.

And expand the test for `readinto()` to test the difference between trying
to read the requested amount by doing multiple underlying IO calls, and
only doing one call.

Signed-off-by: Damien George <damien@micropython.org>
This commit is contained in:
Damien George 2025-09-19 13:09:25 +10:00
parent 2373340aa3
commit b44c4de4fd
3 changed files with 30 additions and 0 deletions

View File

@ -97,6 +97,7 @@ static const mp_rom_map_elem_t rawfile_locals_dict_table[] = {
{ MP_ROM_QSTR(MP_QSTR_write), MP_ROM_PTR(&mp_stream_write_obj) },
{ MP_ROM_QSTR(MP_QSTR_write1), MP_ROM_PTR(&mp_stream_write1_obj) },
{ MP_ROM_QSTR(MP_QSTR_readinto), MP_ROM_PTR(&mp_stream_readinto_obj) },
{ MP_ROM_QSTR(MP_QSTR_readinto1), MP_ROM_PTR(&mp_stream_readinto1_obj) },
{ MP_ROM_QSTR(MP_QSTR_readline), MP_ROM_PTR(&mp_stream_unbuffered_readline_obj) },
{ MP_ROM_QSTR(MP_QSTR_ioctl), MP_ROM_PTR(&mp_stream_ioctl_obj) },
};

View File

@ -33,6 +33,7 @@ print(stream.read()) # read all encounters non-blocking error
print(stream.read(1)) # read 1 byte encounters non-blocking error
print(stream.readline()) # readline encounters non-blocking error
print(stream.readinto(bytearray(10))) # readinto encounters non-blocking error
print(stream.readinto1(bytearray(10))) # readinto1 encounters non-blocking error
print(stream.write(b"1")) # write encounters non-blocking error
print(stream.write1(b"1")) # write1 encounters non-blocking error
stream.set_buf(b"123")
@ -48,6 +49,28 @@ except OSError:
stream.set_error(0)
print(stream.ioctl(0, bytearray(10))) # successful ioctl call
print("# stream.readinto")
# stream.readinto will read 3 bytes then try to read more to fill the buffer,
# but on the second attempt will encounter EIO and should raise that error.
stream.set_error(errno.EIO)
stream.set_buf(b"123")
buf = bytearray(4)
try:
stream.readinto(buf)
except OSError as er:
print("OSError", er.errno == errno.EIO)
print(buf)
# stream.readinto1 will read 3 bytes then should return them immediately, and
# not encounter the EIO.
stream.set_error(errno.EIO)
stream.set_buf(b"123")
buf = bytearray(4)
print(stream.readinto1(buf), buf)
print("# stream textio")
stream2 = data[3] # is textio
print(stream2.read(1)) # read 1 byte encounters non-blocking error with textio stream

View File

@ -214,11 +214,17 @@ None
None
None
None
None
b'123'
b'123'
b'123'
OSError
0
# stream.readinto
OSError True
bytearray(b'123\x00')
3 bytearray(b'123\x00')
# stream textio
None
None
cpp None