# Copyright 2016 Canonical Ltd
#
# 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 json
import sys
import unittest.mock as mock
from subprocess import CalledProcessError

from test_utils import CharmTestCase

# python-apt is not installed as part of test-requirements but is imported by
# some charmhelpers modules so create a fake import.
mock_apt = mock.MagicMock()
sys.modules['apt'] = mock_apt
mock_apt.apt_pkg = mock.MagicMock()

# mocking for rados
mock_rados = mock.MagicMock()
sys.modules['rados'] = mock_rados
mock_rados.connect = mock.MagicMock()

# mocking for psutil
mock_psutil = mock.MagicMock()
sys.modules['psutil'] = mock_psutil
mock_psutil.disks = mock.MagicMock()

with mock.patch('charmhelpers.contrib.hardening.harden.harden') as mock_dec:
    mock_dec.side_effect = (lambda *dargs, **dkwargs: lambda f:
                            lambda *args, **kwargs: f(*args, **kwargs))
    # import health actions as actions
    import ceph_ops as actions


class OpsTestCase(CharmTestCase):

    def setUp(self):
        super(OpsTestCase, self).setUp(
            actions, ["check_output",
                      "action_get",
                      "action_fail",
                      "open"])

    def test_get_health(self):
        actions.get_health()
        cmd = ['ceph', 'health']
        self.check_output.assert_called_once_with(cmd)

    def test_get_version_report_ok(self):
        def _call_rslt():
            with open('unit_tests/ceph_ls_node.json') as f:
                tree = f.read()
            yield tree.encode('UTF-8')
            while True:
                yield ('{'
                       '    "version": "16.2.7",'
                       '    "release": "pacific",'
                       '    "release_type": "stable"'
                       '}').encode('UTF-8')
        self.check_output.side_effect = _call_rslt()
        result = actions.get_versions_report()
        self.assertEqual('{\n'
                         '    "aware-bee": [\n'
                         '        "16.2.7"\n'
                         '    ],\n'
                         '    "grand-ape": [\n'
                         '        "16.2.7"\n'
                         '    ],\n'
                         '    "lucky-muskox": [\n'
                         '        "16.2.7"\n'
                         '    ],\n'
                         '    "juju-c8b0a2-3-lxd-0": [\n'
                         '        "16.2.7"\n'
                         '    ],\n'
                         '    "juju-c8b0a2-4-lxd-0": [\n'
                         '        "16.2.7"\n'
                         '    ],\n'
                         '    "juju-c8b0a2-5-lxd-0": [\n'
                         '        "16.2.7"\n'
                         '    ]\n'
                         '}', result)

    def test_get_version_report_fail(self):
        self.check_output.side_effect = CalledProcessError(1, 'ceph node ls')
        self.assertRaises(actions.CephReportError,
                          lambda: actions.get_versions_report())

    @mock.patch('socket.gethostname')
    def test_get_quorum_status(self, mock_hostname):
        mock_hostname.return_value = 'mockhost'
        cmd_out = (
            '{"election_epoch":4,"quorum":[0,1,2],"quorum_names":["juju-18410c'
            '-zaza-b7061340ed19-1","juju-18410c-zaza-b7061340ed19-0","juju-184'
            '10c-zaza-b7061340ed19-2"],"quorum_leader_name":"juju-18410c-zaza-'
            'b7061340ed19-1","quorum_age":97785,"monmap":{"epoch":1,"fsid":"4f'
            '9dd22a-1b71-11ec-a02a-fa163ee765d3","modified":"2021-09-22 06:51:'
            '10.975225","created":"2021-09-22 06:51:10.975225","min_mon_releas'
            'e":14,"min_mon_release_name":"nautilus","features":{"persistent":'
            '["kraken","luminous","mimic","osdmap-prune","nautilus"],"optional'
            '":[]},"mons":[{"rank":0,"name":"juju-18410c-zaza-b7061340ed19-1",'
            '"public_addrs":{"addrvec":[{"type":"v2","addr":"10.5.0.122:3300",'
            '"nonce":0},{"type":"v1","addr":"10.5.0.122:6789","nonce":0}]},"ad'
            'dr":"10.5.0.122:6789/0","public_addr":"10.5.0.122:6789/0"},{"rank'
            '":1,"name":"juju-18410c-zaza-b7061340ed19-0","public_addrs":{"add'
            'rvec":[{"type":"v2","addr":"10.5.2.239:3300","nonce":0},{"type":"'
            'v1","addr":"10.5.2.239:6789","nonce":0}]},"addr":"10.5.2.239:6789'
            '/0","public_addr":"10.5.2.239:6789/0"},{"rank":2,"name":"juju-184'
            '10c-zaza-b7061340ed19-2","public_addrs":{"addrvec":[{"type":"v2",'
            '"addr":"10.5.3.201:3300","nonce":0},{"type":"v1","addr":"10.5.3.2'
            '01:6789","nonce":0}]},"addr":"10.5.3.201:6789/0","public_addr":"1'
            '0.5.3.201:6789/0"}]}}'
        )
        self.check_output.return_value = cmd_out.encode()

        result = actions.get_quorum_status()
        self.assertDictEqual(result, {
            "election-epoch": 4,
            "quorum-age": 97785,
            "quorum-names": "juju-18410c-zaza-b7061340ed19-1, "
                            "juju-18410c-zaza-b7061340ed19-0, "
                            "juju-18410c-zaza-b7061340ed19-2",
            "quorum-leader-name": "juju-18410c-zaza-b7061340ed19-1",
        })

        result = actions.get_quorum_status(format_type="json")
        self.assertDictEqual(json.loads(result["message"]),
                             json.loads(cmd_out))