def test_refresh_volumes(self):
"""Test for refresh_volumes method."""
udf = self._create_udf()
yield self.main.vm.add_udf(udf)
share = self._create_share()
yield self.main.vm.add_share(share)
volumes = list(self.main.vm.get_volumes(all_volumes=True))
def volumes_changed():
"""Fake volumes_changed."""
self.main.event_q.push('VM_VOLUMES_CHANGED', volumes=volumes)
self.patch(self.main.vm, 'refresh_volumes', volumes_changed)
result = yield self.tool.refresh_volumes()
str_volumes = []
for volume in volumes:
if isinstance(volume, volume_manager.UDF):
str_vol = interaction_interfaces.get_udf_dict(volume)
else:
str_vol = interaction_interfaces.get_share_dict(volume)
str_volumes.append(str_vol)
self.assertEqual(result, str_volumes)
def handle_SYS_QUOTA_EXCEEDED(self, volume_id, free_bytes):
"""Handle the SYS_QUOTA_EXCEEDED event."""
volume = self.vm.get_volume(str(volume_id))
volume_dict = {}
if isinstance(volume, UDF):
volume_dict = get_udf_dict(volume)
to_call = self.status_frontend.udf_quota_exceeded
elif isinstance(volume, Root):
volume_dict = get_share_dict(volume)
to_call = self.status_frontend.root_quota_exceeded
else:
volume_dict = get_share_dict(volume)
to_call = self.status_frontend.share_quota_exceeded
volume_dict['free_bytes'] = str(free_bytes)
to_call(volume_dict)
def emit_folder_delete_error(self, folder, error):
"""Emit the FolderCreateError signal"""
udf_dict = get_udf_dict(folder)
self.emit_signal('on_folder_delete_error', udf_dict, str(error))
def emit_folder_unsubscribed(self, folder):
"""Emit the FolderUnSubscribed signal"""
udf_dict = get_udf_dict(folder)
self.emit_signal('on_folder_unsubscribed', udf_dict)
def emit_folder_deleted(self, folder):
"""Emit the FolderCreated signal"""
udf_dict = get_udf_dict(folder)
self.emit_signal('on_folder_deleted', udf_dict)
