def test_get_cached_exists(self, get_object, get_download):
mock_object = mock.Mock()
get_object.return_value = mock_object
get_download.return_value = '/download'
res = resource.get_cached_file('/candidate', b'', filename='download.csv')
assert res == '/download'
get_download.assert_called_once_with(mock_object, filename='download.csv')
def test_get_cached_exists(self, get_object, get_download):
mock_object = mock.Mock()
get_object.return_value = mock_object
get_download.return_value = '/download'
res = resource.get_cached_file('/candidate', b'', filename='download.csv')
assert res == '/download'
get_download.assert_called_once_with(mock_object, filename='download.csv')
def test_get_cached_not_exists(self, get_object):
mock_object = mock.Mock()
def get_metadata():
raise ClientError({'Error': {}}, 'test')
mock_metadata = mock.PropertyMock(side_effect=get_metadata)
type(mock_object).metadata = mock_metadata
get_object.return_value = mock_object
res = resource.get_cached_file('/candidate', b'')
assert res is None
def test_get_cached_not_exists(self, get_object):
mock_object = mock.Mock()
def get_metadata():
raise ClientError({'Error': {}}, 'test')
mock_metadata = mock.PropertyMock(side_effect=get_metadata)
type(mock_object).metadata = mock_metadata
get_object.return_value = mock_object
res = resource.get_cached_file('/candidate', b'')
assert res is None