class TestGenerateUnitKeyAndMetadata(unittest.TestCase):
def setUp(self):
self.context = mock.MagicMock()
self.context.config = test_config
self.command = UploadOpenstackImageCommand(self.context)
def test_with_cirros(self):
unit_key, metadata = self.command.generate_unit_key_and_metadata(data.cirros_img_path)
self.assertEqual(unit_key, {'image_checksum': '64d7c1cd2b6f60c92c14662941cb7913'})
self.assertEqual(metadata, {'image_protected': True,
'image_filename': 'cirros-0.3.2-x86_64-disk.img',
'image_size': 13167616})
def test_with_cirros_and_metadata(self):
unit_key, metadata = self.command.generate_unit_key_and_metadata(data.cirros_img_path,
image_min_ram=1024,
image_name="fake name")
self.assertEqual(unit_key, {'image_checksum': '64d7c1cd2b6f60c92c14662941cb7913'})
self.assertEqual(metadata, {'image_protected': True, 'image_min_ram': 1024,
'image_name': "fake name",
'image_filename': 'cirros-0.3.2-x86_64-disk.img',
'image_size': 13167616})
def test_file_does_not_exist(self):
self.assertRaises(IOError, self.command.generate_unit_key_and_metadata, '/a/b/c/d')
class TestDetermineID(unittest.TestCase):
def setUp(self):
self.context = mock.MagicMock()
self.context.config = test_config
self.command = UploadOpenstackImageCommand(self.context)
def test_return_value(self):
ret = self.command.determine_type_id('/a/b/c')
self.assertEqual(ret, constants.IMAGE_TYPE_ID)
def add_upload_section(context, parent_section):
"""
add an upload section to the openstack section
:param context: pulp context
:type context: pulp.client.extensions.core.ClientContext
:param parent_section: section of the CLI to which the upload section
should be added
:type parent_section: pulp.client.extensions.extensions.PulpCliSection
:return: populated section
:rtype: PulpCliSection
"""
upload_section = parent_section.create_subsection(SECTION_UPLOADS,
DESC_UPLOADS)
upload_section.add_command(UploadOpenstackImageCommand(context))
return upload_section
def setUp(self):
self.context = mock.MagicMock()
self.context.config = test_config
self.command = UploadOpenstackImageCommand(self.context)