#   Copyright 2016 Red Hat, 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 mock

from tripleoclient.tests.v1.overcloud_deploy import fakes
from tripleoclient.v1 import overcloud_delete


class TestDeleteOvercloud(fakes.TestDeployOvercloud):

    def setUp(self):
        super(TestDeleteOvercloud, self).setUp()

        self.cmd = overcloud_delete.DeleteOvercloud(self.app, None)
        self.app.client_manager.workflow_engine = mock.Mock()
        self.workflow = self.app.client_manager.workflow_engine

    @mock.patch(
        'tripleoclient.workflows.stack_management.plan_undeploy',
        autospec=True)
    def test_plan_undeploy(self, mock_plan_undeploy):
        clients = self.app.client_manager
        orchestration_client = clients.orchestration

        stack = mock.Mock()
        stack.id = 12345
        stack.name = "foobar"
        orchestration_client.stacks.get.return_value = stack

        self.cmd._plan_undeploy(clients, 'overcloud')

        orchestration_client.stacks.get.assert_called_once_with('overcloud')
        mock_plan_undeploy.assert_called_once_with(
            clients, plan="foobar")

    @mock.patch(
        'tripleoclient.workflows.stack_management.base.start_workflow',
        autospec=True)
    def test_plan_undeploy_wf_params(self, mock_plan_undeploy_wf):
        clients = self.app.client_manager
        orchestration_client = clients.orchestration
        workflow_engine = clients.workflow_engine

        stack = mock.Mock()
        stack.id = 12345
        stack.name = "foobar"
        orchestration_client.stacks.get.return_value = stack

        self.cmd._plan_undeploy(clients, 'overcloud')

        orchestration_client.stacks.get.assert_called_once_with('overcloud')
        mock_plan_undeploy_wf.assert_called_once_with(
            workflow_engine,
            "tripleo.deployment.v1.undeploy_plan",
            workflow_input={"container": "foobar"})

    def test_plan_undeploy_no_stack(self):
        clients = self.app.client_manager
        orchestration_client = clients.orchestration
        type(orchestration_client.stacks.get).return_value = None
        self.cmd.log.warning = mock.MagicMock()

        self.cmd._plan_undeploy(clients, 'overcloud')

        orchestration_client.stacks.get.assert_called_once_with('overcloud')
        self.cmd.log.warning.assert_called_once_with(
            "No stack found ('overcloud'), skipping delete")

    @mock.patch(
        'tripleoclient.workflows.plan_management.delete_deployment_plan',
        autospec=True)
    def test_plan_delete(self, delete_deployment_plan_mock):
        self.cmd._plan_delete(self.workflow, 'overcloud')

        delete_deployment_plan_mock.assert_called_once_with(
            self.workflow,
            container='overcloud')