diff --git a/tests/tools/test_local_env_blocklist.py b/tests/tools/test_local_env_blocklist.py index 70a8ae5d11..fdb68f002b 100644 --- a/tests/tools/test_local_env_blocklist.py +++ b/tests/tools/test_local_env_blocklist.py @@ -25,8 +25,7 @@ def _make_fake_popen(captured: dict): proc = MagicMock() proc.poll.return_value = 0 proc.returncode = 0 - proc.stdout = iter([]) - proc.stdout.close = lambda: None + proc.stdout = MagicMock(__iter__=lambda s: iter([]), __next__=lambda s: (_ for _ in ()).throw(StopIteration)) proc.stdin = MagicMock() return proc return fake_popen