def test_transfer(self):
with self.mock_runner() as runner:
with mock.patch.object(self.copier, 'verify') as mock_verify:
self.cfg.set_override('ssh_chunk_size', 10, 'migrate')
runner.run.return_value = sizeof_format.parse_size('100M')
with mock.patch.object(self.copier, 'run_scp') as mock_run_scp:
self.copier.transfer(self.data)
self.assertEqual(10, mock_run_scp.call_count)
self.assertCalledOnce(mock_verify)
def test_transfer(self):
with self.mock_runner() as runner:
with mock.patch.object(self.copier, 'verify') as mock_verify:
self.cfg.set_override('ssh_chunk_size', 10, 'migrate')
runner.run.return_value = sizeof_format.parse_size('100M')
with mock.patch.object(self.copier, 'run_scp') as mock_run_scp:
self.copier.transfer(self.data)
self.assertEqual(10, mock_run_scp.call_count)
self.assertCalledOnce(mock_verify)
def __init__(self, file_obj, speed_limit=None, size=None, name=None,
chunk_size='512K'):
self.file_obj = file_obj
self.chunk_size = sizeof_format.parse_size(chunk_size)
self.speed_limiter = SpeedLimiter(speed_limit or
CONF.migrate.speed_limit)
if size is None:
size = get_file_size(file_obj)
name = name or getattr(file_obj, 'name', '<file object>')
self.view = ProgressView(name=name, size=size)
def __init__(self,
filename='%(scenario)s-%(date)s.log',
date_format='%F-%H-%M-%S',
**kwargs):
self.date_format = date_format
max_bytes = sizeof_format.parse_size(kwargs.pop('maxBytes', 0))
super(RunRotatingFileHandler, self).__init__(
filename=self.get_filename(filename),
maxBytes=max_bytes,
**kwargs)
def __init__(self,
filename='%(scenario)s-%(date)s.log',
date_format='%F-%H-%M-%S',
**kwargs):
self.date_format = date_format
max_bytes = sizeof_format.parse_size(kwargs.pop('maxBytes', 0))
super(RunRotatingFileHandler, self).__init__(
filename=self.get_filename(filename),
maxBytes=max_bytes,
**kwargs)
def __init__(self, name=None, size=None):
self.name = name
self.size = size
self.size_hr = sizeof_format.sizeof_fmt(size) if size else "NAN"
self.show_size = (size / 100 if size else
sizeof_format.parse_size('100MB'))
self.current_show_size = 0
self.progress_message = "Copying %(name)s: %(progress)s of %(size)s"
if size:
self.progress_message += " %(percentage)s%%"
self.progress = 0
self.first_run = None
def parse_size(self, speed_limit=None):
return sizeof_format.parse_size(speed_limit)
def __init__(self, speed_limit=None):
self.speed_limit = sizeof_format.parse_size(speed_limit)
self.prev_time = None
self.sent_size = 0
def parse_size(self, speed_limit=None):
return sizeof_format.parse_size(speed_limit)
