346 lines
12 KiB
Python
346 lines
12 KiB
Python
# Copyright (c) 2020-2024 Wind River Systems, Inc.
|
|
# 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 os
|
|
import tempfile
|
|
|
|
import mock
|
|
|
|
from dcmanagerclient.api.v1 import subcloud_deploy_manager as sdm
|
|
from dcmanagerclient.commands.v1 import subcloud_deploy_manager as \
|
|
subcloud_deploy_cmd
|
|
from dcmanagerclient.exceptions import DCManagerClientException
|
|
from dcmanagerclient.tests import base
|
|
|
|
DEPLOY_PLAYBOOK = "deployment-manager-playbook.yaml"
|
|
DEPLOY_OVERRIDES = "deployment-manager-overrides-subcloud.yaml"
|
|
DEPLOY_CHART = "deployment-manager.tgz"
|
|
DEPLOY_PRESTAGE_IMAGES = "prebuilt-images.lst"
|
|
|
|
SUBCLOUD_DEPLOY_DICT = {
|
|
"DEPLOY_PLAYBOOK": DEPLOY_PLAYBOOK,
|
|
"DEPLOY_OVERRIDES": DEPLOY_OVERRIDES,
|
|
"DEPLOY_CHART": DEPLOY_CHART,
|
|
"DEPLOY_PRESTAGE_IMAGES": DEPLOY_PRESTAGE_IMAGES,
|
|
"SOFTWARE_VERSION": base.SOFTWARE_VERSION,
|
|
}
|
|
|
|
SUBCLOUD_DEPLOY_ALL = sdm.SubcloudDeploy(
|
|
deploy_playbook=SUBCLOUD_DEPLOY_DICT["DEPLOY_PLAYBOOK"],
|
|
deploy_overrides=SUBCLOUD_DEPLOY_DICT["DEPLOY_OVERRIDES"],
|
|
deploy_chart=SUBCLOUD_DEPLOY_DICT["DEPLOY_CHART"],
|
|
prestage_images=SUBCLOUD_DEPLOY_DICT["DEPLOY_PRESTAGE_IMAGES"],
|
|
software_version=SUBCLOUD_DEPLOY_DICT["SOFTWARE_VERSION"],
|
|
)
|
|
|
|
SUBCLOUD_DEPLOY_PRESTAGE = sdm.SubcloudDeploy(
|
|
prestage_images=SUBCLOUD_DEPLOY_DICT["DEPLOY_PRESTAGE_IMAGES"],
|
|
software_version=SUBCLOUD_DEPLOY_DICT["SOFTWARE_VERSION"],
|
|
)
|
|
|
|
SUBCLOUD_DEPLOY_NO_PRESTAGE = sdm.SubcloudDeploy(
|
|
deploy_playbook=SUBCLOUD_DEPLOY_DICT["DEPLOY_PLAYBOOK"],
|
|
deploy_overrides=SUBCLOUD_DEPLOY_DICT["DEPLOY_OVERRIDES"],
|
|
deploy_chart=SUBCLOUD_DEPLOY_DICT["DEPLOY_CHART"],
|
|
software_version=SUBCLOUD_DEPLOY_DICT["SOFTWARE_VERSION"],
|
|
)
|
|
|
|
SUBCLOUD_DEPLOY_NO_PLAYBOOK = sdm.SubcloudDeploy(
|
|
deploy_overrides=SUBCLOUD_DEPLOY_DICT["DEPLOY_OVERRIDES"],
|
|
deploy_chart=SUBCLOUD_DEPLOY_DICT["DEPLOY_CHART"],
|
|
prestage_images=SUBCLOUD_DEPLOY_DICT["DEPLOY_PRESTAGE_IMAGES"],
|
|
software_version=SUBCLOUD_DEPLOY_DICT["SOFTWARE_VERSION"],
|
|
)
|
|
|
|
SUBCLOUD_DEPLOY_NO_PLAYBOOK_OVERRIDES = sdm.SubcloudDeploy(
|
|
deploy_chart=SUBCLOUD_DEPLOY_DICT["DEPLOY_CHART"],
|
|
prestage_images=SUBCLOUD_DEPLOY_DICT["DEPLOY_PRESTAGE_IMAGES"],
|
|
software_version=SUBCLOUD_DEPLOY_DICT["SOFTWARE_VERSION"],
|
|
)
|
|
|
|
SUBCLOUD_DEPLOY_NO_OVERRIDES_CHART = sdm.SubcloudDeploy(
|
|
deploy_playbook=SUBCLOUD_DEPLOY_DICT["DEPLOY_PLAYBOOK"],
|
|
prestage_images=SUBCLOUD_DEPLOY_DICT["DEPLOY_PRESTAGE_IMAGES"],
|
|
software_version=SUBCLOUD_DEPLOY_DICT["SOFTWARE_VERSION"],
|
|
)
|
|
|
|
|
|
class TestCLISubcloudDeployManagerV1(base.BaseCommandTest):
|
|
|
|
def setUp(self):
|
|
super(TestCLISubcloudDeployManagerV1, self).setUp()
|
|
# The client is the subcloud_deploy_manager
|
|
self.client = self.app.client_manager.subcloud_deploy_manager
|
|
|
|
def test_subcloud_deploy_show(self):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_show.return_value = [
|
|
SUBCLOUD_DEPLOY_ALL
|
|
]
|
|
|
|
# Without "--release" parameter
|
|
actual_call1 = self.call(subcloud_deploy_cmd.SubcloudDeployShow)
|
|
|
|
self.assertEqual(
|
|
(
|
|
DEPLOY_PLAYBOOK,
|
|
DEPLOY_OVERRIDES,
|
|
DEPLOY_CHART,
|
|
DEPLOY_PRESTAGE_IMAGES,
|
|
base.SOFTWARE_VERSION,
|
|
),
|
|
actual_call1[1],
|
|
)
|
|
|
|
# With "--release" parameter
|
|
actual_call2 = self.call(
|
|
subcloud_deploy_cmd.SubcloudDeployShow,
|
|
app_args=["--release", base.SOFTWARE_VERSION],
|
|
)
|
|
|
|
self.assertEqual(
|
|
(
|
|
DEPLOY_PLAYBOOK,
|
|
DEPLOY_OVERRIDES,
|
|
DEPLOY_CHART,
|
|
DEPLOY_PRESTAGE_IMAGES,
|
|
base.SOFTWARE_VERSION,
|
|
),
|
|
actual_call2[1],
|
|
)
|
|
|
|
def test_subcloud_deploy_upload_all(self):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_upload.return_value = [
|
|
SUBCLOUD_DEPLOY_ALL
|
|
]
|
|
|
|
with tempfile.NamedTemporaryFile() as f1, \
|
|
tempfile.NamedTemporaryFile() as f2, \
|
|
tempfile.NamedTemporaryFile() as f3, \
|
|
tempfile.NamedTemporaryFile() as f4:
|
|
file_path_1 = os.path.abspath(f1.name)
|
|
file_path_2 = os.path.abspath(f2.name)
|
|
file_path_3 = os.path.abspath(f3.name)
|
|
file_path_4 = os.path.abspath(f4.name)
|
|
actual_call = self.call(
|
|
subcloud_deploy_cmd.SubcloudDeployUpload,
|
|
app_args=[
|
|
"--deploy-playbook",
|
|
file_path_1,
|
|
"--deploy-overrides",
|
|
file_path_2,
|
|
"--deploy-chart",
|
|
file_path_3,
|
|
"--prestage-images",
|
|
file_path_4,
|
|
],
|
|
)
|
|
|
|
self.assertEqual(
|
|
(
|
|
DEPLOY_PLAYBOOK,
|
|
DEPLOY_OVERRIDES,
|
|
DEPLOY_CHART,
|
|
DEPLOY_PRESTAGE_IMAGES,
|
|
base.SOFTWARE_VERSION,
|
|
),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_subcloud_deploy_upload_no_prestage(self):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_upload.return_value = [
|
|
SUBCLOUD_DEPLOY_NO_PRESTAGE
|
|
]
|
|
|
|
with tempfile.NamedTemporaryFile() as f1, \
|
|
tempfile.NamedTemporaryFile() as f2, \
|
|
tempfile.NamedTemporaryFile() as f3:
|
|
file_path_1 = os.path.abspath(f1.name)
|
|
file_path_2 = os.path.abspath(f2.name)
|
|
file_path_3 = os.path.abspath(f3.name)
|
|
actual_call = self.call(
|
|
subcloud_deploy_cmd.SubcloudDeployUpload,
|
|
app_args=[
|
|
"--deploy-playbook",
|
|
file_path_1,
|
|
"--deploy-overrides",
|
|
file_path_2,
|
|
"--deploy-chart",
|
|
file_path_3,
|
|
],
|
|
)
|
|
|
|
self.assertEqual(
|
|
(
|
|
DEPLOY_PLAYBOOK,
|
|
DEPLOY_OVERRIDES,
|
|
DEPLOY_CHART,
|
|
None,
|
|
base.SOFTWARE_VERSION,
|
|
),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_subcloud_deploy_upload_prestage(self):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_upload.return_value = [
|
|
SUBCLOUD_DEPLOY_PRESTAGE
|
|
]
|
|
|
|
with tempfile.NamedTemporaryFile() as f1:
|
|
file_path_1 = os.path.abspath(f1.name)
|
|
actual_call = self.call(
|
|
subcloud_deploy_cmd.SubcloudDeployUpload,
|
|
app_args=["--prestage-images", file_path_1],
|
|
)
|
|
self.assertEqual(
|
|
(None, None, None, DEPLOY_PRESTAGE_IMAGES, base.SOFTWARE_VERSION),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_subcloud_deploy_upload_no_playbook(self):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_upload.return_value = [
|
|
SUBCLOUD_DEPLOY_NO_PLAYBOOK
|
|
]
|
|
|
|
with tempfile.NamedTemporaryFile() as f1, \
|
|
tempfile.NamedTemporaryFile() as f2, \
|
|
tempfile.NamedTemporaryFile() as f3:
|
|
file_path_1 = os.path.abspath(f1.name)
|
|
file_path_2 = os.path.abspath(f2.name)
|
|
file_path_3 = os.path.abspath(f3.name)
|
|
actual_call = self.call(
|
|
subcloud_deploy_cmd.SubcloudDeployUpload,
|
|
app_args=[
|
|
"--deploy-overrides",
|
|
file_path_1,
|
|
"--deploy-chart",
|
|
file_path_2,
|
|
"--prestage-images",
|
|
file_path_3,
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
(
|
|
None,
|
|
DEPLOY_OVERRIDES,
|
|
DEPLOY_CHART,
|
|
DEPLOY_PRESTAGE_IMAGES,
|
|
base.SOFTWARE_VERSION,
|
|
),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_subcloud_deploy_upload_no_playbook_overrides(self):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_upload.return_value = [
|
|
SUBCLOUD_DEPLOY_NO_PLAYBOOK_OVERRIDES
|
|
]
|
|
|
|
with tempfile.NamedTemporaryFile() as f1, \
|
|
tempfile.NamedTemporaryFile() as f2:
|
|
file_path_1 = os.path.abspath(f1.name)
|
|
file_path_2 = os.path.abspath(f2.name)
|
|
actual_call = self.call(
|
|
subcloud_deploy_cmd.SubcloudDeployUpload,
|
|
app_args=[
|
|
"--deploy-chart",
|
|
file_path_1,
|
|
"--prestage-images",
|
|
file_path_2,
|
|
],
|
|
)
|
|
self.assertEqual((
|
|
None, None,
|
|
DEPLOY_CHART,
|
|
DEPLOY_PRESTAGE_IMAGES,
|
|
base.SOFTWARE_VERSION
|
|
), actual_call[1],)
|
|
|
|
def test_subcloud_deploy_upload_no_overrides_chart(self):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_upload.return_value = [
|
|
SUBCLOUD_DEPLOY_NO_OVERRIDES_CHART
|
|
]
|
|
|
|
with tempfile.NamedTemporaryFile() as f1, \
|
|
tempfile.NamedTemporaryFile() as f2:
|
|
file_path_1 = os.path.abspath(f1.name)
|
|
file_path_2 = os.path.abspath(f2.name)
|
|
actual_call = self.call(
|
|
subcloud_deploy_cmd.SubcloudDeployUpload,
|
|
app_args=[
|
|
"--deploy-playbook",
|
|
file_path_1,
|
|
"--prestage-images",
|
|
file_path_2,
|
|
"--release",
|
|
base.SOFTWARE_VERSION,
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
(
|
|
DEPLOY_PLAYBOOK,
|
|
None,
|
|
None,
|
|
DEPLOY_PRESTAGE_IMAGES,
|
|
base.SOFTWARE_VERSION,
|
|
),
|
|
actual_call[1],
|
|
)
|
|
|
|
@mock.patch("builtins.print")
|
|
def test_subcloud_deploy_upload_invalid_path(self, mock_print):
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_upload.return_value = [
|
|
SUBCLOUD_DEPLOY_NO_PRESTAGE
|
|
]
|
|
mock_print.return_value = mock.ANY
|
|
file_path_1 = "not_a_valid_path"
|
|
with tempfile.NamedTemporaryFile() as f2, \
|
|
tempfile.NamedTemporaryFile() as f3:
|
|
file_path_2 = os.path.abspath(f2.name)
|
|
file_path_3 = os.path.abspath(f3.name)
|
|
|
|
e = self.assertRaises(
|
|
DCManagerClientException,
|
|
self.call,
|
|
subcloud_deploy_cmd.SubcloudDeployUpload,
|
|
app_args=[
|
|
"--deploy-playbook",
|
|
file_path_1,
|
|
"--deploy-overrides",
|
|
file_path_2,
|
|
"--deploy-chart",
|
|
file_path_3,
|
|
],
|
|
)
|
|
|
|
self.assertTrue(
|
|
"deploy_playbook file does not exist: not_a_valid_path" in str(e)
|
|
)
|
|
|
|
def test_subcloud_deploy_delete_with_release(self):
|
|
|
|
release_version = base.SOFTWARE_VERSION
|
|
data = {"prestage_images": "False", "deployment_files": "False"}
|
|
app_args = ["--release", release_version]
|
|
|
|
self.call(subcloud_deploy_cmd.SubcloudDeployDelete, app_args=app_args)
|
|
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_delete.\
|
|
assert_called_once_with(release_version, data=data)
|
|
|
|
def test_subcloud_deploy_delete_without_release(self):
|
|
|
|
self.call(subcloud_deploy_cmd.SubcloudDeployDelete)
|
|
data = {"prestage_images": "False", "deployment_files": "False"}
|
|
self.client.subcloud_deploy_manager.subcloud_deploy_delete.\
|
|
assert_called_once_with(None, data=data)
|