# Copyright 2017, Kevin Carter <kevin@cloudnull.com>
#
# 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.
"""Tests for the KVM plugin."""

import json

from click.testing import CliRunner

from monitorstack.cli import cli
from monitorstack.utils.os_utils import OpenStack as Ost


def _runner(module):
    runner = CliRunner()
    result = runner.invoke(cli, [
        '-f', 'json',
        module,
        '--config-file', 'tests/files/test-openstack.ini',
    ])
    return json.loads(result.output)


class MockProject(object):
    """Mock class for OpenStack class."""

    def __init__(self):
        """Mock init."""
        self.id = 'testing'
        self.name = 'testing'


class TestOs(object):
    """Tests for the os_vm.* monitors."""

    def test_os_vm_quota_cores_success(self, monkeypatch):
        """Ensure the run() method works."""
        def mock_get_projects(arg1):
            projects = MockProject()
            return [projects]

        def mock_get_compute_limits(self, project_id, interface):
            return {'quota_set': {'cores': 10}}

        monkeypatch.setattr(Ost, 'get_projects', mock_get_projects)
        monkeypatch.setattr(Ost, 'get_compute_limits', mock_get_compute_limits)

        result = _runner('os_vm_quota_cores')
        from pprint import pprint
        pprint(result)
        assert result['measurement_name'] == 'os_vm_quota_cores'
        assert result['meta'] == {'quotas': 'cores'}

    def test_os_vm_quota_cores_failure(self):
        """Ensure the run() method works."""
        result = _runner('os_vm_quota_cores')
        assert result['measurement_name'] == 'os_vm_quota_cores'
        assert result['meta'] == {'quotas': 'cores'}

    def test_os_vm_quota_instances(self):
        """Ensure the run() method works."""
        pass

    def test_os_vm_quota_ram(self):
        """Ensure the run() method works."""
        pass

    def test_os_vm_used_cores(self):
        """Ensure the run() method works."""
        pass

    def test_os_vm_used_disk(self):
        """Ensure the run() method works."""
        pass

    def test_os_vm_used_instances(self):
        """Ensure the run() method works."""
        pass

    def test_os_vm_used_ram(self):
        """Ensure the run() method works."""
        pass