def make_model_and_disk():
model = FilesystemModel(prober=None)
disk = Disk.from_info(
FakeStorageInfo(name='disk-name',
size=100 * (2**30),
free=50 * (2**30)))
model._available_disks[disk.name] = disk
return model, disk
def make_disk(model, **kw):
serial = 'serial%s' % len(model._actions)
model._actions.append(Disk(
m=model, serial=serial,
info=FakeStorageInfo(size=100*(2**30)),
**kw))
disk = model._actions[-1]
return disk
def make_view(self, partition=None):
controller = mock.create_autospec(spec=FilesystemController)
model = mock.create_autospec(spec=FilesystemModel)
model.fs_by_name = FilesystemModel.fs_by_name
disk = Disk.from_info(
FakeStorageInfo(name='disk-name',
size=100 * (2**20),
free=50 * (2**20)))
return PartitionView(model, controller, disk, partition)
def make_view(self, partition=None):
controller = mock.create_autospec(spec=FilesystemController)
model = mock.create_autospec(spec=FilesystemModel)
model.fs_by_name = FilesystemModel.fs_by_name
info = mock.create_autospec(spec=StorageInfo)
info.name = 'disk-name'
info.size = 100
info.free = 50
disk = Disk.from_info(info)
return PartitionView(model, controller, disk, partition)
def test_one_disk(self):
disk = Disk.from_info(
FakeStorageInfo(name='disk-name',
size=100 * (2**20),
free=50 * (2**20),
serial="DISK-SERIAL"))
view = self.make_view([disk])
w = view_helpers.find_with_pred(
view,
lambda w: isinstance(w, urwid.Text) and "DISK-SERIAL" in w.text)
self.assertIsNotNone(w, "could not find DISK-SERIAL in view")
def test_one_disk(self):
model = mock.create_autospec(spec=FilesystemModel)
disk = Disk(m=model,
serial="DISK-SERIAL",
info=FakeStorageInfo(size=100 * (2**20),
free=50 * (2**20)))
view = self.make_view(model, [disk])
w = view_helpers.find_with_pred(
view,
lambda w: isinstance(w, urwid.Text) and "DISK-SERIAL" in w.text)
self.assertIsNotNone(w, "could not find DISK-SERIAL in view")
def make_disk(fs_model, **kw):
if 'serial' not in kw:
kw['serial'] = 'serial%s' % len(fs_model._actions)
if 'path' not in kw:
kw['path'] = '/dev/thing'
size = kw.pop('size', 100*(2**30))
fs_model._actions.append(Disk(
m=fs_model, info=FakeStorageInfo(size=size),
**kw))
disk = fs_model._actions[-1]
return disk
def make_view(self, partition=None):
controller = mock.create_autospec(spec=FilesystemController)
model = mock.create_autospec(spec=FilesystemModel)
model.fs_by_name = FilesystemModel.fs_by_name
disk = Disk.from_info(
FakeStorageInfo(name='disk-name',
size=100 * (2**20),
free=50 * (2**20)))
base_view = BaseView(urwid.Text(""))
base_view.model = model
base_view.controller = controller
base_view.refresh_model_inputs = lambda: None
stretchy = PartitionStretchy(base_view, disk, partition)
base_view.show_stretchy_overlay(stretchy)
return base_view, stretchy