ansible/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2026-05-05 17:55:11 +00:00
Code Activity

Reformat everything.

Browse source
This commit is contained in:
Felix Fontein 2025-11-01 12:08:41 +01:00
parent 3f2213791a
commit 340ff8586d
1008 changed files with 61301 additions and 58309 deletions
Download patch file Download diff file Expand all files Collapse all files

41
tests/unit/plugins/modules/test_usb_facts.py
View file

@ -8,25 +8,28 @@ import unittest
from unittest import mock
from ansible.module_utils import basic
from ansible_collections.community.general.plugins.modules import usb_facts
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import AnsibleExitJson, set_module_args, exit_json, fail_json
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import (
AnsibleExitJson,
set_module_args,
exit_json,
fail_json,
)
def get_bin_path(self, arg, required=False):
"""Mock AnsibleModule.get_bin_path"""
if arg == 'lsusb':
return '/usr/bin/lsusb'
if arg == "lsusb":
return "/usr/bin/lsusb"
else:
if required:
fail_json(msg=f'{arg!r} not found !')
fail_json(msg=f"{arg!r} not found !")
class TestUsbFacts(unittest.TestCase):
def setUp(self):
self.mock_module_helper = mock.patch.multiple(basic.AnsibleModule,
exit_json=exit_json,
fail_json=fail_json,
get_bin_path=get_bin_path)
self.mock_module_helper = mock.patch.multiple(
basic.AnsibleModule, exit_json=exit_json, fail_json=fail_json, get_bin_path=get_bin_path
)
self.mock_module_helper.start()
self.addCleanup(self.mock_module_helper.stop)
self.testing_data = [
@ -35,39 +38,43 @@ class TestUsbFacts(unittest.TestCase):
"bus": "001",
"device": "001",
"id": "1d6b:0002",
"name": "Linux Foundation 2.0 root hub"
"name": "Linux Foundation 2.0 root hub",
},
{
"input": "Bus 003 Device 002: ID 8087:8008 Intel Corp. Integrated Rate Matching Hub",
"bus": "003",
"device": "002",
"id": "8087:8008",
"name": "Intel Corp. Integrated Rate Matching Hub"
}
"name": "Intel Corp. Integrated Rate Matching Hub",
},
]
self.output_fields = ["bus", "device", "id", "name"]
def test_parsing_single_line(self):
for data in self.testing_data:
with mock.patch.object(basic.AnsibleModule, 'run_command') as mock_run_command:
with mock.patch.object(basic.AnsibleModule, "run_command") as mock_run_command:
command_output = data["input"]
mock_run_command.return_value = 0, command_output, None
with self.assertRaises(AnsibleExitJson) as result:
with set_module_args({}):
usb_facts.main()
for output_field in self.output_fields:
self.assertEqual(result.exception.args[0]["ansible_facts"]["usb_devices"][0][output_field], data[output_field])
self.assertEqual(
result.exception.args[0]["ansible_facts"]["usb_devices"][0][output_field], data[output_field]
)
def test_parsing_multiple_lines(self):
input = ""
for data in self.testing_data:
input += f"{data['input']}\n"
with mock.patch.object(basic.AnsibleModule, 'run_command') as mock_run_command:
with mock.patch.object(basic.AnsibleModule, "run_command") as mock_run_command:
mock_run_command.return_value = 0, input, None
with self.assertRaises(AnsibleExitJson) as result:
with set_module_args({}):
usb_facts.main()
for index in range(0, len(self.testing_data)):
for output_field in self.output_fields:
self.assertEqual(result.exception.args[0]["ansible_facts"]["usb_devices"][index][output_field],
self.testing_data[index][output_field])
self.assertEqual(
result.exception.args[0]["ansible_facts"]["usb_devices"][index][output_field],
self.testing_data[index][output_field],
)