def test_prepare_partition_data_bios_no_gap(self, __salt__, __grains__): partitions = { "devices": { "/dev/sda": { "partitions": [{ "number": 1, "size": "rest", "type": "linux" }], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid self.assertEqual( partmod.prepare_partition_data(partitions), { "/dev/sda": { "label": "msdos", "pmbr_boot": False, "partitions": [ { "part_id": "/dev/sda1", "part_type": "primary", "fs_type": "ext2", "flags": None, "start": "0%", "end": "100%", }, ], }, }, )
def test_prepare_partition_data_fails_fs_type(self, __grains__): partitions = { "devices": { "/dev/sda": { "partitions": [{ "number": 1, "size": "rest", "type": "error" }], }, }, } __grains__.__getitem__.return_value = False with self.assertRaises(SaltInvocationError) as cm: partmod.prepare_partition_data(partitions) self.assertTrue("type error not recognized" in str(cm.exception))
def test_prepare_partition_data_bios_muti_label(self, __salt__, __grains__): partitions = { 'config': { 'label': 'msdos', }, 'devices': { '/dev/sda': { 'partitions': [ { 'number': 1, 'size': 'rest', 'type': 'linux', }, ], }, '/dev/sdb': { 'label': 'gpt', 'partitions': [ { 'number': 1, 'size': 'rest', 'type': 'linux', }, ], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid self.assertEqual(partmod.prepare_partition_data(partitions), { '/dev/sda': { 'label': 'msdos', 'pmbr_boot': False, 'partitions': [ { 'part_id': '/dev/sda1', 'part_type': 'primary', 'fs_type': 'ext2', 'flags': None, 'start': '0%', 'end': '100%', }, ], }, '/dev/sdb': { 'label': 'gpt', 'pmbr_boot': True, 'partitions': [ { 'part_id': '/dev/sdb1', 'part_type': 'primary', 'fs_type': 'ext2', 'flags': None, 'start': '0%', 'end': '100%', }, ], }, })
def test_prepare_partition_data_efi_partitions(self, __salt__, __grains__): partitions = { 'devices': { '/dev/sda': { 'label': 'gpt', 'partitions': [ { 'number': 1, 'size': '500MB', 'type': 'efi', }, { 'number': 2, 'size': '10000MB', 'type': 'linux', }, { 'number': 3, 'size': '5000MB', 'type': 'swap', }, ], }, }, } __grains__.__getitem__.return_value = True __salt__.__getitem__.return_value = filters.is_raid self.assertEqual(partmod.prepare_partition_data(partitions), { '/dev/sda': { 'label': 'gpt', 'pmbr_boot': False, 'partitions': [ { 'part_id': '/dev/sda1', 'part_type': 'primary', 'fs_type': 'fat16', 'flags': ['esp'], 'start': '0MB', 'end': '500.0MB', }, { 'part_id': '/dev/sda2', 'part_type': 'primary', 'fs_type': 'ext2', 'flags': None, 'start': '500.0MB', 'end': '10500.0MB', }, { 'part_id': '/dev/sda3', 'part_type': 'primary', 'fs_type': 'linux-swap', 'flags': None, 'start': '10500.0MB', 'end': '15500.0MB', }, ], }, })
def test_prepare_partition_data_fails_units_invalid( self, __salt__, __grains__): partitions = { "devices": { "/dev/sda": { "partitions": [{ "number": 1, "size": "1Kilo", "type": "swap" }], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid with self.assertRaises(ParseException) as cm: partmod.prepare_partition_data(partitions) self.assertTrue("Kilo not recognized" in str(cm.exception))
def test_prepare_partition_data_fails_fs_type(self, __grains__): partitions = { 'devices': { '/dev/sda': { 'partitions': [ { 'number': 1, 'size': 'rest', 'type': 'error', }, ], }, }, } __grains__.__getitem__.return_value = False with self.assertRaises(SaltInvocationError) as cm: partmod.prepare_partition_data(partitions) self.assertTrue('type error not recognized' in str(cm.exception))
def test_prepare_partition_data_fails_units_invalid(self, __salt__, __grains__): partitions = { 'devices': { '/dev/sda': { 'partitions': [ { 'number': 1, 'size': '1Kilo', 'type': 'swap', }, ], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid with self.assertRaises(ParseException) as cm: partmod.prepare_partition_data(partitions) self.assertTrue('Kilo not recognized' in str(cm.exception))
def test_prepare_partition_data_bios_gpt_post_raid(self, __salt__, __grains__): partitions = { "devices": { "/dev/md0": { "label": "gpt", "partitions": [ { "number": 1, "size": "8MB", "type": "boot" }, { "number": 2, "size": "rest", "type": "linux" }, ], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid self.assertEqual( partmod.prepare_partition_data(partitions), { "/dev/md0": { "label": "gpt", "pmbr_boot": True, "partitions": [ { "part_id": "/dev/md0p1", "part_type": "primary", "fs_type": "ext2", "flags": ["bios_grub"], "start": "0MB", "end": "8.0MB", }, { "part_id": "/dev/md0p2", "part_type": "primary", "fs_type": "ext2", "flags": None, "start": "8.0MB", "end": "100%", }, ], }, }, )
def test_prepare_partition_data_fails_units_initial_gap( self, __salt__, __grains__): partitions = { "config": { "initial_gap": "1024kB" }, "devices": { "/dev/sda": { "partitions": [{ "number": 1, "size": "1MB", "type": "swap" }], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid with self.assertRaises(SaltInvocationError) as cm: partmod.prepare_partition_data(partitions) self.assertTrue("Units needs to be" in str(cm.exception))
def test_prepare_partition_data_fails_units_initial_gap(self, __salt__, __grains__): partitions = { 'config': { 'initial_gap': '1024kB', }, 'devices': { '/dev/sda': { 'partitions': [ { 'number': 1, 'size': '1MB', 'type': 'swap', }, ], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid with self.assertRaises(SaltInvocationError) as cm: partmod.prepare_partition_data(partitions) self.assertTrue('Units needs to be' in str(cm.exception))
def test_prepare_partition_data_fails_rest(self, __salt__, __grains__): partitions = { "devices": { "/dev/sda": { "partitions": [ { "number": 1, "size": "rest", "type": "swap" }, { "number": 2, "size": "rest", "type": "linux" }, ], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid with self.assertRaises(SaltInvocationError) as cm: partmod.prepare_partition_data(partitions) self.assertTrue("rest free space" in str(cm.exception))
def test_prepare_partition_data_fails_rest(self, __salt__, __grains__): partitions = { 'devices': { '/dev/sda': { 'partitions': [ { 'number': 1, 'size': 'rest', 'type': 'swap', }, { 'number': 2, 'size': 'rest', 'type': 'linux', }, ], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid with self.assertRaises(SaltInvocationError) as cm: partmod.prepare_partition_data(partitions) self.assertTrue('rest free space' in str(cm.exception))
def test_prepare_partition_data_bios_gpt_post_raid(self, __salt__, __grains__): partitions = { 'devices': { '/dev/md0': { 'label': 'gpt', 'partitions': [ { 'number': 1, 'size': '8MB', 'type': 'boot', }, { 'number': 2, 'size': 'rest', 'type': 'linux', }, ], }, }, } __grains__.__getitem__.return_value = False __salt__.__getitem__.return_value = filters.is_raid self.assertEqual(partmod.prepare_partition_data(partitions), { '/dev/md0': { 'label': 'gpt', 'pmbr_boot': True, 'partitions': [ { 'part_id': '/dev/md0p1', 'part_type': 'primary', 'fs_type': 'ext2', 'flags': ['bios_grub'], 'start': '0MB', 'end': '8.0MB', }, { 'part_id': '/dev/md0p2', 'part_type': 'primary', 'fs_type': 'ext2', 'flags': None, 'start': '8.0MB', 'end': '100%', }, ], }, })
def test_prepare_partition_data_efi_partitions(self, __salt__, __grains__): partitions = { "devices": { "/dev/sda": { "label": "gpt", "partitions": [ { "number": 1, "size": "500MB", "type": "efi" }, { "number": 2, "size": "10000MB", "type": "linux" }, { "number": 3, "size": "5000MB", "type": "swap" }, ], }, }, } __grains__.__getitem__.return_value = True __salt__.__getitem__.return_value = filters.is_raid self.assertEqual( partmod.prepare_partition_data(partitions), { "/dev/sda": { "label": "gpt", "pmbr_boot": False, "partitions": [ { "part_id": "/dev/sda1", "part_type": "primary", "fs_type": "fat16", "flags": ["esp"], "start": "0MB", "end": "500.0MB", }, { "part_id": "/dev/sda2", "part_type": "primary", "fs_type": "ext2", "flags": None, "start": "500.0MB", "end": "10500.0MB", }, { "part_id": "/dev/sda3", "part_type": "primary", "fs_type": "linux-swap", "flags": None, "start": "10500.0MB", "end": "15500.0MB", }, ], }, }, )