# -*- coding: utf-8 -*-

# Copyright 2018 Whitestack, LLC
# *************************************************************

# This file is part of OSM Monitoring module
# All Rights Reserved to Whitestack, LLC

# 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.
# For those usages not covered by the Apache License, Version 2.0 please
# contact: bdiaz@whitestack.com or glavado@whitestack.com
##
import os
import unittest
from unittest import mock

from osm_mon.collector.collector import Collector
from osm_mon.collector.vnf_collectors.openstack import OpenstackCollector
from osm_mon.core.config import Config
from osm_mon.core.database import DatabaseManager, db


class CollectorTest(unittest.TestCase):
    def setUp(self):
        super().setUp()
        os.environ["OSMMON_SQL_DATABASE_URI"] = "sqlite:///:memory:"
        self.config = Config()
        db_manager = DatabaseManager(self.config)
        db_manager.create_tables()

    def tearDown(self):
        super().tearDown()
        db.close()

    @mock.patch("osm_mon.collector.collector.CommonDbClient", mock.Mock())
    @mock.patch.object(Collector, "_init_backends", mock.Mock())
    @mock.patch.object(OpenstackCollector, "__init__", lambda *args, **kwargs: None)
    @mock.patch.object(OpenstackCollector, "collect")
    @mock.patch.object(DatabaseManager, "get_vim_type")
    def test_init_vim_collector_and_collect_openstack(self, _get_vim_type, collect):
        _get_vim_type.return_value = 'openstack'
        collector = Collector(self.config)
        collector._collect_vim_metrics({}, 'test_vim_account_id')
        collect.assert_called_once_with({})

    @mock.patch("osm_mon.collector.collector.CommonDbClient", mock.Mock())
    @mock.patch.object(Collector, "_init_backends", mock.Mock())
    @mock.patch.object(OpenstackCollector, "collect")
    @mock.patch.object(DatabaseManager, "get_vim_type")
    def test_init_vim_collector_and_collect_unknown(self, _get_vim_type, openstack_collect):
        _get_vim_type.return_value = 'unknown'
        collector = Collector(self.config)
        collector._collect_vim_metrics({}, 'test_vim_account_id')
        openstack_collect.assert_not_called()

    @mock.patch("osm_mon.collector.collector.CommonDbClient", mock.Mock())
    @mock.patch("osm_mon.collector.collector.VCACollector", autospec=True)
    def test_collect_vca_metrics(self, vca_collector):
        collector = Collector(self.config)
        collector._collect_vca_metrics({})
        vca_collector.assert_called_once_with(self.config)
        vca_collector.return_value.collect.assert_called_once_with({})
