From 9549a3e3d4be2a15c222996abff8cb97180ee9be Mon Sep 17 00:00:00 2001 From: Robert Collins Date: Mon, 16 May 2016 15:22:01 +1200 Subject: Issue #26807: mock_open 'files' no longer error on readline at end of file. Patch from Yolanda Robla. --- Lib/unittest/test/testmock/testmock.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'Lib/unittest/test/testmock/testmock.py') diff --git a/Lib/unittest/test/testmock/testmock.py b/Lib/unittest/test/testmock/testmock.py index 03c95de6cd5..5f82b829661 100644 --- a/Lib/unittest/test/testmock/testmock.py +++ b/Lib/unittest/test/testmock/testmock.py @@ -1419,6 +1419,18 @@ class MockTest(unittest.TestCase): self.assertEqual('abc', first) self.assertEqual('abc', second) + def test_mock_open_after_eof(self): + # read, readline and readlines should work after end of file. + _open = mock.mock_open(read_data='foo') + h = _open('bar') + h.read() + self.assertEqual('', h.read()) + self.assertEqual('', h.read()) + self.assertEqual('', h.readline()) + self.assertEqual('', h.readline()) + self.assertEqual([], h.readlines()) + self.assertEqual([], h.readlines()) + def test_mock_parents(self): for Klass in Mock, MagicMock: m = Klass() -- cgit v1.2.3