class HardDriveSourceTearDownTestCase(unittest.TestCase):
def setUp(self):
self.source_module = HardDriveSourceModule()
def tear_down_task_order_test(self):
"""Hard drive source tear down task order."""
self.source_module._uses_iso_mount = True
tasks = self.source_module.tear_down_with_tasks()
self.assertEqual(len(tasks), 2)
self.assertIsInstance(tasks[0], TearDownMountTask)
self.assertIsInstance(tasks[1], TearDownMountTask)
name_suffixes = ["-iso", "-device"]
for task, fragment in zip(tasks, name_suffixes):
self.assertTrue(task._target_mount.endswith(fragment))
def single_tear_down_task_test(self):
"""Hard drive source single tear down task."""
self.source_module._uses_iso_mount = False
tasks = self.source_module.tear_down_with_tasks()
self.assertEqual(len(tasks), 1)
self.assertIsInstance(tasks[0], TearDownMountTask)
self.assertTrue(tasks[0]._target_mount.endswith("-device"))
class HardDriveSourceTearDownTestCase(unittest.TestCase):
def setUp(self):
self.source_module = HardDriveSourceModule()
def test_required_space(self):
"""Test the required_space property."""
assert self.source_module.required_space == 0
def test_tear_down_task_order(self):
"""Hard drive source tear down task order."""
tasks = self.source_module.tear_down_with_tasks()
assert len(tasks) == 2
assert isinstance(tasks[0], TearDownMountTask)
assert isinstance(tasks[1], TearDownMountTask)
name_suffixes = ["-iso", "-device"]
for task, fragment in zip(tasks, name_suffixes):
assert task._target_mount.endswith(fragment)
class HardDriveSourceTearDownTestCase(unittest.TestCase):
def setUp(self):
self.source_module = HardDriveSourceModule()
def required_space_test(self):
"""Test the required_space property."""
self.assertEqual(self.source_module.required_space, 0)
def tear_down_task_order_test(self):
"""Hard drive source tear down task order."""
tasks = self.source_module.tear_down_with_tasks()
self.assertEqual(len(tasks), 2)
self.assertIsInstance(tasks[0], TearDownMountTask)
self.assertIsInstance(tasks[1], TearDownMountTask)
name_suffixes = ["-iso", "-device"]
for task, fragment in zip(tasks, name_suffixes):
self.assertTrue(task._target_mount.endswith(fragment))
