156 lines
4.9 KiB
Python
156 lines
4.9 KiB
Python
#
|
|
# Copyright (c) 2021, 2024 Wind River Systems, Inc.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
|
|
from dcmanagerclient.api.v1.sw_update_options_manager import SwUpdateOptions
|
|
from dcmanagerclient.commands.v1 import (
|
|
sw_update_options_manager as sw_update_options_cmd,
|
|
)
|
|
from dcmanagerclient.tests import base
|
|
|
|
FAKE_MANAGER = None
|
|
FAKE_CLOUD = "1"
|
|
FAKE_STORAGE_APPLY_TYPE = "parallel"
|
|
NEW_FAKE_STORAGE_APPLY_TYPE = "serial"
|
|
FAKE_WORKER_APPLY_TYPE = "serial"
|
|
FAKE_MAX_PARALLEL = "5"
|
|
FAKE_ALARM_RESTRICTIONS = "strict"
|
|
FAKE_DEFAULT_INSTANCE_ACTION = "migrate"
|
|
FAKE_CREATED_AT = None
|
|
FAKE_UPDATED_AT = None
|
|
|
|
SW_UPDATE_OPTION = SwUpdateOptions(
|
|
FAKE_MANAGER,
|
|
FAKE_CLOUD,
|
|
FAKE_STORAGE_APPLY_TYPE,
|
|
FAKE_WORKER_APPLY_TYPE,
|
|
FAKE_MAX_PARALLEL,
|
|
FAKE_ALARM_RESTRICTIONS,
|
|
FAKE_DEFAULT_INSTANCE_ACTION,
|
|
FAKE_CREATED_AT,
|
|
FAKE_UPDATED_AT,
|
|
)
|
|
|
|
UPDATED_SW_UPDATE_OPTION = SwUpdateOptions(
|
|
FAKE_MANAGER,
|
|
FAKE_CLOUD,
|
|
NEW_FAKE_STORAGE_APPLY_TYPE,
|
|
FAKE_WORKER_APPLY_TYPE,
|
|
FAKE_MAX_PARALLEL,
|
|
FAKE_ALARM_RESTRICTIONS,
|
|
FAKE_DEFAULT_INSTANCE_ACTION,
|
|
FAKE_CREATED_AT,
|
|
FAKE_UPDATED_AT,
|
|
)
|
|
|
|
|
|
class TestCLISWUpdateOptionsManagerV1(base.BaseCommandTest):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
# The client is the subcloud_group_manager
|
|
self.client = self.app.client_manager.sw_update_options_manager
|
|
|
|
def test_list_sw_update_options(self):
|
|
self.client.sw_update_options_list.return_value = [SW_UPDATE_OPTION]
|
|
actual_call = self.call(sw_update_options_cmd.ListSwUpdateOptions)
|
|
self.assertEqual(
|
|
[
|
|
(
|
|
FAKE_CLOUD,
|
|
FAKE_STORAGE_APPLY_TYPE,
|
|
FAKE_WORKER_APPLY_TYPE,
|
|
FAKE_MAX_PARALLEL,
|
|
FAKE_ALARM_RESTRICTIONS,
|
|
FAKE_DEFAULT_INSTANCE_ACTION,
|
|
)
|
|
],
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_list_sw_update_options_empty(self):
|
|
self.client.sw_update_options_list.return_value = []
|
|
actual_call = self.call(sw_update_options_cmd.ListSwUpdateOptions)
|
|
self.assertEqual(
|
|
(("<none>", "<none>", "<none>", "<none>", "<none>", "<none>"),),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_show_sw_update_options_default(self):
|
|
self.client.sw_update_options_detail.return_value = [SW_UPDATE_OPTION]
|
|
actual_call = self.call(sw_update_options_cmd.ShowSwUpdateOptions)
|
|
self.assertEqual(
|
|
(
|
|
FAKE_CLOUD,
|
|
FAKE_STORAGE_APPLY_TYPE,
|
|
FAKE_WORKER_APPLY_TYPE,
|
|
FAKE_MAX_PARALLEL,
|
|
FAKE_ALARM_RESTRICTIONS,
|
|
FAKE_DEFAULT_INSTANCE_ACTION,
|
|
FAKE_CREATED_AT,
|
|
FAKE_UPDATED_AT,
|
|
),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_show_sw_update_options_by_ref(self):
|
|
self.client.sw_update_options_detail.return_value = [SW_UPDATE_OPTION]
|
|
actual_call = self.call(
|
|
sw_update_options_cmd.ShowSwUpdateOptions, app_args=[FAKE_CLOUD]
|
|
)
|
|
self.assertEqual(
|
|
(
|
|
FAKE_CLOUD,
|
|
FAKE_STORAGE_APPLY_TYPE,
|
|
FAKE_WORKER_APPLY_TYPE,
|
|
FAKE_MAX_PARALLEL,
|
|
FAKE_ALARM_RESTRICTIONS,
|
|
FAKE_DEFAULT_INSTANCE_ACTION,
|
|
FAKE_CREATED_AT,
|
|
FAKE_UPDATED_AT,
|
|
),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_update_sw_update_options(self):
|
|
self.client.sw_update_options_update.return_value = [
|
|
UPDATED_SW_UPDATE_OPTION
|
|
]
|
|
actual_call = self.call(
|
|
sw_update_options_cmd.UpdateSwUpdateOptions,
|
|
app_args=[
|
|
"--storage-apply-type=" + NEW_FAKE_STORAGE_APPLY_TYPE,
|
|
"--worker-apply-type=" + FAKE_WORKER_APPLY_TYPE,
|
|
"--max-parallel-workers=" + FAKE_MAX_PARALLEL,
|
|
"--alarm-restriction-type=" + FAKE_ALARM_RESTRICTIONS,
|
|
"--default-instance-action=" + FAKE_DEFAULT_INSTANCE_ACTION,
|
|
],
|
|
)
|
|
self.assertEqual(
|
|
(
|
|
FAKE_CLOUD,
|
|
NEW_FAKE_STORAGE_APPLY_TYPE,
|
|
FAKE_WORKER_APPLY_TYPE,
|
|
FAKE_MAX_PARALLEL,
|
|
FAKE_ALARM_RESTRICTIONS,
|
|
FAKE_DEFAULT_INSTANCE_ACTION,
|
|
FAKE_CREATED_AT,
|
|
FAKE_UPDATED_AT,
|
|
),
|
|
actual_call[1],
|
|
)
|
|
|
|
def test_delete_sw_update_options_by_ref(self):
|
|
self.call(sw_update_options_cmd.DeleteSwUpdateOptions, app_args=[FAKE_CLOUD])
|
|
self.client.sw_update_options_delete.assert_called_once_with(FAKE_CLOUD)
|
|
|
|
def test_delete_sw_update_options_without_ref(self):
|
|
self.assertRaises(
|
|
SystemExit,
|
|
self.call,
|
|
sw_update_options_cmd.DeleteSwUpdateOptions,
|
|
app_args=[],
|
|
)
|