def test_output_as_generator(self):
contents = 'this is the simulated file contents'
session = flexmock()
expectation = session.should_receive('downloadTaskOutput')
for chunk in contents:
expectation = expectation.and_return(chunk)
# Empty content to simulate end of stream.
expectation.and_return('')
streamer = koji_util.stream_task_output(session, 123, 'file.ext')
assert ''.join(list(streamer)) == contents
def test_output_as_generator(self):
contents = 'this is the simulated file contents'
session = flexmock()
expectation = session.should_receive('downloadTaskOutput')
for chunk in contents:
expectation = expectation.and_return(chunk)
# Empty content to simulate end of stream.
expectation.and_return('')
streamer = koji_util.stream_task_output(session, 123, 'file.ext')
assert ''.join(list(streamer)) == contents
def download_filesystem(self, task_id, filesystem_regex):
found = self.find_filesystem(task_id, filesystem_regex)
if found is None:
raise RuntimeError('Filesystem not found as task output: {}'
.format(filesystem_regex.pattern))
task_id, file_name = found
self.log.info('Streaming filesystem: %s from task ID: %s',
file_name, task_id)
contents = stream_task_output(self.session, task_id, file_name,
self.blocksize)
return contents
def download_filesystem(self, task_id, filesystem_regex):
found = self.find_filesystem(task_id, filesystem_regex)
if found is None:
raise RuntimeError('Filesystem not found as task output: {}'
.format(filesystem_regex.pattern))
task_id, file_name = found
self.log.info('Streaming filesystem: %s from task ID: %s',
file_name, task_id)
contents = stream_task_output(self.session, task_id, file_name,
self.blocksize)
return contents