def setUp(self): super(TestOvercloudImageBuild, self).setUp() def _force_builder(dummy): builder = overcloud_image.DibImageBuilder() builder._disk_image_create = mock.Mock() self.mock_disk_image_create = builder._disk_image_create builder._ramdisk_image_create = mock.Mock() self.mock_ramdisk_image_create = builder._ramdisk_image_create return builder # Get the command object to test self.cmd = overcloud_image.BuildOvercloudImage(self.app, None) self.cmd._create_builder = _force_builder
def setUp(self): super(TestOvercloudImageBuild, self).setUp() # Get the command object to test self.cmd = overcloud_image.BuildOvercloudImage(self.app, None)