diff options
author | Robert Collins <rbtcollins@hp.com> | 2015-07-23 15:48:45 (GMT) |
---|---|---|
committer | Robert Collins <rbtcollins@hp.com> | 2015-07-23 15:48:45 (GMT) |
commit | 88ba360c88c412c0fe5a75ca50cf11bba08e636b (patch) | |
tree | 5663f9b8330fd997b0c377c92f5718c03244d0e6 /Lib/unittest/test/testmock/testmock.py | |
parent | 4f4913b38bba5e01fc0e7a26f4840bf342365d3d (diff) | |
parent | ca647ef60aa74e3dea81a399ded20c648a69e742 (diff) | |
download | cpython-88ba360c88c412c0fe5a75ca50cf11bba08e636b.zip cpython-88ba360c88c412c0fe5a75ca50cf11bba08e636b.tar.gz cpython-88ba360c88c412c0fe5a75ca50cf11bba08e636b.tar.bz2 |
Issue #21750: Further fixup to be styled like other mock APIs.
Diffstat (limited to 'Lib/unittest/test/testmock/testmock.py')
-rw-r--r-- | Lib/unittest/test/testmock/testmock.py | 24 |
1 files changed, 23 insertions, 1 deletions
diff --git a/Lib/unittest/test/testmock/testmock.py b/Lib/unittest/test/testmock/testmock.py index 683c767..97d844f 100644 --- a/Lib/unittest/test/testmock/testmock.py +++ b/Lib/unittest/test/testmock/testmock.py @@ -1,5 +1,6 @@ import copy import sys +import tempfile import unittest from unittest.test.testmock.support import is_instance @@ -1374,8 +1375,29 @@ class MockTest(unittest.TestCase): def test_mock_open_reuse_issue_21750(self): mocked_open = mock.mock_open(read_data='data') f1 = mocked_open('a-name') + f1_data = f1.read() f2 = mocked_open('another-name') - self.assertEqual(f1.read(), f2.read()) + f2_data = f2.read() + self.assertEqual(f1_data, f2_data) + + def test_mock_open_write(self): + # Test exception in file writing write() + mock_namedtemp = mock.mock_open(mock.MagicMock(name='JLV')) + with mock.patch('tempfile.NamedTemporaryFile', mock_namedtemp): + mock_filehandle = mock_namedtemp.return_value + mock_write = mock_filehandle.write + mock_write.side_effect = OSError('Test 2 Error') + def attempt(): + tempfile.NamedTemporaryFile().write('asd') + self.assertRaises(OSError, attempt) + + def test_mock_open_alter_readline(self): + mopen = mock.mock_open(read_data='foo\nbarn') + mopen.return_value.readline.side_effect = lambda *args:'abc' + first = mopen().readline() + second = mopen().readline() + self.assertEqual('abc', first) + self.assertEqual('abc', second) def test_mock_parents(self): for Klass in Mock, MagicMock: |