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

import os
import tempfile

import mock
import yaml

from dcmanagerclient.commands.v1 import phased_subcloud_deploy_manager as cmd
from dcmanagerclient.exceptions import DCManagerClientException
from dcmanagerclient.tests import base


@mock.patch("getpass.getpass", new=mock.Mock(return_value="testpassword"))
class TestCLIPhasedSubcloudDeployManagerV1(base.BaseCommandTest):

    def setUp(self):
        super().setUp()
        # The client is the subcloud_deploy_manager
        self.client = self.app.client_manager.phased_subcloud_deploy_manager

    def test_subcloud_deploy_create(self):
        self.client.subcloud_deploy_create.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile(
            mode="w"
        ) as bootstrap_file, tempfile.NamedTemporaryFile(
            mode="w"
        ) as config_file, tempfile.NamedTemporaryFile(
            mode="w"
        ) as install_file:

            bootstrap_file_path = os.path.abspath(bootstrap_file.name)
            config_file_path = os.path.abspath(config_file.name)
            install_file_path = os.path.abspath(install_file.name)

            actual_call = self.call(
                cmd.CreatePhasedSubcloudDeploy,
                app_args=[
                    "--bootstrap-address",
                    base.BOOTSTRAP_ADDRESS,
                    "--install-values",
                    install_file_path,
                    "--bootstrap-values",
                    bootstrap_file_path,
                    "--deploy-config",
                    config_file_path,
                    "--release",
                    base.SOFTWARE_VERSION,
                ],
            )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_subcloud_deploy_bootstrap(self):
        self.client.subcloud_deploy_bootstrap.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile(mode="w") as bootstrap_file:
            bootstrap_file_path = os.path.abspath(bootstrap_file.name)

            actual_call = self.call(
                cmd.BootstrapPhasedSubcloudDeploy,
                app_args=[
                    base.ID,
                    "--bootstrap-address",
                    base.BOOTSTRAP_ADDRESS,
                    "--bootstrap-values",
                    bootstrap_file_path,
                ],
            )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_install_subcloud(self):
        self.client.subcloud_deploy_install.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile(mode="w") as f:
            yaml.dump(base.FAKE_INSTALL_VALUES, f)
            file_path = os.path.abspath(f.name)
            actual_call = self.call(
                cmd.InstallPhasedSubcloudDeploy,
                app_args=[
                    base.NAME,
                    "--install-values",
                    file_path,
                ],
            )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_install_subcloud_with_release(self):
        self.client.subcloud_deploy_install.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile(mode="w") as f:
            yaml.dump(base.FAKE_INSTALL_VALUES, f)
            file_path = os.path.abspath(f.name)
            actual_call = self.call(
                cmd.InstallPhasedSubcloudDeploy,
                app_args=[
                    base.NAME,
                    "--install-values",
                    file_path,
                    "--release",
                    base.SOFTWARE_VERSION,
                ],
            )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_install_subcloud_without_install_values(self):
        self.client.subcloud_deploy_install.return_value = [base.SUBCLOUD_RESOURCE]

        actual_call = self.call(
            cmd.InstallPhasedSubcloudDeploy, app_args=[base.NAME]
        )

        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_install_file_does_not_exist(self):
        self.client.subcloud_deploy_install.return_value = [base.SUBCLOUD_RESOURCE]
        with tempfile.NamedTemporaryFile() as f:
            file_path = os.path.abspath(f.name)

        e = self.assertRaises(
            DCManagerClientException,
            self.call,
            cmd.InstallPhasedSubcloudDeploy,
            app_args=[base.NAME, "--install-values", file_path],
        )
        self.assertTrue("install-values does not exist" in str(e))

    def test_configure_subcloud(self):
        self.client.subcloud_deploy_config.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile() as f:
            file_path = os.path.abspath(f.name)
            actual_call = self.call(
                cmd.ConfigPhasedSubcloudDeploy,
                app_args=[base.NAME, "--deploy-config", file_path],
            )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_configure_file_does_not_exist(self):
        self.client.subcloud_deploy_config.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile() as f:
            file_path = os.path.abspath(f.name)

        e = self.assertRaises(
            DCManagerClientException,
            self.call,
            cmd.ConfigPhasedSubcloudDeploy,
            app_args=[base.NAME, "--deploy-config", file_path],
        )
        self.assertTrue("deploy-config file does not exist" in str(e))

    def test_complete_subcloud_deployment(self):
        self.client.subcloud_deploy_complete.return_value = [base.SUBCLOUD_RESOURCE]
        actual_call = self.call(
            cmd.CompletePhasedSubcloudDeploy, app_args=[base.NAME]
        )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_abort_subcloud(self):
        self.client.subcloud_deploy_abort.return_value = [base.SUBCLOUD_RESOURCE]
        actual_call = self.call(cmd.AbortPhasedSubcloudDeploy, app_args=[base.NAME])
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_subcloud_deploy_resume_all_parameters(self):
        self.client.subcloud_deploy_resume.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile(
            mode="w"
        ) as bootstrap_file, tempfile.NamedTemporaryFile(
            mode="w"
        ) as config_file, tempfile.NamedTemporaryFile(
            mode="w"
        ) as install_file:

            bootstrap_file_path = os.path.abspath(bootstrap_file.name)
            config_file_path = os.path.abspath(config_file.name)
            install_file_path = os.path.abspath(install_file.name)

            actual_call = self.call(
                cmd.PhasedSubcloudDeployResume,
                app_args=[
                    base.NAME,
                    "--bootstrap-address",
                    base.BOOTSTRAP_ADDRESS,
                    "--bootstrap-values",
                    bootstrap_file_path,
                    "--install-values",
                    install_file_path,
                    "--deploy-config",
                    config_file_path,
                    "--release",
                    base.SOFTWARE_VERSION,
                ],
            )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_subcloud_deploy_resume_missing_files(self):
        self.client.subcloud_deploy_resume.return_value = [base.SUBCLOUD_RESOURCE]

        with tempfile.NamedTemporaryFile(
            mode="w"
        ) as bootstrap_file, tempfile.NamedTemporaryFile(
            mode="w"
        ) as config_file, tempfile.NamedTemporaryFile(
            mode="w"
        ) as install_file:

            bootstrap_file_path = os.path.abspath(bootstrap_file.name)
            config_file_path = os.path.abspath(config_file.name)
            install_file_path = os.path.abspath(install_file.name)

        # Missing bootstrap values
        app_args_bootstrap = [
            base.NAME,
            "--bootstrap-address",
            base.BOOTSTRAP_ADDRESS,
            "--bootstrap-values",
            bootstrap_file_path,
        ]
        error_msg_bootstrap = "bootstrap-values does not exist"
        call_bootstrap = self.assertRaises(
            DCManagerClientException,
            self.call,
            cmd.PhasedSubcloudDeployResume,
            app_args=app_args_bootstrap,
        )
        self.assertTrue(error_msg_bootstrap in str(call_bootstrap))

        # Missing install values
        app_args_install = [base.NAME, "--install-values", install_file_path]
        error_msg_install = "install-values does not exist"
        call_install = self.assertRaises(
            DCManagerClientException,
            self.call,
            cmd.PhasedSubcloudDeployResume,
            app_args=app_args_install,
        )
        self.assertTrue(error_msg_install in str(call_install))

        # Missing deploy config values
        app_args_config = [base.NAME, "--deploy-config", config_file_path]
        error_msg_config = "deploy-config does not exist"
        call_config = self.assertRaises(
            DCManagerClientException,
            self.call,
            cmd.PhasedSubcloudDeployResume,
            app_args=app_args_config,
        )
        self.assertTrue(error_msg_config in str(call_config))

    def test_subcloud_deploy_resume_no_parameters(self):
        self.client.subcloud_deploy_resume.return_value = [base.SUBCLOUD_RESOURCE]

        actual_call = self.call(cmd.PhasedSubcloudDeployResume, app_args=[base.NAME])
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])

    def test_subcloud_deploy_resume_no_files_only_release(self):
        self.client.subcloud_deploy_resume.return_value = [base.SUBCLOUD_RESOURCE]

        actual_call = self.call(
            cmd.PhasedSubcloudDeployResume,
            app_args=[
                base.NAME,
                "--release",
                base.SOFTWARE_VERSION,
            ],
        )
        self.assertEqual(base.SUBCLOUD_FIELD_RESULT_LIST_WITH_PEERID, actual_call[1])