openstack/vitrage-tempest-plugin
Code Issues Proposed changes

Merge "use vitrage_client attribute where possible"

Browse Source
This commit is contained in:
Zuul 2019-02-26 09:47:06 +00:00 committed by Gerrit Code Review
commit 8f4de82480
10 changed files with 75 additions and 104 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vitrage_tempest_plugin/tests/api/rca/base.py
View File

@ -35,7 +35,6 @@ from vitrage_tempest_plugin.tests.common.constants import VertexProperties \
from vitrage_tempest_plugin.tests.common.constants import VITRAGE_DATASOURCE from vitrage_tempest_plugin.tests.common.constants import VITRAGE_DATASOURCE
from vitrage_tempest_plugin.tests.common import general_utils as g_utils from vitrage_tempest_plugin.tests.common import general_utils as g_utils
from vitrage_tempest_plugin.tests.common import nova_utils from vitrage_tempest_plugin.tests.common import nova_utils
from vitrage_tempest_plugin.tests.common.tempest_clients import TempestClients
from vitrage_tempest_plugin.tests.common import vitrage_utils from vitrage_tempest_plugin.tests.common import vitrage_utils
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -45,7 +44,8 @@ VITRAGE_ALARM_NAME = 'instance_deduce'
class BaseRcaTest(BaseAlarmsTest): class BaseRcaTest(BaseAlarmsTest):
def _clean_all(self): @staticmethod
def _clean_all():
nova_utils.delete_all_instances() nova_utils.delete_all_instances()
aodh_utils.delete_all_aodh_alarms() aodh_utils.delete_all_aodh_alarms()
@ -102,7 +102,7 @@ class BaseRcaTest(BaseAlarmsTest):
LOG.info("The alarms list is : " + str(json.dumps(alarms))) LOG.info("The alarms list is : " + str(json.dumps(alarms)))
# Find the vitrage_id of the deduced alarms using their original id. # Find the vitrage_id of the deduced alarms using their original id.
vitrage_resources = TempestClients.vitrage().resource.list( vitrage_resources = self.vitrage_client.resource.list(
all_tenants=False) all_tenants=False)
vitrage_instance_0_id = g_utils.first_match(vitrage_resources, vitrage_instance_0_id = g_utils.first_match(vitrage_resources,
id=instances[0].id) id=instances[0].id)

10
vitrage_tempest_plugin/tests/api/templates/test_template.py
View File

