From 10402a306f613b37c96b51771e7ad0e1cf60a3a4 Mon Sep 17 00:00:00 2001
From: "Michael W. Hudson" <mwh@python.net>
Date: Thu, 22 Sep 2005 09:19:01 +0000
Subject: Patches #1298449 and #1298499: Add some missing checks for error
 returns in cStringIO.c.  Thanks to Andrew Bennetts.

This must be a backport candidate.
---
 Lib/test/test_StringIO.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'Lib/test/test_StringIO.py')

diff --git a/Lib/test/test_StringIO.py b/Lib/test/test_StringIO.py
index e74a427af0..c61f7cc54a 100644
--- a/Lib/test/test_StringIO.py
+++ b/Lib/test/test_StringIO.py
@@ -44,6 +44,13 @@ class TestGenericStringIO(unittest.TestCase):
         f.seek(0)
         self.assertEqual(f.getvalue(), 'abc')
 
+    def test_writelines_error(self):
+        def errorGen():
+            yield 'a'
+            raise KeyboardInterrupt()
+        f = self.MODULE.StringIO()
+        self.assertRaises(KeyboardInterrupt, f.writelines, errorGen())
+
     def test_truncate(self):
         eq = self.assertEqual
         f = self.MODULE.StringIO()
-- 
cgit v1.2.1