mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
6298ad4faa
* First implementation of usb module * Rename to usb_facts * Working tests * Appease linter * Fix author * Remove splitlines arg for python 2.7 compat * indent usb test further * indent usb test a bit less * Remove trailing whitespace * Update plugins/modules/usb_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Update plugins/modules/usb_facts.py Co-authored-by: Felix Fontein <felix@fontein.de> * Apply suggestions from code review Co-authored-by: Felix Fontein <felix@fontein.de> * Adjust usb_facts PR based on feedback * Add usage example and extend correct documentation fragment * Add myself to BOTMETA.yml * Apply suggestions from code review Co-authored-by: Felix Fontein <felix@fontein.de> --------- Co-authored-by: Felix Fontein <felix@fontein.de>
105 lines
4.1 KiB
Python
105 lines
4.1 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 (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import json
|
|
|
|
from ansible_collections.community.general.tests.unit.compat import mock
|
|
from ansible_collections.community.general.tests.unit.compat import unittest
|
|
from ansible.module_utils import basic
|
|
from ansible.module_utils.common.text.converters import to_bytes
|
|
from ansible_collections.community.general.plugins.modules import usb_facts
|
|
|
|
|
|
def set_module_args(args):
|
|
"""prepare arguments so that they will be picked up during module creation"""
|
|
args = json.dumps({'ANSIBLE_MODULE_ARGS': args})
|
|
basic._ANSIBLE_ARGS = to_bytes(args)
|
|
|
|
|
|
class AnsibleExitJson(Exception):
|
|
"""Exception class to be raised by module.exit_json and caught by the test case"""
|
|
pass
|
|
|
|
|
|
class AnsibleFailJson(Exception):
|
|
"""Exception class to be raised by module.fail_json and caught by the test case"""
|
|
pass
|
|
|
|
|
|
def exit_json(*args, **kwargs):
|
|
"""function to patch over exit_json; package return data into an exception"""
|
|
if 'changed' not in kwargs:
|
|
kwargs['changed'] = False
|
|
raise AnsibleExitJson(kwargs)
|
|
|
|
|
|
def fail_json(*args, **kwargs):
|
|
"""function to patch over fail_json; package return data into an exception"""
|
|
kwargs['failed'] = True
|
|
raise AnsibleFailJson(kwargs)
|
|
|
|
|
|
def get_bin_path(self, arg, required=False):
|
|
"""Mock AnsibleModule.get_bin_path"""
|
|
if arg == 'lsusb':
|
|
return '/usr/bin/lsusb'
|
|
else:
|
|
if required:
|
|
fail_json(msg='%r not found !' % arg)
|
|
|
|
|
|
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.start()
|
|
self.addCleanup(self.mock_module_helper.stop)
|
|
self.testing_data = [
|
|
{
|
|
"input": "Bus 001 Device 001: ID 1d6b:0002 Linux Foundation 2.0 root hub",
|
|
"bus": "001",
|
|
"device": "001",
|
|
"id": "1d6b:0002",
|
|
"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"
|
|
}
|
|
]
|
|
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:
|
|
command_output = data["input"]
|
|
mock_run_command.return_value = 0, command_output, None
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
|
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])
|
|
|
|
def test_parsing_multiple_lines(self):
|
|
input = ""
|
|
for data in self.testing_data:
|
|
input += ("%s\n" % data["input"])
|
|
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:
|
|
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])
|