#   Copyright 2018 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 osc_lib.tests.utils import ParserException
from tripleoclient import constants
from tripleoclient import exceptions
from tripleoclient.tests.v1.overcloud_ceph_upgrade import fakes
from tripleoclient.v1 import overcloud_ceph_upgrade


class TestCephUpgrade(fakes.TestCephUpgrade):

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

        # Get the command object to test
        self.app_args = mock.Mock()
        self.app_args.verbose_level = 1
        self.cmd = overcloud_ceph_upgrade.CephUpgrade(self.app, self.app_args)
        uuid4_patcher = mock.patch('uuid.uuid4', return_value="UUID4")
        self.mock_uuid4 = uuid4_patcher.start()
        self.addCleanup(self.mock_uuid4.stop)

    @mock.patch('tripleoclient.utils.load_container_registry')
    @mock.patch('tripleoclient.utils.get_stack')
    @mock.patch('tripleoclient.workflows.package_update.update', autospec=True)
    @mock.patch(
        'tripleoclient.v1.overcloud_ceph_upgrade.DeployOvercloud.take_action')
    def test_ceph_upgrade(self, mock_deploy, mock_ceph_upgrade, mock_get_stack,
                          mock_load_registry):
        # get a fresh cmd so that the superclass mock takes effect
        cmd = overcloud_ceph_upgrade.CephUpgrade(self.app, self.app_args)

        mock_stack = mock.Mock()
        mock_stack.stack_name = 'mystack'
        mock_get_stack.return_value = mock_stack
        mock_load_registry.return_value = {'fake_container': 'fake_value'}

        argslist = ['--stack', 'mystack', '--templates',
                    '--container-registry-file', 'my-fake-registry.yaml']
        verifylist = [
            ('stack', 'mystack'),
            ('templates', constants.TRIPLEO_HEAT_TEMPLATES),
            ('container_registry_file', 'my-fake-registry.yaml')
        ]
        parsed_args = self.check_parser(self.cmd, argslist, verifylist)

        with mock.patch('os.path.exists') as mock_exists, \
                mock.patch('os.path.isfile') as mock_isfile:
            mock_exists.return_value = True
            mock_isfile.return_value = True
            cmd.take_action(parsed_args)
            mock_deploy.assert_called_once_with(parsed_args)
            mock_ceph_upgrade.assert_called_once_with(
                self.app.client_manager,
                container='mystack',
                container_registry={'fake_container': 'fake_value'},
                ceph_ansible_playbook='/usr/share/ceph-ansible'
                                      '/infrastructure-playbooks'
                                      '/rolling_update.yml',
            )

    @mock.patch('tripleoclient.utils.prepend_environment', autospec=True)
    @mock.patch('tripleoclient.workflows.package_update.update',
                autospec=True)
    @mock.patch('six.moves.builtins.open')
    @mock.patch('os.path.abspath')
    @mock.patch('yaml.load')
    @mock.patch('shutil.copytree', autospec=True)
    @mock.patch(
        'tripleoclient.v1.overcloud_ceph_upgrade.DeployOvercloud.take_action')
    def test_ceph_upgrade_failed(
        self, mock_deploy, mock_copy, mock_yaml, mock_abspath, mock_open,
            mock_ceph_upgrade, mock_prepend_env):
        mock_ceph_upgrade.side_effect = exceptions.DeploymentError()
        mock_abspath.return_value = '/home/fake/my-fake-registry.yaml'
        mock_yaml.return_value = {'fake_container': 'fake_value'}
        argslist = ['--stack', 'overcloud', '--templates',
                    '--container-registry-file', 'my-fake-registry.yaml']
        verifylist = [
            ('stack', 'overcloud'),
            ('templates', constants.TRIPLEO_HEAT_TEMPLATES),
            ('container_registry_file', 'my-fake-registry.yaml')
        ]
        parsed_args = self.check_parser(self.cmd, argslist, verifylist)

        # get a fresh cmd so that the superclass mock takes effect
        cmd = overcloud_ceph_upgrade.CephUpgrade(self.app, self.app_args)

        self.assertRaises(exceptions.DeploymentError,
                          cmd.take_action, parsed_args)

    @mock.patch('tripleoclient.workflows.package_update.update_ansible',
                autospec=True)
    @mock.patch('os.path.expanduser')
    @mock.patch('oslo_concurrency.processutils.execute')
    @mock.patch('six.moves.builtins.open')
    def test_upgrade_no_nodes_or_roles(self, mock_open, mock_execute,
                                       mock_expanduser, upgrade_ansible):
        mock_expanduser.return_value = '/home/fake/'
        argslist = ["--nodes", "controller-1", "--roles", "foo"]
        verifylist = []
        self.assertRaises(ParserException, lambda: self.check_parser(
            self.cmd, argslist, verifylist))