From: Walter Dörwald <walter@livinglogic.de>
Date: Mon, 6 Mar 2006 22:44:03 +0000 (+0000)
Subject: Backport revision 42872:
X-Git-Tag: v2.4.3c1~50
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=a4b4dc328f0769f90c3ebb743280b48d4eae7b7c;p=thirdparty%2FPython%2Fcpython.git

Backport revision 42872:
If size is specified, try to read at least size characters.
This is a alternative version of patch #1379332.
---

diff --git a/Lib/codecs.py b/Lib/codecs.py
index 103a6f62431e..9a217b3a1928 100644
--- a/Lib/codecs.py
+++ b/Lib/codecs.py
@@ -274,7 +274,10 @@ class StreamReader(Codec):
         while True:
             # can the request can be satisfied from the character buffer?
             if chars < 0:
-                if self.charbuffer:
+                if size < 0:
+                    if self.charbuffer:
+                        break
+                elif len(self.charbuffer) >= size:
                     break
             else:
                 if len(self.charbuffer) >= chars:
diff --git a/Lib/test/test_codecs.py b/Lib/test/test_codecs.py
index 44d7896948bb..a1a8073c3965 100644
--- a/Lib/test/test_codecs.py
+++ b/Lib/test/test_codecs.py
@@ -57,19 +57,23 @@ class ReadTest(unittest.TestCase):
             stream = StringIO.StringIO(input.encode(self.encoding))
             return codecs.getreader(self.encoding)(stream)
 
-        def readalllines(input, keepends=True):
+        def readalllines(input, keepends=True, size=None):
             reader = getreader(input)
             lines = []
             while True:
-                line = reader.readline(keepends=keepends)
+                line = reader.readline(size=size, keepends=keepends)
                 if not line:
                     break
                 lines.append(line)
-            return "".join(lines)
+            return "|".join(lines)
 
         s = u"foo\nbar\r\nbaz\rspam\u2028eggs"
-        self.assertEqual(readalllines(s, True), s)
-        self.assertEqual(readalllines(s, False), u"foobarbazspameggs")
+        sexpected = u"foo\n|bar\r\n|baz\r|spam\u2028|eggs"
+        sexpectednoends = u"foo|bar|baz|spam|eggs"
+        self.assertEqual(readalllines(s, True), sexpected)
+        self.assertEqual(readalllines(s, False), sexpectednoends)
+        self.assertEqual(readalllines(s, True, 10), sexpected)
+        self.assertEqual(readalllines(s, False, 10), sexpectednoends)
 
         # Test long lines (multiple calls to read() in readline())
         vw = []