Skip to content

Commit 5a3d0b6

Browse files
committed
test: add getheader mock to account for fetch-content checking content-length
1 parent 4989471 commit 5a3d0b6

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

test/test_scripts_fetch_content.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,13 @@ def mock_urlopen(req, timeout=None, *, cafile=None):
7373
# create a mock context manager
7474
cm = MagicMock()
7575
cm.getcode.return_value = 200
76+
77+
def getheader(field):
78+
if field.lower() == "content-length":
79+
return size
80+
7681
# simulates chunking
82+
cm.getheader = getheader
7783
cm.read.side_effect = [b"foo", b"bar", None]
7884
cm.__enter__.return_value = cm
7985
return cm

0 commit comments

Comments
 (0)