#
# Copyright (c) 2020-2021 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#

from dcmanagerclient.commands.v1 import sw_upgrade_manager as cli_cmd
from dcmanagerclient.tests import base
from dcmanagerclient.tests.v1.mixins import UpdateStrategyMixin


class TestSwUpgradeStrategy(UpdateStrategyMixin, base.BaseCommandTest):

    def setUp(self):
        super(TestSwUpgradeStrategy, self).setUp()
        self.sw_update_manager = \
            self.app.client_manager.sw_upgrade_manager.sw_upgrade_manager
        self.create_command = cli_cmd.CreateSwUpgradeStrategy
        self.show_command = cli_cmd.ShowSwUpgradeStrategy
        self.delete_command = cli_cmd.DeleteSwUpgradeStrategy
        self.apply_command = cli_cmd.ApplySwUpgradeStrategy
        self.abort_command = cli_cmd.AbortSwUpgradeStrategy