Skip to content

Commit 31d5fc9

Browse files
committed
Make the mocking more obvious and reliable
1 parent 9892dfd commit 31d5fc9

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

Lib/test/test_zipfile/test_core.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5801,9 +5801,9 @@ def test_append_keep_efs_flag(self):
58015801
efs_flags = [True, True, False, False]
58025802

58035803
def mock_encode(self):
5804-
if efs_flags[i]:
5805-
zinfo.flag_bits |= zipfile._MASK_UTF_FILENAME
5806-
return (self.filename.encode('ascii'), self.flag_bits)
5804+
idx = names.index(self.filename)
5805+
return (self.filename.encode('ascii'), self.flag_bits | (
5806+
zipfile._MASK_UTF_FILENAME if efs_flags[idx] else 0))
58075807

58085808
with mock.patch('zipfile.ZipInfo._encodeFilenameFlags', mock_encode), \
58095809
zipfile.ZipFile(TESTFN, "w") as zipfp:

0 commit comments

Comments
 (0)