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, 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