It performs the corresponding action and sends a response through a unique topic made of 'alarm_response_' plus a correlation_id, which is a field contained in the request message.
-Alarms are stored in MON database in the OSM MySQL engine.
+Alarms are stored in MON database in the OSM MongoDB engine.
## MON Collector
from osm_mon.collector.collector import Collector
from osm_mon.core.config import Config
-from osm_mon.core.database import DatabaseManager
def main():
- parser = argparse.ArgumentParser(prog='osm-policy-agent')
- parser.add_argument('--config-file', nargs='?', help='POL configuration file')
+ parser = argparse.ArgumentParser(prog='osm-mon-collector')
+ parser.add_argument('--config-file', nargs='?', help='MON configuration file')
args = parser.parse_args()
cfg = Config(args.config_file)
log.info("Starting MON Collector...")
log.debug("Config: %s", cfg.conf)
log.info("Initializing database...")
- db_manager = DatabaseManager(cfg)
- db_manager.create_tables()
- log.info("Database initialized correctly.")
collector = Collector(cfg)
collector.collect_forever()
import sys
from osm_mon.core.config import Config
-from osm_mon.core.database import DatabaseManager
from osm_mon.evaluator.evaluator import Evaluator
def main():
- parser = argparse.ArgumentParser(prog='osm-policy-agent')
- parser.add_argument('--config-file', nargs='?', help='POL configuration file')
+ parser = argparse.ArgumentParser(prog='osm-mon-evaluator')
+ parser.add_argument('--config-file', nargs='?', help='MON configuration file')
args = parser.parse_args()
cfg = Config(args.config_file)
log.info("Starting MON Evaluator...")
log.debug("Config: %s", cfg.conf)
log.info("Initializing database...")
- db_manager = DatabaseManager(cfg)
- db_manager.create_tables()
- log.info("Database initialized correctly.")
evaluator = Evaluator(cfg)
evaluator.evaluate_forever()
import sys
from osm_mon.core.config import Config
-from osm_mon.core.database import DatabaseManager
from osm_mon.server.server import Server
def main():
- parser = argparse.ArgumentParser(prog='osm-policy-agent')
- parser.add_argument('--config-file', nargs='?', help='POL configuration file')
+ parser = argparse.ArgumentParser(prog='osm-mon-server')
+ parser.add_argument('--config-file', nargs='?', help='MON configuration file')
args = parser.parse_args()
cfg = Config(args.config_file)
log.info("Starting MON Server...")
log.debug("Config: %s", cfg.conf)
log.info("Initializing database...")
- db_manager = DatabaseManager(cfg)
- db_manager.create_tables()
- log.info("Database initialized correctly.")
loop = asyncio.get_event_loop()
server = Server(cfg, loop)
server.run()
# For those usages not covered by the Apache License, Version 2.0 please
# contact: bdiaz@whitestack.com or glavado@whitestack.com
##
+from typing import List
+
from osm_common import dbmongo, dbmemory
from osm_mon.core.config import Config
+from osm_mon.core.models import Alarm
class CommonDbClient:
def get_sdnc(self, sdnc_id: str):
return self.common_db.get_one('sdns', {'_id': sdnc_id})
+
+ def create_alarm(self, alarm: Alarm):
+ return self.common_db.create('alarms', alarm.to_dict())
+
+ def delete_alarm(self, alarm_uuid: str):
+ return self.common_db.del_one('alarms', {'uuid': alarm_uuid})
+
+ def get_alarms(self) -> List[Alarm]:
+ alarms = []
+ alarm_dicts = self.common_db.get_list('alarms')
+ for alarm_dict in alarm_dicts:
+ alarms.append(Alarm.from_dict(alarm_dict))
+ return alarms
+++ /dev/null
-# -*- 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 logging
-import os
-from typing import Iterable
-
-from peewee import CharField, FloatField, Model, AutoField, Proxy, ForeignKeyField
-from peewee_migrate import Router
-from playhouse.db_url import connect
-
-from osm_mon import migrations
-from osm_mon.core.config import Config
-
-log = logging.getLogger(__name__)
-
-db = Proxy()
-
-
-class BaseModel(Model):
- id = AutoField(primary_key=True)
-
- class Meta:
- database = db
-
-
-class Alarm(BaseModel):
- uuid = CharField(unique=True)
- name = CharField()
- severity = CharField()
- threshold = FloatField()
- operation = CharField()
- statistic = CharField()
- metric = CharField()
-
-
-class AlarmTag(BaseModel):
- name = CharField()
- value = CharField()
- alarm = ForeignKeyField(Alarm, related_name='tags', on_delete='CASCADE')
-
-
-class DatabaseManager:
- def __init__(self, config: Config):
- db.initialize(connect(config.get('sql', 'database_uri')))
-
- def create_tables(self) -> None:
- db.connect()
- with db.atomic():
- router = Router(db, os.path.dirname(migrations.__file__))
- router.run()
- db.close()
-
-
-class AlarmTagRepository:
- @staticmethod
- def create(**query) -> Alarm:
- return AlarmTag.create(**query)
-
-
-class AlarmRepository:
- @staticmethod
- def create(**query) -> Alarm:
- return Alarm.create(**query)
-
- @staticmethod
- def get(*expressions) -> Alarm:
- return Alarm.select().where(*expressions).get()
-
- @staticmethod
- def list(*expressions) -> Iterable[Alarm]:
- if expressions == ():
- return Alarm.select()
- else:
- return Alarm.select().where(*expressions)
--- /dev/null
+# -*- 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 uuid
+
+
+class Alarm:
+
+ def __init__(self, name: str = None, severity: str = None, threshold: float = None, operation: str = None,
+ statistic: str = None, metric: str = None, tags: dict = {}):
+ self.uuid = str(uuid.uuid4())
+ self.name = name
+ self.severity = severity
+ self.threshold = threshold
+ self.operation = operation
+ self.statistic = statistic
+ self.metric = metric
+ self.tags = tags
+
+ def to_dict(self) -> dict:
+ alarm = {
+ 'uuid': self.uuid,
+ 'name': self.name,
+ 'severity': self.severity,
+ 'threshold': self.threshold,
+ 'statistic': self.statistic,
+ 'metric': self.metric,
+ 'tags': self.tags
+ }
+ return alarm
+
+ @staticmethod
+ def from_dict(data: dict):
+ alarm = Alarm()
+ alarm.uuid = data.get('uuid', str(uuid.uuid4()))
+ alarm.name = data.get('name')
+ alarm.severity = data.get('severity')
+ alarm.threshold = data.get('threshold')
+ alarm.statistic = data.get('statistic')
+ alarm.metric = data.get('metric')
+ alarm.tags = data.get('tags')
+ return alarm
import peewee
from osm_mon.core.config import Config
-from osm_mon.core.database import Alarm
from osm_mon.core.message_bus_client import MessageBusClient
+from osm_mon.core.models import Alarm
from osm_mon.core.response import ResponseBuilder
from osm_mon.evaluator.service import EvaluatorService, AlarmStatus
from enum import Enum
from typing import Tuple, List
-from osm_mon.core import database
from osm_mon.core.common_db import CommonDbClient
from osm_mon.core.config import Config
-from osm_mon.core.database import Alarm, AlarmRepository
+from osm_mon.core.models import Alarm
from osm_mon.evaluator.backends.prometheus import PrometheusBackend
log = logging.getLogger(__name__)
return BACKENDS[self.conf.get('evaluator', 'backend')](self.conf).get_metric_value(metric_name, tags)
def _evaluate_metric(self,
- alarm: Alarm, tags: dict):
+ alarm: Alarm):
log.debug("_evaluate_metric")
- metric_value = self._get_metric_value(alarm.metric, tags)
+ metric_value = self._get_metric_value(alarm.metric, alarm.tags)
if metric_value is None:
- log.warning("No metric result for alarm %s", alarm.id)
+ log.warning("No metric result for alarm %s", alarm.uuid)
self.queue.put((alarm, AlarmStatus.INSUFFICIENT))
else:
if alarm.operation.upper() == 'GT':
def evaluate_alarms(self) -> List[Tuple[Alarm, AlarmStatus]]:
log.debug('evaluate_alarms')
processes = []
- database.db.connect()
- try:
- with database.db.atomic():
- for alarm in AlarmRepository.list():
- # Tags need to be passed inside a dict to avoid database locking issues related to process forking
- tags = {}
- for tag in alarm.tags:
- tags[tag.name] = tag.value
- p = multiprocessing.Process(target=self._evaluate_metric,
- args=(alarm, tags))
- processes.append(p)
- p.start()
-
- for process in processes:
- process.join(timeout=10)
- alarms_tuples = []
- log.info("Appending alarms to queue")
- while not self.queue.empty():
- alarms_tuples.append(self.queue.get())
- return alarms_tuples
- finally:
- database.db.close()
+ for alarm in self.common_db.get_alarms():
+ p = multiprocessing.Process(target=self._evaluate_metric,
+ args=(alarm,))
+ processes.append(p)
+ p.start()
+
+ for process in processes:
+ process.join(timeout=10)
+ alarms_tuples = []
+ while not self.queue.empty():
+ alarms_tuples.append(self.queue.get())
+ return alarms_tuples
+++ /dev/null
-"""Peewee migrations -- 001_initial.py.
-
-Some examples (model - class or model name)::
-
- > Model = migrator.orm['model_name'] # Return model in current state by name
-
- > migrator.sql(sql) # Run custom SQL
- > migrator.python(func, *args, **kwargs) # Run python code
- > migrator.create_model(Model) # Create a model (could be used as decorator)
- > migrator.remove_model(model, cascade=True) # Remove a model
- > migrator.add_fields(model, **fields) # Add fields to a model
- > migrator.change_fields(model, **fields) # Change fields
- > migrator.remove_fields(model, *field_names, cascade=True)
- > migrator.rename_field(model, old_field_name, new_field_name)
- > migrator.rename_table(model, new_table_name)
- > migrator.add_index(model, *col_names, unique=False)
- > migrator.drop_index(model, *col_names)
- > migrator.add_not_null(model, *field_names)
- > migrator.drop_not_null(model, *field_names)
- > migrator.add_default(model, field_name, default)
-
-"""
-
-import peewee as pw
-
-SQL = pw.SQL
-
-
-def migrate(migrator, database, fake=False, **kwargs):
- """Write your migrations here."""
-
- @migrator.create_model
- class Alarm(pw.Model):
- id = pw.AutoField()
- uuid = pw.CharField(max_length=255, unique=True)
- name = pw.CharField(max_length=255)
- severity = pw.CharField(max_length=255)
- threshold = pw.FloatField()
- operation = pw.CharField(max_length=255)
- statistic = pw.CharField(max_length=255)
- monitoring_param = pw.CharField(max_length=255)
- vdur_name = pw.CharField(max_length=255)
- vnf_member_index = pw.CharField(max_length=255)
- nsr_id = pw.CharField(max_length=255)
-
- class Meta:
- table_name = "alarm"
-
- @migrator.create_model
- class BaseModel(pw.Model):
- id = pw.AutoField()
-
- class Meta:
- table_name = "basemodel"
-
- @migrator.create_model
- class VimCredentials(pw.Model):
- id = pw.AutoField()
- uuid = pw.CharField(max_length=255, unique=True)
- name = pw.CharField(max_length=255)
- type = pw.CharField(max_length=255)
- url = pw.CharField(max_length=255)
- user = pw.CharField(max_length=255)
- password = pw.CharField(max_length=255)
- tenant_name = pw.CharField(max_length=255)
- config = pw.TextField()
-
- class Meta:
- table_name = "vimcredentials"
-
-
-def rollback(migrator, database, fake=False, **kwargs):
- """Write your rollback migrations here."""
-
- migrator.remove_model('vimcredentials')
-
- migrator.remove_model('basemodel')
-
- migrator.remove_model('alarm')
+++ /dev/null
-# -*- 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
-##
-"""Peewee migrations -- 002_add_alarm_tags.py.
-
-Some examples (model - class or model name)::
-
- > Model = migrator.orm['model_name'] # Return model in current state by name
-
- > migrator.sql(sql) # Run custom SQL
- > migrator.python(func, *args, **kwargs) # Run python code
- > migrator.create_model(Model) # Create a model (could be used as decorator)
- > migrator.remove_model(model, cascade=True) # Remove a model
- > migrator.add_fields(model, **fields) # Add fields to a model
- > migrator.change_fields(model, **fields) # Change fields
- > migrator.remove_fields(model, *field_names, cascade=True)
- > migrator.rename_field(model, old_field_name, new_field_name)
- > migrator.rename_table(model, new_table_name)
- > migrator.add_index(model, *col_names, unique=False)
- > migrator.drop_index(model, *col_names)
- > migrator.add_not_null(model, *field_names)
- > migrator.drop_not_null(model, *field_names)
- > migrator.add_default(model, field_name, default)
-
-"""
-
-import peewee as pw
-
-SQL = pw.SQL
-
-
-def migrate(migrator, database, fake=False, **kwargs):
- """Write your migrations here."""
-
- @migrator.create_model
- class AlarmTag(pw.Model):
- id = pw.AutoField()
- name = pw.CharField(max_length=255)
- value = pw.CharField(max_length=255)
- alarm = pw.ForeignKeyField(backref='tags', column_name='alarm_id', field='id',
- model=migrator.orm['alarm'], on_delete='CASCADE')
-
- class Meta:
- table_name = "alarmtag"
-
-
-def rollback(migrator, database, fake=False, **kwargs):
- """Write your rollback migrations here."""
-
- migrator.remove_model('alarmtag')
+++ /dev/null
-# -*- 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
-##
-"""Peewee migrations -- 003_rename_monitoring_param.py.
-
-Some examples (model - class or model name)::
-
- > Model = migrator.orm['model_name'] # Return model in current state by name
-
- > migrator.sql(sql) # Run custom SQL
- > migrator.python(func, *args, **kwargs) # Run python code
- > migrator.create_model(Model) # Create a model (could be used as decorator)
- > migrator.remove_model(model, cascade=True) # Remove a model
- > migrator.add_fields(model, **fields) # Add fields to a model
- > migrator.change_fields(model, **fields) # Change fields
- > migrator.remove_fields(model, *field_names, cascade=True)
- > migrator.rename_field(model, old_field_name, new_field_name)
- > migrator.rename_table(model, new_table_name)
- > migrator.add_index(model, *col_names, unique=False)
- > migrator.drop_index(model, *col_names)
- > migrator.add_not_null(model, *field_names)
- > migrator.drop_not_null(model, *field_names)
- > migrator.add_default(model, field_name, default)
-
-"""
-
-import peewee as pw
-
-SQL = pw.SQL
-
-
-def migrate(migrator, database, fake=False, **kwargs):
- """Write your migrations here."""
-
- migrator.rename_field('alarm', 'monitoring_param', 'metric')
-
-
-def rollback(migrator, database, fake=False, **kwargs):
- """Write your rollback migrations here."""
-
- migrator.rename_field('alarm', 'metric', 'monitoring_param')
+++ /dev/null
-# -*- 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
-##
-"""Peewee migrations -- 004_remove_alarm_fields.py.
-
-Some examples (model - class or model name)::
-
- > Model = migrator.orm['model_name'] # Return model in current state by name
-
- > migrator.sql(sql) # Run custom SQL
- > migrator.python(func, *args, **kwargs) # Run python code
- > migrator.create_model(Model) # Create a model (could be used as decorator)
- > migrator.remove_model(model, cascade=True) # Remove a model
- > migrator.add_fields(model, **fields) # Add fields to a model
- > migrator.change_fields(model, **fields) # Change fields
- > migrator.remove_fields(model, *field_names, cascade=True)
- > migrator.rename_field(model, old_field_name, new_field_name)
- > migrator.rename_table(model, new_table_name)
- > migrator.add_index(model, *col_names, unique=False)
- > migrator.drop_index(model, *col_names)
- > migrator.add_not_null(model, *field_names)
- > migrator.drop_not_null(model, *field_names)
- > migrator.add_default(model, field_name, default)
-
-"""
-
-import peewee as pw
-
-SQL = pw.SQL
-
-
-def migrate(migrator, database, fake=False, **kwargs):
- """Write your migrations here."""
-
- migrator.remove_fields('alarm', 'vdur_name', 'vnf_member_index', 'nsr_id')
-
-
-def rollback(migrator, database, fake=False, **kwargs):
- """Write your rollback migrations here."""
-
- migrator.add_fields('alarm',
- vdur_name=pw.CharField(max_length=255),
- vnf_member_index=pw.CharField(max_length=255),
- nsr_id=pw.CharField(max_length=255))
+++ /dev/null
-# -*- 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
-##
+++ /dev/null
-# -*- 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
-
-DATABASE = os.getenv('OSMMON_SQL_DATABASE_URI', 'sqlite://')
# contact: bdiaz@whitestack.com or glavado@whitestack.com
##
import logging
-import uuid
-from osm_mon.core import database
from osm_mon.core.common_db import CommonDbClient
from osm_mon.core.config import Config
-from osm_mon.core.database import AlarmRepository, Alarm, AlarmTagRepository
+from osm_mon.core.models import Alarm
log = logging.getLogger(__name__)
def create_alarm(self,
name: str,
- threshold: str,
+ threshold: float,
operation: str,
severity: str,
statistic: str,
metric_name: str,
tags: dict) -> Alarm:
- database.db.connect()
- try:
- with database.db.atomic():
- alarm = AlarmRepository.create(
- uuid=str(uuid.uuid4()),
- name=name,
- threshold=threshold,
- operation=operation.lower(),
- severity=severity.lower(),
- statistic=statistic.lower(),
- metric=metric_name
- )
- for k, v in tags.items():
- AlarmTagRepository.create(
- name=k,
- value=v,
- alarm=alarm
- )
- return alarm
-
- finally:
- database.db.close()
+ alarm = Alarm(name, severity, threshold, operation, statistic, metric_name, tags)
+ self.common_db.create_alarm(alarm)
+ return alarm
def delete_alarm(self,
alarm_uuid: str) -> None:
- database.db.connect()
- try:
- with database.db.atomic():
- alarm = AlarmRepository.get(Alarm.uuid == alarm_uuid)
- alarm.delete_instance()
- finally:
- database.db.close()
+ self.common_db.delete_alarm(alarm_uuid)
from osm_mon.core.common_db import CommonDbClient
from osm_mon.core.config import Config
+from osm_mon.core.models import Alarm
class CommonDbClientTest(unittest.TestCase):
decrypt_vim_password.assert_any_call('vim_password', schema_version, vim_id)
self.assertRaises(AssertionError, decrypt_vim_password.assert_any_call, 'vrops_password', schema_version,
vim_id)
+
+ @mock.patch.object(dbmongo.DbMongo, "db_connect", mock.Mock())
+ @mock.patch.object(dbmongo.DbMongo, "get_list")
+ def test_get_alarms(self, get_list):
+ get_list.return_value = [{
+ 'uuid': '1',
+ 'name': 'name',
+ 'severity': 'severity',
+ 'threshold': 50,
+ 'operation': 'operation',
+ 'statistic': 'statistic',
+ 'tags': {},
+ }]
+
+ common_db_client = CommonDbClient(self.config)
+ alarms = common_db_client.get_alarms()
+ self.assertEqual('1', alarms[0].uuid)
+
+ @mock.patch.object(dbmongo.DbMongo, "db_connect", mock.Mock())
+ @mock.patch.object(dbmongo.DbMongo, "create")
+ def test_create_alarm(self, create):
+ alarm = Alarm('name', 'severity', 50.0, 'operation', 'statistic', 'metric', {})
+ alarm.uuid = '1'
+ common_db_client = CommonDbClient(self.config)
+ common_db_client.create_alarm(alarm)
+ create.assert_called_with('alarms', {'tags': {}, 'threshold': 50.0, 'metric': 'metric', 'severity': 'severity',
+ 'statistic': 'statistic', 'name': 'name', 'uuid': '1'})
+
+ @mock.patch.object(dbmongo.DbMongo, "db_connect", mock.Mock())
+ @mock.patch.object(dbmongo.DbMongo, "del_one")
+ def test_delete_alarm(self, delete):
+ common_db_client = CommonDbClient(self.config)
+ common_db_client.delete_alarm('1')
+ delete.assert_called_with('alarms', {'uuid': '1'})
@mock.patch('multiprocessing.Process')
@mock.patch.object(Evaluator, "notify_alarm")
@mock.patch.object(EvaluatorService, "evaluate_alarms")
- @mock.patch('osm_mon.core.database.db')
- def test_evaluate(self, db, evaluate_alarms, notify_alarm, process):
+ def test_evaluate(self, evaluate_alarms, notify_alarm, process):
mock_alarm = mock.Mock()
mock_alarm.operation = 'gt'
mock_alarm.threshold = 50.0
from osm_mon.core.common_db import CommonDbClient
from osm_mon.core.config import Config
-from osm_mon.core.database import AlarmRepository, AlarmTag
from osm_mon.core.message_bus_client import MessageBusClient
from osm_mon.evaluator.backends.prometheus import PrometheusBackend
from osm_mon.evaluator.evaluator import AlarmStatus
self.config = Config()
@mock.patch.object(EvaluatorService, "_get_metric_value")
- @mock.patch('osm_mon.core.database.db')
- def test_evaluate_metric(self, db, get_metric_value):
+ def test_evaluate_metric(self, get_metric_value):
mock_alarm = mock.Mock()
mock_alarm.operation = 'gt'
mock_alarm.threshold = 50.0
service = EvaluatorService(self.config)
service.queue = mock.Mock()
- service._evaluate_metric(mock_alarm, {})
+ service._evaluate_metric(mock_alarm)
service.queue.put.assert_called_with((mock_alarm, AlarmStatus.ALARM))
service.queue.reset_mock()
mock_alarm.operation = 'lt'
- service._evaluate_metric(mock_alarm, {})
+ service._evaluate_metric(mock_alarm)
service.queue.put.assert_called_with((mock_alarm, AlarmStatus.OK))
service.queue.reset_mock()
get_metric_value.return_value = None
- service._evaluate_metric(mock_alarm, {})
+ service._evaluate_metric(mock_alarm)
service.queue.put.assert_called_with((mock_alarm, AlarmStatus.INSUFFICIENT))
@mock.patch('multiprocessing.Process')
@mock.patch.object(EvaluatorService, "_evaluate_metric")
@mock.patch.object(CommonDbClient, "get_vnfd")
@mock.patch.object(CommonDbClient, "get_vnfr")
- @mock.patch.object(AlarmRepository, "list")
- @mock.patch('osm_mon.core.database.db')
- def test_evaluate(self, db, alarm_list, get_vnfr, get_vnfd, evaluate_metric, process):
+ @mock.patch.object(CommonDbClient, "get_alarms")
+ def test_evaluate_alarms(self, alarm_list, get_vnfr, get_vnfd, evaluate_metric, process):
mock_alarm = mock.Mock()
mock_alarm.vdur_name = 'cirros_ns-1-cirros_vnfd-VM-1'
mock_alarm.monitoring_param = 'cirros_vnf_memory_util'
- mock_tag = AlarmTag()
- mock_tag.name = 'name'
- mock_tag.value = 'value'
- mock_alarm.tags = [mock_tag]
+ mock_alarm.tags = {'name': 'value'}
alarm_list.return_value = [mock_alarm]
get_vnfr.return_value = vnfr_record_mock
get_vnfd.return_value = vnfd_record_mock
evaluator = EvaluatorService(self.config)
evaluator.evaluate_alarms()
- process.assert_called_with(target=evaluate_metric, args=(mock_alarm, {'name': 'value'}))
+ process.assert_called_with(target=evaluate_metric, args=(mock_alarm,))
@mock.patch.object(PrometheusBackend, "get_metric_value")
- @mock.patch('osm_mon.core.database.db')
- def test_get_metric_value_prometheus(self, db, get_metric_value):
+ def test_get_metric_value_prometheus(self, get_metric_value):
self.config.set('evaluator', 'backend', 'prometheus')
evaluator = EvaluatorService(self.config)
evaluator._get_metric_value('test', {})
projects / osm / MON.git / commitdiff
