from osm_mon.collector.collector import Collector
from osm_mon.collector.vnf_collectors.openstack import OpenstackCollector
from osm_mon.collector.collector import Collector
from osm_mon.collector.vnf_collectors.openstack import OpenstackCollector
from osm_mon.core.database import DatabaseManager, db
class CollectorTest(unittest.TestCase):
def setUp(self):
super().setUp()
from osm_mon.core.database import DatabaseManager, db
class CollectorTest(unittest.TestCase):
def setUp(self):
super().setUp()
@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'
@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._collect_vim_metrics({}, 'test_vim_account_id')
collect.assert_called_once_with({})
collector._collect_vim_metrics({}, 'test_vim_account_id')
collect.assert_called_once_with({})
@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'
@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._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._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):