#11488: Add tests for writelines method of SpooledTemporaryFile.

bitdancer · bitdancer · commit d89ee79d19e1 · 2011-03-14T09:55:46.000-04:00
Patch by Evan Dandrea.
diff --git a/Lib/test/test_tempfile.py b/Lib/test/test_tempfile.py
@@ -697,6 +697,23 @@ def test_write_sequential(self):
         f.write(b'x')
         self.assertTrue(f._rolled)
 
+    def test_writelines(self):
+        # Verify writelines with a SpooledTemporaryFile
+        f = self.do_create()
+        f.writelines((b'x', b'y', b'z'))
+        f.seek(0)
+        buf = f.read()
+        self.assertEqual(buf, b'xyz')
+
+    def test_writelines_sequential(self):
+        # A SpooledTemporaryFile should hold exactly max_size bytes, and roll
+        # over afterward
+        f = self.do_create(max_size=35)
+        f.writelines((b'x' * 20, b'x' * 10, b'x' * 5))
+        self.assertFalse(f._rolled)
+        f.write(b'x')
+        self.assertTrue(f._rolled)
+
     def test_sparse(self):
         # A SpooledTemporaryFile that is written late in the file will extend
         # when that occurs
diff --git a/Misc/ACKS b/Misc/ACKS
@@ -176,6 +176,7 @@ Tom Culliton
 Lisandro Dalcin
 Andrew Dalke
 Lars Damerow
+Evan Dandrea
 Eric Daniel
 Scott David Daniels
 Ben Darnell
