AbitraryYu
0caacf870c
* Added Tests for Notification and Search Pipeline.ml Signed-off-by: AbitraryYu <nikkoyhc@gmail.com> * Add back the lines of comment in indices.py during resolving merge conflict. Signed-off-by: AbitraryYu <nikkoyhc@gmail.com> * Fix notification plugin tests, from async def to def. Signed-off-by: AbitraryYu <nikkoyhc@gmail.com> * Fix test errors. Rename function names and rewrite assertion tests. Signed-off-by: AbitraryYu <nikkoyhc@gmail.com> * Fix formatting errors. Added typings to CONTENT. Signed-off-by: AbitraryYu <nikkoyhc@gmail.com> --------- Signed-off-by: AbitraryYu <nikkoyhc@gmail.com>
129 lines
4.2 KiB
Python
129 lines
4.2 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# The OpenSearch Contributors require contributions made to
|
|
# this file be licensed under the Apache-2.0 license or a
|
|
# compatible open source license.
|
|
#
|
|
# Modifications Copyright OpenSearch Contributors. See
|
|
# GitHub history for details.
|
|
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import unittest
|
|
from typing import Any, Dict
|
|
|
|
from opensearchpy.exceptions import NotFoundError
|
|
from opensearchpy.helpers.test import OPENSEARCH_VERSION
|
|
|
|
from .. import OpenSearchTestCase
|
|
|
|
|
|
class TestNotificationPlugin(OpenSearchTestCase):
|
|
CONFIG_ID = "sample-id"
|
|
CONTENT: Dict[str, Any] = {
|
|
"config_id": "sample-id",
|
|
"name": "sample-name",
|
|
"config": {
|
|
"name": "Sample Slack Channel",
|
|
"description": "This ialerting.create_destination(dummy_destination)s a Slack channel",
|
|
"config_type": "slack",
|
|
"is_enabled": True,
|
|
"slack": {"url": "https://sample-slack-webhook"},
|
|
},
|
|
}
|
|
|
|
@unittest.skipUnless(
|
|
(OPENSEARCH_VERSION) and (OPENSEARCH_VERSION >= (2, 0, 0)),
|
|
"Plugin not supported for opensearch version",
|
|
)
|
|
def test_create_config(self) -> None:
|
|
response = self.client.plugins.notifications.create_config(self.CONTENT)
|
|
|
|
self.assertNotIn("errors", response)
|
|
self.assertIn("config_id", response)
|
|
|
|
@unittest.skipUnless(
|
|
(OPENSEARCH_VERSION) and (OPENSEARCH_VERSION >= (2, 0, 0)),
|
|
"Plugin not supported for opensearch version",
|
|
)
|
|
def test_list_channel_config(self) -> None:
|
|
self.test_create_config()
|
|
|
|
response = self.client.plugins.notifications.list_features()
|
|
|
|
self.assertNotIn("errors", response)
|
|
self.assertIn("allowed_config_type_list", response)
|
|
|
|
@unittest.skipUnless(
|
|
(OPENSEARCH_VERSION) and (OPENSEARCH_VERSION >= (2, 0, 0)),
|
|
"Plugin not supported for opensearch version",
|
|
)
|
|
def test_list_all_config(self) -> None:
|
|
self.test_create_config()
|
|
|
|
response = self.client.plugins.notifications.get_configs()
|
|
|
|
self.assertNotIn("errors", response)
|
|
self.assertIn("config_list", response)
|
|
|
|
@unittest.skipUnless(
|
|
(OPENSEARCH_VERSION) and (OPENSEARCH_VERSION >= (2, 0, 0)),
|
|
"Plugin not supported for opensearch version",
|
|
)
|
|
def test_get_config(self) -> None:
|
|
self.test_create_config()
|
|
|
|
response = self.client.plugins.notifications.get_config(
|
|
config_id=self.CONFIG_ID
|
|
)
|
|
|
|
self.assertNotIn("errors", response)
|
|
self.assertIn("config_list", response)
|
|
self.assertEqual(response["config_list"][0]["config_id"], self.CONFIG_ID)
|
|
|
|
@unittest.skipUnless(
|
|
(OPENSEARCH_VERSION) and (OPENSEARCH_VERSION >= (2, 0, 0)),
|
|
"Plugin not supported for opensearch version",
|
|
)
|
|
def test_update_config(self) -> None:
|
|
# Create configuration
|
|
self.test_create_config()
|
|
|
|
# Fetch the configuration
|
|
response = self.client.plugins.notifications.get_config(self.CONFIG_ID)
|
|
|
|
channel_content = self.CONTENT.copy()
|
|
channel_content["config"]["name"] = "Slack Channel"
|
|
channel_content["config"][
|
|
"description"
|
|
] = "This is an updated channel configuration"
|
|
channel_content["config"]["config_type"] = "slack"
|
|
channel_content["config"]["is_enabled"] = True
|
|
channel_content["config"]["slack"]["url"] = "https://hooks.slack.com/sample-url"
|
|
|
|
# Test to updat configuration
|
|
response = self.client.plugins.notifications.update_config(
|
|
config_id=self.CONFIG_ID, body=channel_content
|
|
)
|
|
|
|
self.assertNotIn("errors", response)
|
|
self.assertIn("config_id", response)
|
|
|
|
@unittest.skipUnless(
|
|
(OPENSEARCH_VERSION) and (OPENSEARCH_VERSION >= (2, 0, 0)),
|
|
"Plugin not supported for opensearch version",
|
|
)
|
|
def test_delete_config(self) -> None:
|
|
self.test_create_config()
|
|
|
|
response = self.client.plugins.notifications.delete_config(
|
|
config_id=self.CONFIG_ID
|
|
)
|
|
|
|
self.assertNotIn("errors", response)
|
|
|
|
# Try fetching the policy
|
|
with self.assertRaises(NotFoundError):
|
|
response = self.client.plugins.notifications.get_config(self.CONFIG_ID)
|