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

Raise TypeError when trying to read from filehandles not opened for reading

Browse source
This commit is contained in:
Erik Johnson 2018-06-23 21:27:58 -05:00
parent 09aae0c82b
commit 3b6356f4b0
No known key found for this signature in database
GPG key ID: 5E5583C437808F3F
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/support/mock.py
View file

@ -105,6 +105,7 @@ class MockFH(object):
except IndexError:
self.mode = kwargs.get('mode', 'r')
self.binary_mode = 'b' in self.mode
self.read_mode = any(x in self.mode for x in ('r', '+'))
self.write_mode = any(x in self.mode for x in ('w', 'a', '+'))
self.empty_string = b'' if self.binary_mode else ''
self.call = MockCall(filename, *args, **kwargs)
@ -186,6 +187,8 @@ class MockFH(object):
def _read(self, size=0):
self.__check_read_data()
if not self.read_mode:
raise IOError('File not open for reading')
if not isinstance(size, six.integer_types) or size < 0:
raise TypeError('a positive integer is required')
@ -206,6 +209,8 @@ class MockFH(object):
def _readlines(self, size=None): # pylint: disable=unused-argument
# TODO: Implement "size" argument
self.__check_read_data()
if not self.read_mode:
raise IOError('File not open for reading')
ret = list(self.read_data_iter)
self.__loc += sum(len(x) for x in ret)
return ret
@ -213,6 +218,8 @@ class MockFH(object):
def _readline(self, size=None): # pylint: disable=unused-argument
# TODO: Implement "size" argument
self.__check_read_data()
if not self.read_mode:
raise IOError('File not open for reading')
try:
ret = next(self.read_data_iter)
self.__loc += len(ret)
@ -222,6 +229,8 @@ class MockFH(object):
def __iter__(self):
self.__check_read_data()
if not self.read_mode:
raise IOError('File not open for reading')
while True:
try:
ret = next(self.read_data_iter)