mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-10-23 04:24:00 -07:00
8f8a0e1d7c
Some checks are pending
EOL CI / EOL Sanity (Ⓐ2.17) (push) Waiting to run
EOL CI / EOL Units (Ⓐ2.17+py3.10) (push) Waiting to run
EOL CI / EOL Units (Ⓐ2.17+py3.12) (push) Waiting to run
EOL CI / EOL Units (Ⓐ2.17+py3.7) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+alpine319+py:azp/posix/1/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+alpine319+py:azp/posix/2/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+alpine319+py:azp/posix/3/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+fedora39+py:azp/posix/1/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+fedora39+py:azp/posix/2/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+fedora39+py:azp/posix/3/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+ubuntu2004+py:azp/posix/1/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+ubuntu2004+py:azp/posix/2/) (push) Waiting to run
EOL CI / EOL I (Ⓐ2.17+ubuntu2004+py:azp/posix/3/) (push) Waiting to run
nox / Run extra sanity tests (push) Waiting to run
* Adjust all __future__ imports: for i in $(grep -REl "__future__.*absolute_import" plugins/ tests/); do sed -e 's/from __future__ import .*/from __future__ import annotations/g' -i $i; done * Remove all UTF-8 encoding specifications for Python source files: for i in $(grep -REl '[-][*]- coding: utf-8 -[*]-' plugins/ tests/); do sed -e '/^# -\*- coding: utf-8 -\*-/d' -i $i; done * Remove __metaclass__ = type: for i in $(grep -REl '__metaclass__ = type' plugins/ tests/); do sed -e '/^__metaclass__ = type/d' -i $i; done
100 lines
4.3 KiB
Python
100 lines
4.3 KiB
Python
# 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
|
|
|
|
from ansible_collections.community.general.plugins.modules import statsd
|
|
from ansible_collections.community.internal_test_tools.tests.unit.compat.mock import patch, MagicMock
|
|
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import AnsibleExitJson, AnsibleFailJson, ModuleTestCase, set_module_args
|
|
|
|
|
|
class FakeStatsD(MagicMock):
|
|
|
|
def incr(self, *args, **kwargs):
|
|
pass
|
|
|
|
def gauge(self, *args, **kwargs):
|
|
pass
|
|
|
|
def close(self, *args, **kwargs):
|
|
pass
|
|
|
|
|
|
class TestStatsDModule(ModuleTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestStatsDModule, self).setUp()
|
|
statsd.HAS_STATSD = True
|
|
self.module = statsd
|
|
|
|
def tearDown(self):
|
|
super(TestStatsDModule, self).tearDown()
|
|
|
|
def patch_udp_statsd_client(self, **kwargs):
|
|
return patch('ansible_collections.community.general.plugins.modules.statsd.udp_statsd_client', autospec=True, **kwargs)
|
|
|
|
def patch_tcp_statsd_client(self, **kwargs):
|
|
return patch('ansible_collections.community.general.plugins.modules.statsd.tcp_statsd_client', autospec=True, **kwargs)
|
|
|
|
def test_udp_without_parameters(self):
|
|
"""Test udp without parameters"""
|
|
with self.patch_udp_statsd_client(side_effect=FakeStatsD) as fake_statsd:
|
|
with self.assertRaises(AnsibleFailJson) as result:
|
|
with set_module_args({}):
|
|
self.module.main()
|
|
|
|
def test_tcp_without_parameters(self):
|
|
"""Test tcp without parameters"""
|
|
with self.patch_tcp_statsd_client(side_effect=FakeStatsD) as fake_statsd:
|
|
with self.assertRaises(AnsibleFailJson) as result:
|
|
with set_module_args({}):
|
|
self.module.main()
|
|
|
|
def test_udp_with_parameters(self):
|
|
"""Test udp with parameters"""
|
|
with self.patch_udp_statsd_client(side_effect=FakeStatsD) as fake_statsd:
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
|
with set_module_args({
|
|
'metric': 'my_counter',
|
|
'metric_type': 'counter',
|
|
'value': 1,
|
|
}):
|
|
self.module.main()
|
|
self.assertEqual(result.exception.args[0]['msg'], 'Sent counter my_counter -> 1 to StatsD')
|
|
self.assertEqual(result.exception.args[0]['changed'], True)
|
|
with self.patch_udp_statsd_client(side_effect=FakeStatsD) as fake_statsd:
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
|
with set_module_args({
|
|
'metric': 'my_gauge',
|
|
'metric_type': 'gauge',
|
|
'value': 3,
|
|
}):
|
|
self.module.main()
|
|
self.assertEqual(result.exception.args[0]['msg'], 'Sent gauge my_gauge -> 3 (delta=False) to StatsD')
|
|
self.assertEqual(result.exception.args[0]['changed'], True)
|
|
|
|
def test_tcp_with_parameters(self):
|
|
"""Test tcp with parameters"""
|
|
with self.patch_tcp_statsd_client(side_effect=FakeStatsD) as fake_statsd:
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
|
with set_module_args({
|
|
'protocol': 'tcp',
|
|
'metric': 'my_counter',
|
|
'metric_type': 'counter',
|
|
'value': 1,
|
|
}):
|
|
self.module.main()
|
|
self.assertEqual(result.exception.args[0]['msg'], 'Sent counter my_counter -> 1 to StatsD')
|
|
self.assertEqual(result.exception.args[0]['changed'], True)
|
|
with self.patch_tcp_statsd_client(side_effect=FakeStatsD) as fake_statsd:
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
|
with set_module_args({
|
|
'protocol': 'tcp',
|
|
'metric': 'my_gauge',
|
|
'metric_type': 'gauge',
|
|
'value': 3,
|
|
}):
|
|
self.module.main()
|
|
self.assertEqual(result.exception.args[0]['msg'], 'Sent gauge my_gauge -> 3 (delta=False) to StatsD')
|
|
self.assertEqual(result.exception.args[0]['changed'], True)
|