def test__cache_tftp_images_master_path(self): temp_dir = tempfile.mkdtemp() self.config(tftp_root=temp_dir, group='pxe') self.config(tftp_master_path=os.path.join(temp_dir, 'tftp_master_path'), group='pxe') image_info = {'deploy_kernel': ['deploy_kernel', os.path.join(temp_dir, self.node.uuid, 'deploy_kernel')]} fileutils.ensure_tree(CONF.pxe.tftp_master_path) fd, tmp_master_image = tempfile.mkstemp(dir=CONF.pxe.tftp_master_path) with mock.patch.object(images, 'fetch_to_raw') as fetch_to_raw_mock: with mock.patch.object(tempfile, 'mkstemp') as mkstemp_mock: fetch_to_raw_mock.return_value = None mkstemp_mock.return_value = (fd, tmp_master_image) pxe._cache_tftp_images(None, self.node, image_info) fetch_to_raw_mock.assert_called_once_with(None, 'deploy_kernel', tmp_master_image, None) mkstemp_mock.assert_called_once_with( dir=CONF.pxe.tftp_master_path)
def test__cache_tftp_images_master_path(self): temp_dir = tempfile.mkdtemp() self.config(tftp_root=temp_dir, group='pxe') self.config(tftp_master_path=os.path.join(temp_dir, 'tftp_master_path'), group='pxe') image_info = { 'deploy_kernel': [ 'deploy_kernel', os.path.join(temp_dir, self.node.uuid, 'deploy_kernel') ] } fileutils.ensure_tree(CONF.pxe.tftp_master_path) fd, tmp_master_image = tempfile.mkstemp(dir=CONF.pxe.tftp_master_path) with mock.patch.object(images, 'fetch_to_raw') as fetch_to_raw_mock: with mock.patch.object(tempfile, 'mkstemp') as mkstemp_mock: fetch_to_raw_mock.return_value = None mkstemp_mock.return_value = (fd, tmp_master_image) pxe._cache_tftp_images(None, self.node, image_info) fetch_to_raw_mock.assert_called_once_with( None, 'deploy_kernel', tmp_master_image, None) mkstemp_mock.assert_called_once_with( dir=CONF.pxe.tftp_master_path)
def test__cache_tftp_images_master_path(self, mock_fetch_image): temp_dir = tempfile.mkdtemp() self.config(tftp_root=temp_dir, group="pxe") self.config(tftp_master_path=os.path.join(temp_dir, "tftp_master_path"), group="pxe") image_path = os.path.join(temp_dir, self.node.uuid, "deploy_kernel") image_info = {"deploy_kernel": ("deploy_kernel", image_path)} fileutils.ensure_tree(CONF.pxe.tftp_master_path) pxe._cache_tftp_images(None, self.node, image_info) mock_fetch_image.assert_called_once_with(None, mock.ANY, [("deploy_kernel", image_path)])
def test__cache_tftp_images_no_master_path(self): temp_dir = tempfile.mkdtemp() CONF.set_default('tftp_root', temp_dir, group='pxe') CONF.set_default('tftp_master_path', None, group='pxe') image_info = {'deploy_kernel': ['deploy_kernel', os.path.join(temp_dir, 'instance_uuid_123/deploy_kernel')]} self.mox.StubOutWithMock(images, 'fetch_to_raw') images.fetch_to_raw(None, 'deploy_kernel', os.path.join(temp_dir, 'instance_uuid_123/deploy_kernel'), None).AndReturn(None) self.mox.ReplayAll() pxe._cache_tftp_images(None, self.node, image_info) self.mox.VerifyAll()
def test__cache_tftp_images_master_path(self, mock_fetch_image): temp_dir = tempfile.mkdtemp() self.config(tftp_root=temp_dir, group='pxe') self.config(tftp_master_path=os.path.join(temp_dir, 'tftp_master_path'), group='pxe') image_path = os.path.join(temp_dir, self.node.uuid, 'deploy_kernel') image_info = {'deploy_kernel': ['deploy_kernel', image_path]} fileutils.ensure_tree(CONF.pxe.tftp_master_path) pxe._cache_tftp_images(None, self.node, image_info) mock_fetch_image.assert_called_once_with('deploy_kernel', image_path, ctx=None)
def test__cache_tftp_images_no_master_path(self): temp_dir = tempfile.mkdtemp() CONF.set_default('tftp_root', temp_dir, group='pxe') CONF.set_default('tftp_master_path', None, group='pxe') image_info = {'deploy_kernel': ['deploy_kernel', os.path.join(temp_dir, 'instance_uuid_123/deploy_kernel')]} with mock.patch.object(images, 'fetch_to_raw') as fetch_to_raw_mock: fetch_to_raw_mock.return_value = None pxe._cache_tftp_images(None, self.node, image_info) fetch_to_raw_mock.assert_called_once_with(None, 'deploy_kernel', os.path.join(temp_dir, 'instance_uuid_123/deploy_kernel'), None)
def test__cache_tftp_images_no_master_path(self): temp_dir = tempfile.mkdtemp() self.config(tftp_root=temp_dir, group='pxe') self.config(tftp_master_path=None, group='pxe') image_info = {'deploy_kernel': ['deploy_kernel', os.path.join(temp_dir, self.node.uuid, 'deploy_kernel')]} with mock.patch.object(images, 'fetch_to_raw') as fetch_to_raw_mock: fetch_to_raw_mock.return_value = None pxe._cache_tftp_images(None, self.node, image_info) fetch_to_raw_mock.assert_called_once_with(None, 'deploy_kernel', os.path.join(temp_dir, self.node.uuid, 'deploy_kernel'), None)
def test__cache_tftp_images_master_path(self, mock_fetch_image): temp_dir = tempfile.mkdtemp() self.config(tftp_root=temp_dir, group='pxe') self.config(tftp_master_path=os.path.join(temp_dir, 'tftp_master_path'), group='pxe') image_path = os.path.join(temp_dir, self.node.uuid, 'deploy_kernel') image_info = {'deploy_kernel': ('deploy_kernel', image_path)} fileutils.ensure_tree(CONF.pxe.tftp_master_path) pxe._cache_tftp_images(None, self.node, image_info) mock_fetch_image.assert_called_once_with(None, mock.ANY, [('deploy_kernel', image_path)])
def test__cache_tftp_images_no_master_path(self): temp_dir = tempfile.mkdtemp() self.config(tftp_root=temp_dir, group='pxe') self.config(tftp_master_path=None, group='pxe') image_info = { 'deploy_kernel': [ 'deploy_kernel', os.path.join(temp_dir, self.node.uuid, 'deploy_kernel') ] } with mock.patch.object(images, 'fetch_to_raw') as fetch_to_raw_mock: fetch_to_raw_mock.return_value = None pxe._cache_tftp_images(None, self.node, image_info) fetch_to_raw_mock.assert_called_once_with( None, 'deploy_kernel', os.path.join(temp_dir, self.node.uuid, 'deploy_kernel'), None)
def test__cache_tftp_images_master_path(self): temp_dir = tempfile.mkdtemp() CONF.set_default('tftp_root', temp_dir, group='pxe') CONF.set_default('tftp_master_path', os.path.join(temp_dir, 'tftp_master_path'), group='pxe') image_info = {'deploy_kernel': ['deploy_kernel', temp_dir + '/instance_uuid_123/deploy_kernel']} fileutils.ensure_tree(CONF.pxe.tftp_master_path) fd, tmp_master_image = tempfile.mkstemp(dir=CONF.pxe.tftp_master_path) self.mox.StubOutWithMock(images, 'fetch_to_raw') self.mox.StubOutWithMock(tempfile, 'mkstemp') tempfile.mkstemp(dir=CONF.pxe.tftp_master_path).\ AndReturn((fd, tmp_master_image)) images.fetch_to_raw(None, 'deploy_kernel', tmp_master_image, None).\ AndReturn(None) self.mox.ReplayAll() pxe._cache_tftp_images(None, self.node, image_info) self.mox.VerifyAll()