Skip to content

Commit 2ece545

Browse files
committed
Fix test FakeStderr to use read1() for progress parsing
1 parent f227f74 commit 2ece545

1 file changed

Lines changed: 13 additions & 1 deletion

File tree

tests/test_flash_usb_and_find_usb_fixes.py

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -372,12 +372,24 @@ class TestFlashUsbProgress:
372372

373373
def _fake_popen_factory(self, stderr_lines, popen_calls, envs):
374374
class FakePopen:
375+
class FakeStderr:
376+
def __init__(self, lines):
377+
self._lines = [line.encode("utf-8") for line in lines]
378+
self._index = 0
379+
380+
def read1(self, size=-1):
381+
if self._index >= len(self._lines):
382+
return b""
383+
result = self._lines[self._index]
384+
self._index += 1
385+
return result
386+
375387
def __init__(self, cmd, **kwargs):
376388
popen_calls.append((cmd, kwargs))
377389
envs.append(kwargs.get("env"))
378390
self.pid = 99999
379391
self.returncode = 0
380-
self.stderr = iter(line.encode("utf-8") for line in stderr_lines)
392+
self.stderr = FakePopen.FakeStderr(stderr_lines)
381393

382394
def wait(self):
383395
return self.returncode

0 commit comments

Comments
 (0)