diff --git a/Lib/codecs.py b/Lib/codecs.py index f05c4f7e2db..9490602ad3c 100644 --- a/Lib/codecs.py +++ b/Lib/codecs.py @@ -699,6 +699,10 @@ class StreamReaderWriter: self.reader.reset() self.writer.reset() + def seek(self, offset, whence=0): + self.reader.seek(offset, whence) + self.writer.seek(offset, whence) + def __getattr__(self, name, getattr=getattr): diff --git a/Lib/test/test_codecs.py b/Lib/test/test_codecs.py index 227e37513f5..13165723739 100644 --- a/Lib/test/test_codecs.py +++ b/Lib/test/test_codecs.py @@ -1592,6 +1592,26 @@ class SurrogateEscapeTest(unittest.TestCase): b"\xe4\xeb\xef\xf6\xfc") +class BomTest(unittest.TestCase): + def test_seek0(self): + data = "1234567890" + tests = ("utf-16", + "utf-16-le", + "utf-16-be", + "utf-32", + "utf-32-le", + "utf-32-be") + for encoding in tests: + with codecs.open('foo', 'w+', encoding=encoding) as f: + # Check if the BOM is written only once + f.write(data) + f.write(data) + f.seek(0) + self.assertEquals(f.read(), data * 2) + f.seek(0) + self.assertEquals(f.read(), data * 2) + + def test_main(): support.run_unittest( UTF32Test, @@ -1619,6 +1639,7 @@ def test_main(): WithStmtTest, TypesTest, SurrogateEscapeTest, + BomTest, ) diff --git a/Misc/NEWS b/Misc/NEWS index 32f188c6cac..54ebf596d7f 100644 --- a/Misc/NEWS +++ b/Misc/NEWS @@ -54,6 +54,9 @@ C-API Library ------- +- Issue #6268: Fix seek() method of codecs.open(), don't read the BOM twice + after seek(0) + - Issue #8782: Add a trailing newline in linecache.updatecache to the last line of files without one.