def setup(self): disk_provider = mock.Mock() disk_provider.get_device = mock.Mock(return_value='/dev/loop0') self.partitioner = PartitionerGpt(disk_provider)
def setup(self): disk_provider = mock.Mock() disk_provider.get_device = mock.Mock( return_value='/dev/loop0' ) self.partitioner = PartitionerGpt(disk_provider)
class TestPartitionerGpt(object): def setup(self): disk_provider = mock.Mock() disk_provider.get_device = mock.Mock(return_value='/dev/loop0') self.partitioner = PartitionerGpt(disk_provider) def test_get_id(self): assert self.partitioner.get_id() == 0 @patch('kiwi.partitioner_gpt.Command.run') @patch('kiwi.partitioner_gpt.PartitionerGpt.set_flag') def test_create(self, mock_flag, mock_command): self.partitioner.create('name', 100, 't.linux', ['t.csm']) mock_command.assert_called_once_with( ['sgdisk', '-n', '1:0:+100M', '-c', '1:name', '/dev/loop0']) call = mock_flag.call_args_list[0] assert mock_flag.call_args_list[0] == \ call(1, 't.linux') call = mock_flag.call_args_list[1] assert mock_flag.call_args_list[1] == \ call(1, 't.csm') @patch('kiwi.partitioner_gpt.Command.run') @patch('kiwi.partitioner_gpt.PartitionerGpt.set_flag') def test_create_all_free(self, mock_flag, mock_command): self.partitioner.create('name', 'all_free', 't.linux') mock_command.assert_called_once_with( ['sgdisk', '-n', '1:0:0', '-c', '1:name', '/dev/loop0']) @raises(KiwiPartitionerGptFlagError) def test_set_flag_invalid(self): self.partitioner.set_flag(1, 'foo') @patch('kiwi.partitioner_gpt.Command.run') def test_set_flag(self, mock_command): self.partitioner.set_flag(1, 't.csm') mock_command.assert_called_once_with( ['sgdisk', '-t', '1:EF02', '/dev/loop0']) @patch('kiwi.logger.log.warning') def test_set_flag_ignored(self, mock_warn): self.partitioner.set_flag(1, 'f.active') assert mock_warn.called
class TestPartitionerGpt(object): def setup(self): disk_provider = mock.Mock() disk_provider.get_device = mock.Mock( return_value='/dev/loop0' ) self.partitioner = PartitionerGpt(disk_provider) def test_get_id(self): assert self.partitioner.get_id() == 0 @patch('kiwi.partitioner_gpt.Command.run') @patch('kiwi.partitioner_gpt.PartitionerGpt.set_flag') def test_create(self, mock_flag, mock_command): self.partitioner.create('name', 100, 't.linux', ['t.csm']) mock_command.assert_called_once_with( ['sgdisk', '-n', '1:0:+100M', '-c', '1:name', '/dev/loop0'] ) call = mock_flag.call_args_list[0] assert mock_flag.call_args_list[0] == \ call(1, 't.linux') call = mock_flag.call_args_list[1] assert mock_flag.call_args_list[1] == \ call(1, 't.csm') @patch('kiwi.partitioner_gpt.Command.run') @patch('kiwi.partitioner_gpt.PartitionerGpt.set_flag') def test_create_all_free(self, mock_flag, mock_command): self.partitioner.create('name', 'all_free', 't.linux') mock_command.assert_called_once_with( ['sgdisk', '-n', '1:0:0', '-c', '1:name', '/dev/loop0'] ) @raises(KiwiPartitionerGptFlagError) def test_set_flag_invalid(self): self.partitioner.set_flag(1, 'foo') @patch('kiwi.partitioner_gpt.Command.run') def test_set_flag(self, mock_command): self.partitioner.set_flag(1, 't.csm') mock_command.assert_called_once_with( ['sgdisk', '-t', '1:EF02', '/dev/loop0'] ) @patch('kiwi.logger.log.warning') def test_set_flag_ignored(self, mock_warn): self.partitioner.set_flag(1, 'f.active') assert mock_warn.called