def test_tape_empty_when_OSError_then_raise_exception(self, mock_tarfile):
exception = OSError()
mock_tarfile.open.side_effect = exception
with self.assertRaises(OSError):
tape_empty("some_drive")
mock_tarfile.open.assert_called_once()
def test_tape_empty_when_tarfile_ReadError_and_unknown_message_then_raise_exception(self, mock_tarfile_open):
exception = tarfile.ReadError()
exception.message = 'some other message'
mock_tarfile_open.side_effect = exception
with self.assertRaises(tarfile.ReadError):
tape_empty("some_drive")
mock_tarfile_open.assert_called_once()
def test_tape_empty_when_tarfile_ReadError_with_msg_then_drive_is_empty(self, mock_rewind_tape, mock_tarfile_open):
exception = tarfile.ReadError('empty file')
mock_tarfile_open.side_effect = exception
self.assertTrue(tape_empty("some_drive"))
mock_tarfile_open.assert_called_once()
mock_rewind_tape.assert_called_once_with("some_drive")
def test_tape_empty_when_IO_error_then_drive_is_empty(self, mock_rewind_tape, mock_tarfile):
exception = IOError()
exception.errno = errno.EIO
mock_tarfile.open.side_effect = exception
self.assertEqual(tape_empty("some_drive"), True)
mock_tarfile.open.assert_called_once()
mock_rewind_tape.assert_called_once_with("some_drive")
def test_tape_empty_when_drive_not_empty(self, mock_rewind_tape, mock_tarfile):
mocked_tar = mock.Mock()
mock_tarfile.open.return_value = mocked_tar
self.assertEqual(tape_empty("some_drive"), False)
mock_tarfile.open.assert_called_once()
mocked_tar.close.assert_called_once()
mock_rewind_tape.assert_called_once_with("some_drive")
def run(self, medium=None, drive=None, timeout=120):
"""
Mounts tape into drive
Args:
medium: Which medium to mount
drive: Which drive to load to
"""
medium = StorageMedium.objects.get(pk=medium)
slot = medium.tape_slot.slot_id
tape_drive = TapeDrive.objects.get(pk=drive)
mount_tape(tape_drive.robot.device, slot, drive)
wait_to_come_online(tape_drive.device, timeout)
if medium.format not in [100, 101]:
label_root = Path.objects.get(entity='label').value
xmlpath = os.path.join(label_root,
'%s_label.xml' % medium.medium_id)
if tape_empty(tape_drive.device):
create_tape_label(medium, xmlpath)
rewind_tape(tape_drive.device)
write_to_tape(tape_drive.device, xmlpath)
else:
tar = tarfile.open(tape_drive.device, 'r|')
first_member = tar.getmembers()[0]
if first_member.name.endswith('_label.xml'):
xmlstring = tar.extractfile(first_member).read()
tar.close()
if not verify_tape_label(medium, xmlstring):
raise ValueError(
'Tape contains labelfile with wrong tapeid')
elif first_member.name == 'reuse':
tar.close()
create_tape_label(medium, xmlpath)
rewind_tape(tape_drive.device)
write_to_tape(tape_drive.device, xmlpath)
else:
raise ValueError('Tape contains unknown information')
TapeDrive.objects.filter(pk=drive).update(
num_of_mounts=F('num_of_mounts') + 1, )
StorageMedium.objects.filter(pk=medium.pk).update(
num_of_mounts=F('num_of_mounts') + 1, tape_drive_id=drive)
def write_medium_label_to_drive(drive_id, medium, slot_id, tape_drive):
xmlfile = tempfile.NamedTemporaryFile(delete=False)
try:
arcname = '%s_label.xml' % medium.medium_id
if medium.format not in [100, 101]:
if tape_empty(tape_drive.device):
create_tape_label(medium, xmlfile.name)
rewind_tape(tape_drive.device)
write_to_tape(tape_drive.device, xmlfile.name, arcname=arcname)
else:
tar = tarfile.open(tape_drive.device, 'r|')
first_member = tar.getmembers()[0]
tar.close()
rewind_tape(tape_drive.device)
if first_member.name.endswith('_label.xml'):
tar = tarfile.open(tape_drive.device, 'r|')
xmlstring = tar.extractfile(first_member).read()
tar.close()
if not verify_tape_label(medium, xmlstring):
raise ValueError('Tape contains invalid label file')
elif first_member.name == 'reuse':
create_tape_label(medium, xmlfile.name)
rewind_tape(tape_drive.device)
write_to_tape(tape_drive.device,
xmlfile.name,
arcname=arcname)
else:
raise ValueError('Tape contains unknown information')
rewind_tape(tape_drive.device)
except BaseException:
StorageMedium.objects.filter(pk=medium.pk).update(
status=100,
last_changed_local=timezone.now(),
)
TapeDrive.objects.filter(pk=drive_id).update(locked=False, status=100)
TapeSlot.objects.filter(slot_id=slot_id).update(status=100)
raise
finally:
xmlfile.close()
TapeDrive.objects.filter(pk=drive_id).update(locked=False)
def run(self, medium=None, drive=None, timeout=120):
"""
Mounts tape into drive
Args:
medium: Which medium to mount
drive: Which drive to load to
"""
medium = StorageMedium.objects.get(pk=medium)
slot = medium.tape_slot.slot_id
tape_drive = TapeDrive.objects.get(pk=drive)
if tape_drive.locked:
raise TapeDriveLockedError()
tape_drive.locked = True
tape_drive.save(update_fields=['locked'])
try:
mount_tape(tape_drive.robot.device, slot, tape_drive.drive_id)
wait_to_come_online(tape_drive.device, timeout)
except:
StorageMedium.objects.filter(pk=medium.pk).update(status=100)
TapeDrive.objects.filter(pk=drive).update(locked=False, status=100)
TapeSlot.objects.filter(slot_id=slot).update(status=100)
raise
TapeDrive.objects.filter(pk=drive).update(
num_of_mounts=F('num_of_mounts') + 1,
last_change=timezone.now(),
)
StorageMedium.objects.filter(pk=medium.pk).update(
num_of_mounts=F('num_of_mounts') + 1, tape_drive_id=drive)
xmlfile = tempfile.NamedTemporaryFile(delete=False)
try:
arcname = '%s_label.xml' % medium.medium_id
if medium.format not in [100, 101]:
if tape_empty(tape_drive.device):
create_tape_label(medium, xmlfile.name)
rewind_tape(tape_drive.device)
write_to_tape(tape_drive.device,
xmlfile.name,
arcname=arcname)
else:
tar = tarfile.open(tape_drive.device, 'r|')
first_member = tar.getmembers()[0]
tar.close()
rewind_tape(tape_drive.device)
if first_member.name.endswith('_label.xml'):
tar = tarfile.open(tape_drive.device, 'r|')
xmlstring = tar.extractfile(first_member).read()
tar.close()
if not verify_tape_label(medium, xmlstring):
raise ValueError(
'Tape contains invalid label file')
elif first_member.name == 'reuse':
create_tape_label(medium, xmlfile.name)
rewind_tape(tape_drive.device)
write_to_tape(tape_drive.device,
xmlfile.name,
arcname=arcname)
else:
raise ValueError('Tape contains unknown information')
rewind_tape(tape_drive.device)
except:
StorageMedium.objects.filter(pk=medium.pk).update(status=100)
TapeDrive.objects.filter(pk=drive).update(locked=False, status=100)
TapeSlot.objects.filter(slot_id=slot).update(status=100)
raise
finally:
xmlfile.close()
TapeDrive.objects.filter(pk=drive).update(locked=False)