aa48f8223a
When deploy OpenStack on XenServer, we need install some XAPI plugins to dom0. This commit is to add utils for this purpose. If the os-xenapi version is different from the os-xenapi which contains the utils. Users can specify the version to the utils. So that the utils will download the right version of os-xenapi and copy plugins from there. Otherwise the utils will by default copy plugins from current installed packages. Change-Id: I269a444b952f63fd73b3825b23dc95d6e825ce8f
101 lines
4.3 KiB
Python
101 lines
4.3 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import mock
|
|
import os
|
|
import shutil
|
|
import tempfile
|
|
|
|
from os_xenapi.tests import base
|
|
from os_xenapi.utils import common_function
|
|
from os_xenapi.utils import xapi_plugin
|
|
|
|
|
|
class XenapiPluginTestCase(base.TestCase):
|
|
@mock.patch.object(tempfile, 'mkdtemp',
|
|
return_value='/tmp/tmp1VyeJn')
|
|
@mock.patch.object(common_function, 'execute')
|
|
def test_get_os_xenapi_dir_with_version(self, mock_exec, mock_tmp):
|
|
VERSION = '0.3.1'
|
|
is_tmp, dir = xapi_plugin.get_os_xenapi_dir(VERSION)
|
|
|
|
self.assertEqual(2, mock_exec.call_count)
|
|
self.assertEqual(dir, '/tmp/tmp1VyeJn')
|
|
self.assertEqual(is_tmp, True)
|
|
|
|
@mock.patch.object(tempfile, 'mkdtemp')
|
|
@mock.patch.object(common_function, 'execute')
|
|
def test_get_os_xenapi_dir_no_version(self, mock_exec, mock_tmp):
|
|
fake_loc = '/fake/install/loc'
|
|
mock_exec.return_value = 'Location: %s\n' % fake_loc
|
|
|
|
is_tmp, dir = xapi_plugin.get_os_xenapi_dir()
|
|
|
|
self.assertEqual(dir, fake_loc)
|
|
self.assertEqual(is_tmp, False)
|
|
mock_exec.assert_called_with('pip', 'show',
|
|
xapi_plugin.OS_XENAPI_PKG)
|
|
mock_tmp.assert_not_called()
|
|
|
|
@mock.patch.object(xapi_plugin, 'get_os_xenapi_dir')
|
|
@mock.patch.object(os, 'listdir')
|
|
@mock.patch.object(shutil, 'rmtree')
|
|
def test_install_plugins_to_dom0_no_version(self, mock_rm, mock_dir,
|
|
mock_get):
|
|
FAKE_PKG_PATH = '/fake/pkg/path'
|
|
ssh_client = mock.Mock()
|
|
mock_get.return_value = (False, FAKE_PKG_PATH)
|
|
mock_dir.return_value = ['file1', 'file2']
|
|
|
|
xapi_plugin.install_plugins_to_dom0(ssh_client)
|
|
|
|
dom0_path = xapi_plugin.DOM0_PLUGIN_PATH
|
|
local_path = '%s/%s' % (FAKE_PKG_PATH,
|
|
xapi_plugin.PKG_PLUGIN_PATH)
|
|
scp_expect = [mock.call('%s/file1' % local_path,
|
|
'%s/file1' % dom0_path),
|
|
mock.call('%s/file2' % local_path,
|
|
'%s/file2' % dom0_path)]
|
|
ssh_expect = [mock.call('chmod +x %s/file1' % dom0_path),
|
|
mock.call('chmod +x %s/file2' % dom0_path)]
|
|
self.assertEqual(scp_expect, ssh_client.scp.call_args_list)
|
|
self.assertEqual(ssh_expect, ssh_client.ssh.call_args_list)
|
|
# Shouldn't invoke mock_rm to remove the package dir.
|
|
mock_rm.assert_not_called()
|
|
|
|
@mock.patch.object(xapi_plugin, 'get_os_xenapi_dir')
|
|
@mock.patch.object(os, 'listdir')
|
|
@mock.patch.object(shutil, 'rmtree')
|
|
def test_install_plugins_to_dom0_with_version(self, mock_rm, mock_dir,
|
|
mock_get):
|
|
VERSION = '0.3.1'
|
|
FAKE_PKG_PATH = '/tmp/tmp1VyeJn'
|
|
ssh_client = mock.Mock()
|
|
mock_get.return_value = (True, FAKE_PKG_PATH)
|
|
mock_dir.return_value = ['file1', 'file2']
|
|
|
|
xapi_plugin.install_plugins_to_dom0(ssh_client, VERSION)
|
|
|
|
dom0_path = xapi_plugin.DOM0_PLUGIN_PATH
|
|
local_path = '%s/%s' % (FAKE_PKG_PATH,
|
|
xapi_plugin.PKG_PLUGIN_PATH)
|
|
scp_expect = [mock.call('%s/file1' % local_path,
|
|
'%s/file1' % dom0_path),
|
|
mock.call('%s/file2' % local_path,
|
|
'%s/file2' % dom0_path)]
|
|
ssh_expect = [mock.call('chmod +x %s/file1' % dom0_path),
|
|
mock.call('chmod +x %s/file2' % dom0_path)]
|
|
self.assertEqual(scp_expect, ssh_client.scp.call_args_list)
|
|
self.assertEqual(ssh_expect, ssh_client.ssh.call_args_list)
|
|
# Should invoke mock_rm to remove the package dir.
|
|
mock_rm.assert_called_with(FAKE_PKG_PATH)
|