1 # -*- coding: utf-8 -*-
3 # Copyright 2018 Whitestack, LLC
4 # *************************************************************
6 # This file is part of OSM Monitoring module
7 # All Rights Reserved to Whitestack, LLC
9 # Licensed under the Apache License, Version 2.0 (the "License"); you may
10 # not use this file except in compliance with the License. You may obtain
11 # a copy of the License at
13 # http://www.apache.org/licenses/LICENSE-2.0
15 # Unless required by applicable law or agreed to in writing, software
16 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
17 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
18 # License for the specific language governing permissions and limitations
20 # For those usages not covered by the Apache License, Version 2.0 please
21 # contact: bdiaz@whitestack.com or glavado@whitestack.com
24 from unittest
import mock
26 from peewee
import SqliteDatabase
28 from osm_mon
.collector
.collector
import Collector
29 from osm_mon
.collector
.collectors
.openstack
import OpenstackCollector
30 from osm_mon
.core
import database
31 from osm_mon
.core
.database
import DatabaseManager
, VimCredentials
, Alarm
33 test_db
= SqliteDatabase(':memory:')
35 MODELS
= [VimCredentials
, Alarm
]
38 class CollectorTest(unittest
.TestCase
):
44 test_db
.drop_tables(MODELS
)
45 test_db
.create_tables(MODELS
)
51 @mock.patch("osm_mon.collector.collector.CommonDbClient", mock
.Mock())
52 @mock.patch
.object(OpenstackCollector
, "__init__", lambda *args
, **kwargs
: None)
53 @mock.patch
.object(OpenstackCollector
, "collect")
54 @mock.patch
.object(DatabaseManager
, "get_vim_type")
55 def test_init_vim_collector_and_collect_openstack(self
, _get_vim_type
, collect
):
56 _get_vim_type
.return_value
= 'openstack'
57 collector
= Collector()
58 collector
._collect
_vim
_metrics
({}, 'test_vim_account_id')
59 collect
.assert_called_once_with({})
61 @mock.patch("osm_mon.collector.collector.CommonDbClient", mock
.Mock())
62 @mock.patch
.object(OpenstackCollector
, "collect")
63 @mock.patch
.object(DatabaseManager
, "get_vim_type")
64 def test_init_vim_collector_and_collect_unknown(self
, _get_vim_type
, openstack_collect
):
65 _get_vim_type
.return_value
= 'unknown'
66 collector
= Collector()
67 collector
._collect
_vim
_metrics
({}, 'test_vim_account_id')
68 openstack_collect
.assert_not_called()
70 @mock.patch("osm_mon.collector.collector.CommonDbClient", mock
.Mock())
71 @mock.patch("osm_mon.collector.collector.VCACollector", autospec
=True)
72 def test_collect_vca_metrics(self
, vca_collector
):
73 collector
= Collector()
74 collector
._collect
_vca
_metrics
({})
75 vca_collector
.assert_called_once_with()
76 vca_collector
.return_value
.collect
.assert_called_once_with({})