@ -23,7 +23,6 @@ from vitrage_tempest_plugin.tests.common.constants import TemplateStatus
from vitrage_tempest_plugin.tests.common.constants import TemplateTypes as \ from vitrage_tempest_plugin.tests.common.constants import TemplateTypes as \
TTypes TTypes
from vitrage_tempest_plugin.tests.common import general_utils as g_utils from vitrage_tempest_plugin.tests.common import general_utils as g_utils
from vitrage_tempest_plugin.tests.common.tempest_clients import TempestClients
from vitrage_tempest_plugin.tests.common import vitrage_utils as v_utils from vitrage_tempest_plugin.tests.common import vitrage_utils as v_utils
import vitrage_tempest_plugin.tests.utils as utils import vitrage_tempest_plugin.tests.utils as utils
@ -110,11 +109,6 @@ class TestValidate(BaseTemplateTest):
class TemplatesDBTest(BaseTemplateTest): class TemplatesDBTest(BaseTemplateTest):
"""Template DB test class for vitrage API tests""" """Template DB test class for vitrage API tests"""
@classmethod
def setUpClass(cls):
super(TemplatesDBTest, cls).setUpClass()
cls.client = TempestClients.vitrage()
def test_template_add(self): def test_template_add(self):
"""template add test """template add test
@ -193,7 +187,7 @@ class TemplatesDBTest(BaseTemplateTest):
templates_names = self._add_templates() templates_names = self._add_templates()
cli_templates_list = utils.run_vitrage_command( cli_templates_list = utils.run_vitrage_command(
"vitrage template list") "vitrage template list")
api_templates_list = self.client.template.list() api_templates_list = self.vitrage_client.template.list()
self.assertThat(api_templates_list, IsNotEmpty(), self.assertThat(api_templates_list, IsNotEmpty(),
'The template list taken from api is empty') 'The template list taken from api is empty')
@ -220,7 +214,7 @@ class TemplatesDBTest(BaseTemplateTest):
name='host_high_memory_usage_scenarios', name='host_high_memory_usage_scenarios',
type=TTypes.STANDARD, type=TTypes.STANDARD,
status=TemplateStatus.ACTIVE) status=TemplateStatus.ACTIVE)
payload_from_db = self.client.template.show(db_row['uuid']) payload_from_db = self.vitrage_client.template.show(db_row['uuid'])
with open(template_path, 'r') as stream: with open(template_path, 'r') as stream:
payload_from_file = yaml.load(stream, Loader=yaml.BaseLoader) payload_from_file = yaml.load(stream, Loader=yaml.BaseLoader)
self.assert_dict_equal(payload_from_file, payload_from_db, self.assert_dict_equal(payload_from_file, payload_from_db,

39
vitrage_tempest_plugin/tests/api/webhook/test_webhook.py
View File

@ -16,7 +16,6 @@ from oslo_log import log as logging
from testtools import matchers from testtools import matchers
from vitrage_tempest_plugin.tests.base import BaseVitrageTempest from vitrage_tempest_plugin.tests.base import BaseVitrageTempest
from vitrage_tempest_plugin.tests.common.tempest_clients import TempestClients
from vitrageclient.exceptions import ClientException from vitrageclient.exceptions import ClientException
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -35,17 +34,17 @@ class TestWebhook(BaseVitrageTempest):
def setUpClass(cls): def setUpClass(cls):
super(TestWebhook, cls).setUpClass() super(TestWebhook, cls).setUpClass()
cls.pre_test_webhook_count = \ cls.pre_test_webhook_count = \
len(TempestClients.vitrage().webhook.list()) len(cls.vitrage_client.webhook.list())
def test_add_webhook(self): def test_add_webhook(self):
webhooks = TempestClients.vitrage().webhook.list() webhooks = self.vitrage_client.webhook.list()
self.assertThat(webhooks, self.assertThat(webhooks,
matchers.HasLength(self.pre_test_webhook_count), matchers.HasLength(self.pre_test_webhook_count),
'Amount of webhooks should be ' 'Amount of webhooks should be '
'the same as before the test') 'the same as before the test')
created_webhook = TempestClients.vitrage().webhook.add( created_webhook = self.vitrage_client.webhook.add(
url="https://www.test.com", url="https://www.test.com",
regex_filter=REGEX_PROPS, regex_filter=REGEX_PROPS,
headers=HEADERS_PROPS headers=HEADERS_PROPS
@ -63,27 +62,27 @@ class TestWebhook(BaseVitrageTempest):
"https://www.test.com", "https://www.test.com",
'URL not created correctly') 'URL not created correctly')
webhooks = TempestClients.vitrage().webhook.list() webhooks = self.vitrage_client.webhook.list()
self.assertThat(webhooks, self.assertThat(webhooks,
matchers.HasLength(self.pre_test_webhook_count + 1)) matchers.HasLength(self.pre_test_webhook_count + 1))
TempestClients.vitrage().webhook.delete( self.vitrage_client.webhook.delete(
created_webhook['id']) created_webhook['id'])
def test_delete_webhook(self): def test_delete_webhook(self):
webhooks = TempestClients.vitrage().webhook.list() webhooks = self.vitrage_client.webhook.list()
self.assertThat(webhooks, self.assertThat(webhooks,
matchers.HasLength(self.pre_test_webhook_count), matchers.HasLength(self.pre_test_webhook_count),
'Amount of webhooks should ' 'Amount of webhooks should '
'be the same as before the test') 'be the same as before the test')
created_webhook = TempestClients.vitrage().webhook.add( created_webhook = self.vitrage_client.webhook.add(
url="https://www.test.com", url="https://www.test.com",
regex_filter=REGEX_PROPS, regex_filter=REGEX_PROPS,
headers=HEADERS_PROPS headers=HEADERS_PROPS
) )
created_webhook = TempestClients.vitrage().webhook.delete( created_webhook = self.vitrage_client.webhook.delete(
id=created_webhook['id']) id=created_webhook['id'])
self.assertIsNotNone(created_webhook.get('SUCCESS'), self.assertIsNotNone(created_webhook.get('SUCCESS'),
'failed to delete') 'failed to delete')
@ -93,24 +92,24 @@ class TestWebhook(BaseVitrageTempest):
def test_delete_non_existing_webhook(self): def test_delete_non_existing_webhook(self):
self.assertRaises(ClientException, self.assertRaises(ClientException,
TempestClients.vitrage().webhook.delete, self.vitrage_client.webhook.delete,
('non existant')) ('non existant'))
def test_list_webhook(self): def test_list_webhook(self):
webhooks = TempestClients.vitrage().webhook.list() webhooks = self.vitrage_client.webhook.list()
self.assertThat(webhooks, self.assertThat(webhooks,
matchers.HasLength(self.pre_test_webhook_count), matchers.HasLength(self.pre_test_webhook_count),
'Amount of webhooks should be ' 'Amount of webhooks should be '
'the same as before the test') 'the same as before the test')
created_webhook = TempestClients.vitrage().webhook.add( created_webhook = self.vitrage_client.webhook.add(
url="https://www.test.com", url="https://www.test.com",
regex_filter=REGEX_PROPS, regex_filter=REGEX_PROPS,
headers=HEADERS_PROPS headers=HEADERS_PROPS
) )
webhooks = TempestClients.vitrage().webhook.list() webhooks = self.vitrage_client.webhook.list()
self.assertThat(webhooks, self.assertThat(webhooks,
matchers.HasLength(self.pre_test_webhook_count + 1)) matchers.HasLength(self.pre_test_webhook_count + 1))
self.assertEqual(created_webhook[HEADERS], webhooks[0][HEADERS]) self.assertEqual(created_webhook[HEADERS], webhooks[0][HEADERS])
@ -118,25 +117,22 @@ class TestWebhook(BaseVitrageTempest):
self.assertEqual(created_webhook[REGEX_FILTER], self.assertEqual(created_webhook[REGEX_FILTER],
webhooks[0][REGEX_FILTER]) webhooks[0][REGEX_FILTER])
TempestClients.vitrage().webhook.delete( self.vitrage_client.webhook.delete(created_webhook['id'])
created_webhook['id'])
def test_show_webhook(self): def test_show_webhook(self):
webhooks = TempestClients.vitrage().webhook.list() webhooks = self.vitrage_client.webhook.list()
self.assertThat(webhooks, self.assertThat(webhooks,
matchers.HasLength(self.pre_test_webhook_count), matchers.HasLength(self.pre_test_webhook_count),
'Amount of webhooks should be ' 'Amount of webhooks should be '
'the same as before the test') 'the same as before the test')
created_webhook = TempestClients.vitrage().webhook.add( created_webhook = self.vitrage_client.webhook.add(
url="https://www.test.com", url="https://www.test.com",
regex_filter=REGEX_PROPS, regex_filter=REGEX_PROPS,
headers=HEADERS_PROPS headers=HEADERS_PROPS
) )
show_webhook = TempestClients.vitrage().webhook.show( show_webhook = self.vitrage_client.webhook.show(created_webhook['id'])
created_webhook['id']
)
self.assertIsNotNone(show_webhook, 'webhook not listed') self.assertIsNotNone(show_webhook, 'webhook not listed')
self.assertEqual(created_webhook[HEADERS], self.assertEqual(created_webhook[HEADERS],
@ -149,5 +145,4 @@ class TestWebhook(BaseVitrageTempest):
show_webhook[URL], show_webhook[URL],
'URL mismatch') 'URL mismatch')
TempestClients.vitrage().webhook.delete( self.vitrage_client.webhook.delete(created_webhook['id'])
created_webhook['id'])

2
vitrage_tempest_plugin/tests/base.py
View File

@ -117,6 +117,8 @@ class BaseVitrageTempest(test.BaseTestCase):
warnings.filterwarnings(action="ignore", warnings.filterwarnings(action="ignore",
message="unclosed", message="unclosed",
category=ResourceWarning) category=ResourceWarning)
logger = logging.getLogger('urllib3.connectionpool').logger
logger.setLevel(logging.INFO)
def tearDown(self): def tearDown(self):
super(BaseVitrageTempest, self).tearDown() super(BaseVitrageTempest, self).tearDown()

6
vitrage_tempest_plugin/tests/e2e/test_actions_base.py
View File

@ -20,7 +20,6 @@ from vitrage_tempest_plugin.tests.base import BaseVitrageTempest
from vitrage_tempest_plugin.tests.common.constants import VertexProperties as \ from vitrage_tempest_plugin.tests.common.constants import VertexProperties as \
VProps VProps
from vitrage_tempest_plugin.tests.common import general_utils as g_utils from vitrage_tempest_plugin.tests.common import general_utils as g_utils
from vitrage_tempest_plugin.tests.common.tempest_clients import TempestClients
from vitrage_tempest_plugin.tests.common import vitrage_utils from vitrage_tempest_plugin.tests.common import vitrage_utils
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
@ -54,9 +53,8 @@ class TestActionsBase(BaseVitrageTempest):
time.sleep(2) time.sleep(2)
def _check_deduced(self, deduced_count, deduced_props, resource_id): def _check_deduced(self, deduced_count, deduced_props, resource_id):
alarms = TempestClients.vitrage().alarm.list( alarms = self.vitrage_client.alarm.list(vitrage_id=resource_id,
vitrage_id=resource_id, all_tenants=True)
all_tenants=True)
deduces = g_utils.all_matches(alarms, **deduced_props) deduces = g_utils.all_matches(alarms, **deduced_props)
self.assertThat(deduces, matchers.HasLength(deduced_count), self.assertThat(deduces, matchers.HasLength(deduced_count),
'Expected %s deduces\n - \n%s\n - \n%s' % 'Expected %s deduces\n - \n%s\n - \n%s' %

10
vitrage_tempest_plugin/tests/e2e/test_basic_actions.py
View File

@ -253,7 +253,7 @@ class TestBasicActions(TestActionsBase):
try: try:
# Do # Do
self._trigger_do_action(trigger_name) self._trigger_do_action(trigger_name)
alarms = TempestClients.vitrage().alarm.list( alarms = self.vitrage_client.alarm.list(
vitrage_id=self.orig_host.get(VProps.VITRAGE_ID), vitrage_id=self.orig_host.get(VProps.VITRAGE_ID),
all_tenants=True) all_tenants=True)
self.assertTrue(len(alarms) >= 2, 'alarms %s' % str(alarms)) self.assertTrue(len(alarms) >= 2, 'alarms %s' % str(alarms))
@ -262,13 +262,13 @@ class TestBasicActions(TestActionsBase):
trigger = g_utils.first_match(alarms, **trigger_alarm_props) trigger = g_utils.first_match(alarms, **trigger_alarm_props)
# Get Rca for the deduced # Get Rca for the deduced
rca = TempestClients.vitrage().rca.get( rca = self.vitrage_client.rca.get(deduced[VProps.VITRAGE_ID],
deduced[VProps.VITRAGE_ID], all_tenants=True) all_tenants=True)
self._check_rca(rca, [deduced, trigger], deduced_props) self._check_rca(rca, [deduced, trigger], deduced_props)
# Get Rca for the trigger # Get Rca for the trigger
rca = TempestClients.vitrage().rca.get( rca = self.vitrage_client.rca.get(trigger[VProps.VITRAGE_ID],
trigger[VProps.VITRAGE_ID], all_tenants=True) all_tenants=True)
self._check_rca(rca, [deduced, trigger], trigger_alarm_props) self._check_rca(rca, [deduced, trigger], trigger_alarm_props)
finally: finally:
self._trigger_undo_action(trigger_name) self._trigger_undo_action(trigger_name)

50
vitrage_tempest_plugin/tests/e2e/test_e2e_webhook.py
View File

@ -24,7 +24,6 @@ import time
from vitrage_tempest_plugin.tests.base import IsEmpty from vitrage_tempest_plugin.tests.base import IsEmpty
from vitrage_tempest_plugin.tests.common.constants import VertexProperties as \ from vitrage_tempest_plugin.tests.common.constants import VertexProperties as \
VProps VProps
from vitrage_tempest_plugin.tests.common.tempest_clients import TempestClients
from vitrage_tempest_plugin.tests.common import vitrage_utils as v_utils from vitrage_tempest_plugin.tests.common import vitrage_utils as v_utils
from vitrage_tempest_plugin.tests.e2e.test_actions_base import TestActionsBase from vitrage_tempest_plugin.tests.e2e.test_actions_base import TestActionsBase
@ -100,11 +99,9 @@ class TestWebhook(TestActionsBase):
try: try:
# Add webhook with filter matching alarm # Add webhook with filter matching alarm
TempestClients.vitrage().webhook.add( self.vitrage_client.webhook.add(url=self.URL_PROPS,
url=self.URL_PROPS, regex_filter=NAME_FILTER,
regex_filter=NAME_FILTER, headers=HEADERS_PROPS)
headers=HEADERS_PROPS
)
# Raise alarm # Raise alarm
self._trigger_do_action(TRIGGER_ALARM_1) self._trigger_do_action(TRIGGER_ALARM_1)
@ -133,10 +130,8 @@ class TestWebhook(TestActionsBase):
try: try:
# Add webhook # Add webhook
TempestClients.vitrage().webhook.add( self.vitrage_client.webhook.add(url=self.URL_PROPS,
url=self.URL_PROPS, regex_filter=NAME_FILTER)
regex_filter=NAME_FILTER,
)
# Raise alarm # Raise alarm
self._trigger_do_action(TRIGGER_ALARM_1) self._trigger_do_action(TRIGGER_ALARM_1)
@ -163,10 +158,8 @@ class TestWebhook(TestActionsBase):
try: try:
# Add webhook # Add webhook
TempestClients.vitrage().webhook.add( self.vitrage_client.webhook.add(url=self.URL_PROPS,
url=self.URL_PROPS, regex_filter=FILTER_NO_MATCH)
regex_filter=FILTER_NO_MATCH,
)
# Raise alarm # Raise alarm
self._trigger_do_action(TRIGGER_ALARM_1) self._trigger_do_action(TRIGGER_ALARM_1)
@ -196,14 +189,10 @@ class TestWebhook(TestActionsBase):
try: try:
# Add webhook # Add webhook
TempestClients.vitrage().webhook.add( self.vitrage_client.webhook.add(url=self.URL_PROPS,
url=self.URL_PROPS, regex_filter=TYPE_FILTER)
regex_filter=TYPE_FILTER,
)
TempestClients.vitrage().webhook.add( self.vitrage_client.webhook.add(url=self.URL_PROPS)
url=self.URL_PROPS
)
# Raise alarm # Raise alarm
self._trigger_do_action(TRIGGER_ALARM_1) self._trigger_do_action(TRIGGER_ALARM_1)
@ -228,11 +217,10 @@ class TestWebhook(TestActionsBase):
try: try:
# Add webhook with filter for the deduced alarm # Add webhook with filter for the deduced alarm
TempestClients.vitrage().webhook.add( self.vitrage_client.webhook.add(
url=self.URL_PROPS, url=self.URL_PROPS,
regex_filter=NAME_FILTER_FOR_DEDUCED, regex_filter=NAME_FILTER_FOR_DEDUCED,
headers=HEADERS_PROPS headers=HEADERS_PROPS)
)
# Raise the trigger alarm # Raise the trigger alarm
self._trigger_do_action(TRIGGER_ALARM_WITH_DEDUCED) self._trigger_do_action(TRIGGER_ALARM_WITH_DEDUCED)
@ -261,10 +249,8 @@ class TestWebhook(TestActionsBase):
try: try:
TempestClients.vitrage().webhook.add( self.vitrage_client.webhook.add(url=self.URL_PROPS,
url=self.URL_PROPS, headers=HEADERS_PROPS)
headers=HEADERS_PROPS
)
# Raise the trigger alarm # Raise the trigger alarm
self._trigger_do_action(TRIGGER_ALARM_1) self._trigger_do_action(TRIGGER_ALARM_1)
@ -319,11 +305,11 @@ class TestWebhook(TestActionsBase):
finally: finally:
self._trigger_undo_action(TRIGGER_ALARM_1) self._trigger_undo_action(TRIGGER_ALARM_1)
@staticmethod @classmethod
def _delete_webhooks(): def _delete_webhooks(cls):
webhooks = TempestClients.vitrage().webhook.list() webhooks = cls.vitrage_client.webhook.list()
for webhook in webhooks: for webhook in webhooks:
TempestClients.vitrage().webhook.delete(webhook['id']) cls.vitrage_client.webhook.delete(webhook['id'])
def _get_free_port(): def _get_free_port():

26
vitrage_tempest_plugin/tests/e2e/test_overlapping_actions.py
View File

@ -169,7 +169,7 @@ class TestOverlappingActions(TestActionsBase):
# ---- Do first & second ---- # ---- Do first & second ----
self._trigger_do_action(TRIGGER_ALARM_1) self._trigger_do_action(TRIGGER_ALARM_1)
self._trigger_do_action(TRIGGER_ALARM_2) self._trigger_do_action(TRIGGER_ALARM_2)
alarms = TempestClients.vitrage().alarm.list( alarms = self.vitrage_client.alarm.list(
vitrage_id=self.orig_host.get(VProps.VITRAGE_ID), vitrage_id=self.orig_host.get(VProps.VITRAGE_ID),
all_tenants=True) all_tenants=True)
@ -178,23 +178,23 @@ class TestOverlappingActions(TestActionsBase):
trigger2 = g_utils.first_match(alarms, **TRIGGER_ALARM_2_PROPS) trigger2 = g_utils.first_match(alarms, **TRIGGER_ALARM_2_PROPS)
# Get Rca for the deduced # Get Rca for the deduced
rca = TempestClients.vitrage().rca.get( rca = self.vitrage_client.rca.get(deduced[VProps.VITRAGE_ID],
deduced[VProps.VITRAGE_ID], all_tenants=True) all_tenants=True)
self._check_rca(rca, [deduced, trigger1, trigger2], DEDUCED_PROPS) self._check_rca(rca, [deduced, trigger1, trigger2], DEDUCED_PROPS)
# Get Rca for trigger 1 # Get Rca for trigger 1
rca = TempestClients.vitrage().rca.get( rca = self.vitrage_client.rca.get(trigger1[VProps.VITRAGE_ID],
trigger1[VProps.VITRAGE_ID], all_tenants=True) all_tenants=True)
self._check_rca(rca, [deduced, trigger1], TRIGGER_ALARM_1_PROPS) self._check_rca(rca, [deduced, trigger1], TRIGGER_ALARM_1_PROPS)
# Get Rca for trigger 2 # Get Rca for trigger 2
rca = TempestClients.vitrage().rca.get( rca = self.vitrage_client.rca.get(trigger2[VProps.VITRAGE_ID],
trigger2[VProps.VITRAGE_ID], all_tenants=True) all_tenants=True)
self._check_rca(rca, [deduced, trigger2], TRIGGER_ALARM_2_PROPS) self._check_rca(rca, [deduced, trigger2], TRIGGER_ALARM_2_PROPS)
# ---- Undo - first ---- # ---- Undo - first ----
self._trigger_undo_action(TRIGGER_ALARM_1) self._trigger_undo_action(TRIGGER_ALARM_1)
alarms = TempestClients.vitrage().alarm.list( alarms = self.vitrage_client.alarm.list(
vitrage_id=self.orig_host.get(VProps.VITRAGE_ID), vitrage_id=self.orig_host.get(VProps.VITRAGE_ID),
all_tenants=True) all_tenants=True)
@ -202,18 +202,18 @@ class TestOverlappingActions(TestActionsBase):
trigger2 = g_utils.first_match(alarms, **TRIGGER_ALARM_2_PROPS) trigger2 = g_utils.first_match(alarms, **TRIGGER_ALARM_2_PROPS)
# Get Rca for the deduced # Get Rca for the deduced
rca = TempestClients.vitrage().rca.get( rca = self.vitrage_client.rca.get(deduced[VProps.VITRAGE_ID],
deduced[VProps.VITRAGE_ID], all_tenants=True) all_tenants=True)
self._check_rca(rca, [deduced, trigger2], DEDUCED_PROPS) self._check_rca(rca, [deduced, trigger2], DEDUCED_PROPS)
# Get Rca for trigger 2 # Get Rca for trigger 2
rca = TempestClients.vitrage().rca.get( rca = self.vitrage_client.rca.get(trigger2[VProps.VITRAGE_ID],
trigger2[VProps.VITRAGE_ID], all_tenants=True) all_tenants=True)
self._check_rca(rca, [deduced, trigger2], TRIGGER_ALARM_2_PROPS) self._check_rca(rca, [deduced, trigger2], TRIGGER_ALARM_2_PROPS)
# ---- Undo - second ---- # ---- Undo - second ----
self._trigger_undo_action(TRIGGER_ALARM_2) self._trigger_undo_action(TRIGGER_ALARM_2)
alarms = TempestClients.vitrage().alarm.list( alarms = self.vitrage_client.alarm.list(
vitrage_id=self.orig_host.get(VProps.VITRAGE_ID), vitrage_id=self.orig_host.get(VProps.VITRAGE_ID),
all_tenants=True) all_tenants=True)
self.assertThat( self.assertThat(

8
vitrage_tempest_plugin/tests/notifiers/test_mistral_notifier.py
View File

@ -178,10 +178,10 @@ class TestMistralNotifier(BaseTestEvents):
num_alarms=num_alarms), num_alarms=num_alarms),
'Vitrage trigger alarm was not deleted') 'Vitrage trigger alarm was not deleted')
@staticmethod @classmethod
def _check_num_vitrage_alarms(num_alarms): def _check_num_vitrage_alarms(cls, num_alarms):
vitrage_alarms = TempestClients.vitrage().alarm.list(vitrage_id='all', vitrage_alarms = cls.vitrage_client.alarm.list(vitrage_id='all',
all_tenants=True) all_tenants=True)
if len(vitrage_alarms) == num_alarms: if len(vitrage_alarms) == num_alarms:
return True return True
return False return False

22
vitrage_tempest_plugin/tests/resources/mock_datasource/test_3rd_degree_scenarios.py
View File

@ -21,7 +21,6 @@ from tempest import config
from vitrage_tempest_plugin.tests.common.constants import VertexProperties from vitrage_tempest_plugin.tests.common.constants import VertexProperties
from vitrage_tempest_plugin.tests.common import general_utils as g_utils from vitrage_tempest_plugin.tests.common import general_utils as g_utils
from vitrage_tempest_plugin.tests.common.tempest_clients import TempestClients
from vitrage_tempest_plugin.tests.common import vitrage_utils as v_utils from vitrage_tempest_plugin.tests.common import vitrage_utils as v_utils
from vitrage_tempest_plugin.tests.e2e.test_actions_base import TestActionsBase from vitrage_tempest_plugin.tests.e2e.test_actions_base import TestActionsBase
from vitrage_tempest_plugin.tests import utils from vitrage_tempest_plugin.tests import utils
@ -86,8 +85,7 @@ class TestLongProcessing(TestActionsBase):
time.sleep(MAX_FAIL_OVER_TIME) time.sleep(MAX_FAIL_OVER_TIME)
doctor_events_thread.join(timeout=10) doctor_events_thread.join(timeout=10)
alarm_count = TempestClients.vitrage().alarm.count( alarm_count = self.vitrage_client.alarm.count(all_tenants=True)
all_tenants=True)
self.assertTrue(self.num_of_sent_events > 0, self.assertTrue(self.num_of_sent_events > 0,
'Test did not create events') 'Test did not create events')
self.assertEqual( self.assertEqual(
@ -106,12 +104,12 @@ class TestLongProcessing(TestActionsBase):
self._check_template_instance_3rd_degree_scenarios() self._check_template_instance_3rd_degree_scenarios()
# 2. check fast fail-over - start from database # 2. check fast fail-over - start from database
topo1 = TempestClients.vitrage().topology.get(all_tenants=True) topo1 = self.vitrage_client.topology.get(all_tenants=True)
v_utils.restart_graph() v_utils.restart_graph()
time.sleep(MAX_FAIL_OVER_TIME) time.sleep(MAX_FAIL_OVER_TIME)
for i in range(5): for i in range(5):
self._check_template_instance_3rd_degree_scenarios() self._check_template_instance_3rd_degree_scenarios()
topo2 = TempestClients.vitrage().topology.get(all_tenants=True) topo2 = self.vitrage_client.topology.get(all_tenants=True)
self.assert_graph_equal( self.assert_graph_equal(
topo1, topo2, 'comparing graph items iteration ' + str(i)) topo1, topo2, 'comparing graph items iteration ' + str(i))
time.sleep(CONF.root_cause_analysis_service.snapshots_interval) time.sleep(CONF.root_cause_analysis_service.snapshots_interval)
@ -122,8 +120,7 @@ class TestLongProcessing(TestActionsBase):
def _check_template_instance_3rd_degree_scenarios(self): def _check_template_instance_3rd_degree_scenarios(self):
alarm_count = TempestClients.vitrage().alarm.count( alarm_count = self.vitrage_client.alarm.count(all_tenants=True)
all_tenants=True)
self.assertEqual( self.assertEqual(
CONF.root_cause_analysis_service.instances_per_host, CONF.root_cause_analysis_service.instances_per_host,
alarm_count['SEVERE'], alarm_count['SEVERE'],
@ -138,8 +135,8 @@ class TestLongProcessing(TestActionsBase):
expected_rca.extend([{'name': DEDUCED_1}, {'name': DEDUCED_2}]) expected_rca.extend([{'name': DEDUCED_1}, {'name': DEDUCED_2}])
def check_rca(alarm): def check_rca(alarm):
rca = TempestClients.vitrage().rca.get(alarm['vitrage_id'], rca = self.vitrage_client.rca.get(alarm['vitrage_id'],
all_tenants=True) all_tenants=True)
try: try:
self._check_rca(rca, expected_rca, alarm) self._check_rca(rca, expected_rca, alarm)
return True return True
@ -148,8 +145,8 @@ class TestLongProcessing(TestActionsBase):
return False return False
# 10 threads calling rca api # 10 threads calling rca api
alarms = TempestClients.vitrage().alarm.list(all_tenants=True, alarms = self.vitrage_client.alarm.list(all_tenants=True,
vitrage_id='all') vitrage_id='all')
deduced_alarms = g_utils.all_matches( deduced_alarms = g_utils.all_matches(
alarms, vitrage_type='vitrage', name=DEDUCED_2) alarms, vitrage_type='vitrage', name=DEDUCED_2)
workers = futures.ThreadPoolExecutor(max_workers=10) workers = futures.ThreadPoolExecutor(max_workers=10)
@ -158,8 +155,7 @@ class TestLongProcessing(TestActionsBase):
self.assertTrue(all(workers_result)) self.assertTrue(all(workers_result))
def _check_template_instance_3rd_degree_scenarios_deleted(self): def _check_template_instance_3rd_degree_scenarios_deleted(self):
alarm_count = TempestClients.vitrage().alarm.count( alarm_count = self.vitrage_client.alarm.count(all_tenants=True)
all_tenants=True)
self.assertEqual( self.assertEqual(
0, 0,
alarm_count['SEVERE'], alarm_count['SEVERE'],