def test_display_size(self):
sizes = (
(45, '45 bytes'),
(1000, '1.0 kB'),
(1000 * 1000, '1.0 MB'),
(1000 * 1000 * 500, '500.0 MB'),
(1000 * 1000 * 1000, '1.0 GB'),
(1000 * 1000 * 1000 * 1000, '1.0 TB'),
)
block_device = BlockDevice()
for (size, display_size) in sizes:
block_device.size = size
self.expectThat(block_device.display_size(), Equals(display_size))
def test_minimum_block_device_size(self):
node = factory.make_Node()
blockdevice = BlockDevice(node=node,
name='sda',
block_size=511,
size=143360)
self.assertRaises(ValidationError, blockdevice.save)
def test_negative_size(self):
node = factory.make_Node()
blockdevice = BlockDevice(node=node,
name='sda',
block_size=512,
size=-1)
self.assertRaises(ValidationError, blockdevice.save)
def test_negative_block_device_size(self):
node = factory.make_Node()
blockdevice = BlockDevice(node=node,
name="sda",
block_size=-1,
size=143360)
self.assertRaises(ValidationError, blockdevice.save)
def test_minimum_size(self):
node = factory.make_Node()
blockdevice = BlockDevice(node=node,
name="sda",
block_size=512,
size=143359)
self.assertRaises(ValidationError, blockdevice.save)
def test__get_idx_from_block_name(self):
self.assertEqual(self.idx,
BlockDevice._get_idx_from_block_name(self.name))
def test__get_block_name_from_idx(self):
self.assertEqual(self.name,
BlockDevice._get_block_name_from_idx(self.idx))
def test_get_partition_table_returns_none_for_non_partitioned_device(self):
blockdevice = BlockDevice()
self.assertIsNone(blockdevice.get_partitiontable())
def test_remove_tag_doesnt_error_on_missing_tag(self):
block_device = BlockDevice()
tag = factory.make_name('tag')
#: Test is this doesn't raise an exception
block_device.remove_tag(tag)
def test_remove_tag_deletes_tag(self):
block_device = BlockDevice()
tag = factory.make_name('tag')
block_device.add_tag(tag)
block_device.remove_tag(tag)
self.assertItemsEqual([], block_device.tags)
def test_add_tag_doesnt_duplicate(self):
block_device = BlockDevice()
tag = factory.make_name('tag')
block_device.add_tag(tag)
block_device.add_tag(tag)
self.assertItemsEqual([tag], block_device.tags)
def test_add_tag_adds_new_tag(self):
block_device = BlockDevice()
tag = factory.make_name('tag')
block_device.add_tag(tag)
self.assertItemsEqual([tag], block_device.tags)
def test_get_name(self):
name = factory.make_name("name")
block_device = BlockDevice(name=name)
self.assertEqual(name, block_device.get_name())
