mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-07-04 09:38:52 +00:00
add new google_chat module (#12181)
* feat(module): add new `google_chat` module incl. tests # Conflicts: # .github/BOTMETA.yml * fix: address `check_mode` and `diff_mode` feedback * refactor: switch from message_reply_option to create_new_thread * refactor: split webhook_url into separate module parameters * fix: remove unused pytest import * refactor: remove unused `validate_certs` * fix: add type hints Co-authored-by: Felix Fontein <felix@fontein.de> * style: format files once more * fix: move types behind guard to prevent issues on python =< 3.8 Co-authored-by: Felix Fontein <felix@fontein.de> --------- Co-authored-by: Tom Scholz <> Co-authored-by: Felix Fontein <felix@fontein.de>
This commit is contained in:
parent
57739cbbd9
commit
5a9b0ec81f
3 changed files with 431 additions and 0 deletions
178
tests/unit/plugins/modules/test_google_chat.py
Normal file
178
tests/unit/plugins/modules/test_google_chat.py
Normal file
|
|
@ -0,0 +1,178 @@
|
|||
# Copyright (c) Ansible project
|
||||
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||||
# SPDX-License-Identifier: GPL-3.0-or-later
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from unittest.mock import Mock, patch
|
||||
|
||||
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import (
|
||||
AnsibleExitJson,
|
||||
AnsibleFailJson,
|
||||
ModuleTestCase,
|
||||
set_module_args,
|
||||
)
|
||||
|
||||
from ansible_collections.community.general.plugins.modules import google_chat
|
||||
|
||||
SPACE = "SPACE_ID"
|
||||
KEY = "KEY"
|
||||
TOKEN = "TOKEN"
|
||||
BASE = "https://chat.googleapis.com/v1/spaces/SPACE_ID/messages"
|
||||
|
||||
|
||||
def make_response(payload):
|
||||
"""Build a fake fetch_url file-like response whose read() returns JSON text."""
|
||||
mock_response = Mock()
|
||||
mock_response.read.return_value = json.dumps(payload)
|
||||
return mock_response
|
||||
|
||||
|
||||
class TestGoogleChatModule(ModuleTestCase):
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
self.module = google_chat
|
||||
|
||||
def tearDown(self):
|
||||
super().tearDown()
|
||||
|
||||
def test_without_required_parameters(self):
|
||||
"""Failure must occur when all parameters are missing"""
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
with set_module_args({}):
|
||||
self.module.main()
|
||||
|
||||
def test_missing_text(self):
|
||||
"""Failure when connection params are given but text is missing"""
|
||||
with set_module_args({"space": SPACE, "key": KEY, "token": TOKEN}):
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
self.module.main()
|
||||
|
||||
def test_missing_space(self):
|
||||
"""Failure when text is given but space is missing"""
|
||||
with set_module_args({"key": KEY, "token": TOKEN, "text": "test"}):
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
self.module.main()
|
||||
|
||||
def test_successful_message(self):
|
||||
"""tests sending a plain message"""
|
||||
with set_module_args({"space": SPACE, "key": KEY, "token": TOKEN, "text": "test"}):
|
||||
with patch.object(google_chat, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (
|
||||
make_response({"name": "spaces/AAAA/messages/BBBB.BBBB"}),
|
||||
{"status": 200},
|
||||
)
|
||||
with self.assertRaises(AnsibleExitJson) as result:
|
||||
self.module.main()
|
||||
|
||||
self.assertTrue(fetch_url_mock.call_count, 1)
|
||||
call_data = json.loads(fetch_url_mock.call_args[1]["data"])
|
||||
assert call_data["text"] == "test"
|
||||
assert "thread" not in call_data
|
||||
url = fetch_url_mock.call_args[1]["url"]
|
||||
assert url.startswith(BASE + "?")
|
||||
assert "key=KEY" in url
|
||||
assert "token=TOKEN" in url
|
||||
assert "messageReplyOption" not in url
|
||||
assert fetch_url_mock.call_args[1]["method"] == "POST"
|
||||
assert result.exception.args[0]["changed"]
|
||||
assert result.exception.args[0]["name"] == "spaces/AAAA/messages/BBBB.BBBB"
|
||||
|
||||
def test_failed_message(self):
|
||||
"""tests failing to send a message (non-200 response)"""
|
||||
with set_module_args({"space": SPACE, "key": KEY, "token": TOKEN, "text": "test"}):
|
||||
with patch.object(google_chat, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (
|
||||
None,
|
||||
{"status": 404, "msg": "not found", "body": b"NOT_FOUND"},
|
||||
)
|
||||
with self.assertRaises(AnsibleFailJson) as result:
|
||||
self.module.main()
|
||||
|
||||
assert "Google Chat" in result.exception.args[0]["msg"]
|
||||
assert "404" in result.exception.args[0]["msg"]
|
||||
|
||||
def test_message_with_thread(self):
|
||||
"""tests sending a message with a thread_key and reading back the thread name"""
|
||||
with set_module_args({"space": SPACE, "key": KEY, "token": TOKEN, "text": "test", "thread_key": "deploy-1"}):
|
||||
with patch.object(google_chat, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (
|
||||
make_response(
|
||||
{
|
||||
"name": "spaces/AAAA/messages/BBBB.BBBB",
|
||||
"thread": {"name": "spaces/AAAA/threads/CCCC"},
|
||||
}
|
||||
),
|
||||
{"status": 200},
|
||||
)
|
||||
with self.assertRaises(AnsibleExitJson) as result:
|
||||
self.module.main()
|
||||
|
||||
self.assertTrue(fetch_url_mock.call_count, 1)
|
||||
call_data = json.loads(fetch_url_mock.call_args[1]["data"])
|
||||
assert call_data["text"] == "test"
|
||||
assert call_data["thread"]["threadKey"] == "deploy-1"
|
||||
assert result.exception.args[0]["thread_name"] == "spaces/AAAA/threads/CCCC"
|
||||
|
||||
def test_create_new_thread_false_appends_reply_or_fail(self):
|
||||
"""create_new_thread=false must map to REPLY_MESSAGE_OR_FAIL in the URL"""
|
||||
with set_module_args(
|
||||
{
|
||||
"space": SPACE,
|
||||
"key": KEY,
|
||||
"token": TOKEN,
|
||||
"text": "test",
|
||||
"thread_key": "deploy-1",
|
||||
"create_new_thread": False,
|
||||
}
|
||||
):
|
||||
with patch.object(google_chat, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (
|
||||
make_response({"name": "spaces/AAAA/messages/BBBB.BBBB"}),
|
||||
{"status": 200},
|
||||
)
|
||||
with self.assertRaises(AnsibleExitJson):
|
||||
self.module.main()
|
||||
|
||||
url = fetch_url_mock.call_args[1]["url"]
|
||||
assert "messageReplyOption=REPLY_MESSAGE_OR_FAIL" in url
|
||||
|
||||
def test_check_mode(self):
|
||||
"""check mode reports changed and never calls the API"""
|
||||
with set_module_args({"space": SPACE, "key": KEY, "token": TOKEN, "text": "test", "_ansible_check_mode": True}):
|
||||
with patch.object(google_chat, "fetch_url") as fetch_url_mock:
|
||||
with self.assertRaises(AnsibleExitJson) as result:
|
||||
self.module.main()
|
||||
|
||||
fetch_url_mock.assert_not_called()
|
||||
assert result.exception.args[0]["changed"]
|
||||
assert "name" not in result.exception.args[0]
|
||||
|
||||
|
||||
def test_build_payload_without_thread():
|
||||
payload = google_chat.build_payload("hello", None)
|
||||
assert payload == {"text": "hello"}
|
||||
|
||||
|
||||
def test_build_payload_with_thread():
|
||||
payload = google_chat.build_payload("hello", "deploy-1")
|
||||
assert payload == {"text": "hello", "thread": {"threadKey": "deploy-1"}}
|
||||
|
||||
|
||||
def test_build_url_without_thread():
|
||||
url = google_chat.build_url(SPACE, KEY, TOKEN, None, True)
|
||||
assert url.startswith(BASE + "?")
|
||||
assert "key=KEY" in url
|
||||
assert "token=TOKEN" in url
|
||||
assert "messageReplyOption" not in url
|
||||
|
||||
|
||||
def test_build_url_create_new_thread_true():
|
||||
url = google_chat.build_url(SPACE, KEY, TOKEN, "deploy-1", True)
|
||||
assert "messageReplyOption=REPLY_MESSAGE_FALLBACK_TO_NEW_THREAD" in url
|
||||
|
||||
|
||||
def test_build_url_create_new_thread_false():
|
||||
url = google_chat.build_url(SPACE, KEY, TOKEN, "deploy-1", False)
|
||||
assert "messageReplyOption=REPLY_MESSAGE_OR_FAIL" in url
|
||||
Loading…
Add table
Add a link
Reference in a new issue