diff options
| author | Antoine Pitrou <solipsis@pitrou.net> | 2012-06-30 15:32:41 (GMT) |
|---|---|---|
| committer | Antoine Pitrou <solipsis@pitrou.net> | 2012-06-30 15:32:41 (GMT) |
| commit | f93b705bd0d6c0338670d051d83748ea861d37b5 (patch) | |
| tree | 77673973358acbd9f90345c568a210f1604b66be /Lib/test/test_zipfile.py | |
| parent | 24ef3e967f4698da6c85778408e920f6437ed344 (diff) | |
| parent | c399185fcc9ca795840db4c94f273effa2d40947 (diff) | |
| download | cpython-f93b705bd0d6c0338670d051d83748ea861d37b5.zip cpython-f93b705bd0d6c0338670d051d83748ea861d37b5.tar.gz cpython-f93b705bd0d6c0338670d051d83748ea861d37b5.tar.bz2 | |
Issue #9239: add tests for modifying zipfile comments in append mode.
Diffstat (limited to 'Lib/test/test_zipfile.py')
| -rw-r--r-- | Lib/test/test_zipfile.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/Lib/test/test_zipfile.py b/Lib/test/test_zipfile.py index 0772236..2e52c68 100644 --- a/Lib/test/test_zipfile.py +++ b/Lib/test/test_zipfile.py @@ -1134,6 +1134,24 @@ class OtherTests(unittest.TestCase): with zipfile.ZipFile(TESTFN, mode="r") as zipfr: self.assertEqual(zipfr.comment, comment2) + # check that comments are correctly modified in append mode + with zipfile.ZipFile(TESTFN,mode="w") as zipf: + zipf.comment = b"original comment" + zipf.writestr("foo.txt", "O, for a Muse of Fire!") + with zipfile.ZipFile(TESTFN,mode="a") as zipf: + zipf.comment = b"an updated comment" + with zipfile.ZipFile(TESTFN,mode="r") as zipf: + self.assertEqual(zipf.comment, b"an updated comment") + + # check that comments are correctly shortened in append mode + with zipfile.ZipFile(TESTFN,mode="w") as zipf: + zipf.comment = b"original comment that's longer" + zipf.writestr("foo.txt", "O, for a Muse of Fire!") + with zipfile.ZipFile(TESTFN,mode="a") as zipf: + zipf.comment = b"shorter comment" + with zipfile.ZipFile(TESTFN,mode="r") as zipf: + self.assertEqual(zipf.comment, b"shorter comment") + def test_unicode_comment(self): with zipfile.ZipFile(TESTFN, "w", zipfile.ZIP_STORED) as zipf: zipf.writestr("foo.txt", "O, for a Muse of Fire!") |