[Python-checkins] r68349 - sandbox/trunk/io-c/test_io.py

[antoine.pitrou]

Author: antoine.pitrou
Date: Mon Jan  5 22:01:21 2009
New Revision: 68349

Log:
add a readinto() test to Buffered{Reader,Random}



Modified:
   sandbox/trunk/io-c/test_io.py

Modified: sandbox/trunk/io-c/test_io.py
==============================================================================
--- sandbox/trunk/io-c/test_io.py	(original)
+++ sandbox/trunk/io-c/test_io.py	Mon Jan  5 22:01:21 2009
@@ -462,6 +462,21 @@
         self.assertEquals(rawio._reads, 3)
         self.assertEquals(b"", bufio.read1(100))
 
+    def testReadinto(self):
+        rawio = MockRawIO((b"abc", b"d", b"efg"))
+        bufio = self.tp(rawio)
+        b = bytearray(2)
+        self.assertEquals(bufio.readinto(b), 2)
+        self.assertEquals(b, b"ab")
+        self.assertEquals(bufio.readinto(b), 2)
+        self.assertEquals(b, b"cd")
+        self.assertEquals(bufio.readinto(b), 2)
+        self.assertEquals(b, b"ef")
+        self.assertEquals(bufio.readinto(b), 1)
+        self.assertEquals(b, b"gf")
+        self.assertEquals(bufio.readinto(b), 0)
+        self.assertEquals(b, b"gf")
+
     def testBuffering(self):
         data = b"abcdefghi"
         dlen = len(data)