saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

add a dict containing the handles to the mock_open return object

Browse source 
This allows tests to access the MockFH objects opened during the
execution of the unit test.
This commit is contained in:
Erik Johnson 2018-06-17 10:42:53 -05:00
parent 852ba4b982
commit b9200dbc3e
No known key found for this signature in database
GPG key ID: 5E5583C437808F3F
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/support/mock.py
View file

@ -226,6 +226,9 @@ def mock_open(read_data=''):
for x, y in six.iteritems(read_data)]
)
mock = MagicMock(name='open', spec=open)
mock.handles = {}
def _fopen_side_effect(name, *args, **kwargs):
for pat in read_data:
if pat == '*':
@ -237,10 +240,13 @@ def mock_open(read_data=''):
# No non-glob match in read_data, fall back to '*'
matched_pattern = '*'
try:
return MockFH(name, read_data[matched_pattern])
ret = MockFH(name, read_data[matched_pattern])
mock.handles.setdefault(name, []).append(ret)
return ret
except KeyError:
# No matching glob in read_data, treat this as a file that does
# not exist and raise the appropriate exception.
raise IOError(errno.ENOENT, 'No such file or directory', name)
return MagicMock(name='open', spec=open, side_effect=_fopen_side_effect)
mock.side_effect = _fopen_side_effect
return mock