You've already forked community.general
mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-07-30 18:43:09 +03:00
Unflatmap community.general (#5461)
* Move files. * Update imports and references. * Move wrongly placed files. * Reverse redirects, deprecate long → short name redirects. * Simplify contribution guidelines for new modules. * Rewrite BOTMETA. * Add changelog fragment. * Fix ignore.txt files.
This commit is contained in:
105
tests/unit/plugins/modules/test_discord.py
Normal file
105
tests/unit/plugins/modules/test_discord.py
Normal file
@ -0,0 +1,105 @@
|
||||
# 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 (absolute_import, division, print_function)
|
||||
__metaclass__ = type
|
||||
|
||||
import json
|
||||
import pytest
|
||||
from ansible_collections.community.general.tests.unit.compat.mock import Mock, patch
|
||||
from ansible_collections.community.general.plugins.modules import discord
|
||||
from ansible_collections.community.general.tests.unit.plugins.modules.utils import AnsibleExitJson, AnsibleFailJson, ModuleTestCase, set_module_args
|
||||
|
||||
|
||||
class TestDiscordModule(ModuleTestCase):
|
||||
|
||||
def setUp(self):
|
||||
super(TestDiscordModule, self).setUp()
|
||||
self.module = discord
|
||||
|
||||
def tearDown(self):
|
||||
super(TestDiscordModule, self).tearDown()
|
||||
|
||||
@pytest.fixture
|
||||
def fetch_url_mock(self, mocker):
|
||||
return mocker.patch('ansible.module_utils.notification.discord.fetch_url')
|
||||
|
||||
def test_without_parameters(self):
|
||||
"""Failure if no parameters set"""
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
set_module_args({})
|
||||
self.module.main()
|
||||
|
||||
def test_without_content(self):
|
||||
"""Failure if content and embeds both are missing"""
|
||||
set_module_args({
|
||||
'webhook_id': 'xxx',
|
||||
'webhook_token': 'xxx'
|
||||
})
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
self.module.main()
|
||||
|
||||
def test_successful_message(self):
|
||||
"""Test a basic message successfully."""
|
||||
set_module_args({
|
||||
'webhook_id': 'xxx',
|
||||
'webhook_token': 'xxx',
|
||||
'content': 'test'
|
||||
})
|
||||
|
||||
with patch.object(discord, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (None, {"status": 204, 'msg': 'OK (0 bytes)'})
|
||||
with self.assertRaises(AnsibleExitJson):
|
||||
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['content'] == "test"
|
||||
|
||||
def test_message_with_username(self):
|
||||
"""Test a message with username set successfully."""
|
||||
set_module_args({
|
||||
'webhook_id': 'xxx',
|
||||
'webhook_token': 'xxx',
|
||||
'content': 'test',
|
||||
'username': 'Ansible Bot'
|
||||
})
|
||||
|
||||
with patch.object(discord, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (None, {"status": 204, 'msg': 'OK (0 bytes)'})
|
||||
with self.assertRaises(AnsibleExitJson):
|
||||
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['username'] == "Ansible Bot"
|
||||
assert call_data['content'] == "test"
|
||||
|
||||
def test_failed_message(self):
|
||||
"""Test failure because webhook id is wrong."""
|
||||
|
||||
set_module_args({
|
||||
'webhook_id': 'wrong',
|
||||
'webhook_token': 'xxx',
|
||||
'content': 'test'
|
||||
})
|
||||
|
||||
with patch.object(discord, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (None, {"status": 404, 'msg': 'HTTP Error 404: Not Found', 'body': '{"message": "Unknown Webhook", "code": 10015}'})
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
self.module.main()
|
||||
|
||||
def test_failed_message_without_body(self):
|
||||
"""Test failure with empty response body."""
|
||||
|
||||
set_module_args({
|
||||
'webhook_id': 'wrong',
|
||||
'webhook_token': 'xxx',
|
||||
'content': 'test'
|
||||
})
|
||||
|
||||
with patch.object(discord, "fetch_url") as fetch_url_mock:
|
||||
fetch_url_mock.return_value = (None, {"status": 404, 'msg': 'HTTP Error 404: Not Found'})
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
self.module.main()
|
Reference in New Issue
Block a user