mirror of
https://github.com/python/cpython.git
synced 2025-11-02 03:01:58 +00:00
Issue #21750: mock_open.read_data can now be read from each instance, as it
could in Python 3.3.
This commit is contained in:
parent
579db160b3
commit
5329aaa74b
4 changed files with 57 additions and 31 deletions
|
|
@ -1326,6 +1326,11 @@ class MockTest(unittest.TestCase):
|
|||
self.assertEqual(m.mock_calls, [call.__int__(), call.__float__()])
|
||||
self.assertEqual(m.method_calls, [])
|
||||
|
||||
def test_mock_open_reuse_issue_21750(self):
|
||||
mocked_open = mock.mock_open(read_data='data')
|
||||
f1 = mocked_open('a-name')
|
||||
f2 = mocked_open('another-name')
|
||||
self.assertEqual(f1.read(), f2.read())
|
||||
|
||||
def test_mock_parents(self):
|
||||
for Klass in Mock, MagicMock:
|
||||
|
|
|
|||
|
|
@ -141,7 +141,6 @@ class TestMockOpen(unittest.TestCase):
|
|||
|
||||
def test_mock_open_context_manager(self):
|
||||
mock = mock_open()
|
||||
handle = mock.return_value
|
||||
with patch('%s.open' % __name__, mock, create=True):
|
||||
with open('foo') as f:
|
||||
f.read()
|
||||
|
|
@ -149,8 +148,23 @@ class TestMockOpen(unittest.TestCase):
|
|||
expected_calls = [call('foo'), call().__enter__(), call().read(),
|
||||
call().__exit__(None, None, None)]
|
||||
self.assertEqual(mock.mock_calls, expected_calls)
|
||||
self.assertIs(f, handle)
|
||||
# mock_open.return_value is no longer static, because
|
||||
# readline support requires that it mutate state
|
||||
|
||||
def test_mock_open_context_manager_multiple_times(self):
|
||||
mock = mock_open()
|
||||
with patch('%s.open' % __name__, mock, create=True):
|
||||
with open('foo') as f:
|
||||
f.read()
|
||||
with open('bar') as f:
|
||||
f.read()
|
||||
|
||||
expected_calls = [
|
||||
call('foo'), call().__enter__(), call().read(),
|
||||
call().__exit__(None, None, None),
|
||||
call('bar'), call().__enter__(), call().read(),
|
||||
call().__exit__(None, None, None)]
|
||||
self.assertEqual(mock.mock_calls, expected_calls)
|
||||
|
||||
def test_explicit_mock(self):
|
||||
mock = MagicMock()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue