def setUp(self): super(TestUploadOvercloudImage, self).setUp() # Get the command object to test self.cmd = overcloud_image.UploadOvercloudImage(self.app, None) self.app.client_manager.image = mock.Mock() self.app.client_manager.image.images.create.return_value = ( mock.Mock(id=10, name='imgname', properties={'kernel_id': 10, 'ramdisk_id': 10}, created_at='2015-07-31T14:37:22.000000')) self.cmd._read_image_file_pointer = mock.Mock(return_value=b'IMGDATA') self.cmd._check_file_exists = mock.Mock(return_value=True)
def setUp(self): super(TestUploadOvercloudImageFull, self).setUp() # Get the command object to test self.cmd = overcloud_image.UploadOvercloudImage(self.app, None) self.app.client_manager.image = mock.Mock() self.app.client_manager.image.version = 2.0 self._arch = tripleo_common.arch.kernel_arch() self.app.client_manager.image.images.create.return_value = ( mock.Mock(id=10, name='imgname', properties={'hw_architecture': self._arch}, created_at='2015-07-31T14:37:22.000000')) self.cmd._read_image_file_pointer = mock.Mock(return_value=b'IMGDATA') self.cmd._check_file_exists = mock.Mock(return_value=True)
def setUp(self): super(TestUploadOvercloudImageFullMultiArch, self).setUp() # Get the command object to test self.cmd = overcloud_image.UploadOvercloudImage(self.app, None) self.app.client_manager.image = mock.Mock() self.app.client_manager.image.version = 2.0 # NOTE(tonyb): This is a little fragile. It works because # GlanceV2ClientAdapter.upload_image() calls # self.client.images.create() and self.client.images.get() once each # call so this way we always create() and get() the same mocked "image" self.app.client_manager.image.images.create.side_effect = self.images self.app.client_manager.image.images.get.side_effect = self.images self.cmd._read_image_file_pointer = mock.Mock(return_value=b'IMGDATA') self.cmd._check_file_exists = mock.Mock(return_value=True)
def setUp(self): super(TestUploadOnlyExisting, self).setUp() # Get the command object to test self.cmd = overcloud_image.UploadOvercloudImage(self.app, None) self.app.client_manager.image = mock.Mock() self.app.client_manager.image.version = 2.0 self.app.client_manager.image.images.create.return_value = (mock.Mock( id=10, name='imgname', properties={}, created_at='2015-07-31T14:37:22.000000')) self.cmd._check_file_exists = mock.Mock() self._file_handle = FakeHandle() self.cmd._read_image_file_pointer = mock.Mock( return_value=self._file_handle)