# Copyright 2015 - Savoir-Faire Linux 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 requests_mock

from surveilclient.tests.v2_0 import clienttest


class TestMetrics(clienttest.ClientTest):

    def test_list_host_metrics(self):
        with requests_mock.mock() as m:
            m.post("http://localhost:5311/v2/status/"
                   "hosts/localhost/metrics/load1",
                   text='[{"min": "2", "warning": "15", "value": "3"},'
                        '{"min": "5", "warning": "200", "value": "150"}]'
                   )

            metrics = self.client.status.hosts.metrics.list(
                'localhost',
                'load1'
            )

            self.assertEqual(
                metrics,
                [{"min": "2", "warning": "15", "value": "3"},
                 {"min": "5", "warning": "200", "value": "150"}]
            )

    def test_list_host_metrics_service(self):
        with requests_mock.mock() as m:
            m.post("http://localhost:5311/v2/status/hosts/localhost"
                   "/services/load/metrics/load1",
                   text='[{"min": "2", "warning": "15", "value": "3"},'
                        '{"min": "5", "warning": "200", "value": "150"}]'
                   )

            live_query = ('{"time_interval": { "start_time": '
                          '"2015-05-22T13:38:08Z",'
                          '"end_time": "2015-05-22T13:38:08Z"}}')

            metrics = self.client.status.hosts.metrics.list('localhost',
                                                            'load1',
                                                            'load',
                                                            query=live_query
                                                            )

            self.assertEqual(
                metrics,
                [{"min": "2", "warning": "15", "value": "3"},
                 {"min": "5", "warning": "200", "value": "150"}]
            )

    def test_show_host_metrics(self):
        with requests_mock.mock() as m:
            m.get("http://localhost:5311/v2/status/hosts/localhost"
                  "/metrics/load1",
                  text='{"min": "2", "warning": "15", "value": "3"}'
                  )

            metrics = self.client.status.hosts.metrics.get('localhost',
                                                           'load1')

            self.assertEqual(
                metrics,
                {"min": "2", "warning": "15", "value": "3"}
            )

    def test_show_host_service_metrics(self):
        with requests_mock.mock() as m:
            m.get("http://localhost:5311/v2/status/hosts/localhost"
                  "/services/load/metrics/load1",
                  text='{"value": "3"}'
                  )

            metrics = self.client.status.hosts.metrics.get('localhost',
                                                           'load1',
                                                           'load')

            self.assertEqual(
                metrics,
                {"value": "3"}
            )

    def test_show_host_service(self):
        with requests_mock.mock() as m:
            m.get("http://localhost:5311/v2/status/hosts/localhost"
                  "/services/load/metrics",
                  text='[{"metric_name": "load1"},{"metric_name": "load5"}]'
                  )

            metrics = self.client.status.hosts.metrics.get(
                'localhost',
                service_description='load')

            self.assertEqual(
                metrics,
                [{"metric_name": "load1"}, {"metric_name": "load5"}]
            )