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
116 lines
4 KiB
Python
116 lines
4 KiB
Python
# (c) 2015, Michael Scherer <mscherer@redhat.com>
|
|
# Copyright (c) 2017 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 pytest
|
|
|
|
from ansible_collections.community.general.plugins.module_utils import known_hosts
|
|
|
|
|
|
URLS = {
|
|
'ssh://one.example.org/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': 'one.example.org',
|
|
'add_host_key_cmd': " -t rsa one.example.org",
|
|
'port': None,
|
|
},
|
|
'ssh+git://two.example.org/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': 'two.example.org',
|
|
'add_host_key_cmd': " -t rsa two.example.org",
|
|
'port': None,
|
|
},
|
|
'rsync://three.example.org/user/example.git': {
|
|
'is_ssh_url': False,
|
|
'get_fqdn': 'three.example.org',
|
|
'add_host_key_cmd': None, # not called for non-ssh urls
|
|
'port': None,
|
|
},
|
|
'git@four.example.org:user/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': 'four.example.org',
|
|
'add_host_key_cmd': " -t rsa four.example.org",
|
|
'port': None,
|
|
},
|
|
'git+ssh://five.example.org/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': 'five.example.org',
|
|
'add_host_key_cmd': " -t rsa five.example.org",
|
|
'port': None,
|
|
},
|
|
'ssh://six.example.org:21/example.org': {
|
|
# ssh on FTP Port?
|
|
'is_ssh_url': True,
|
|
'get_fqdn': 'six.example.org',
|
|
'add_host_key_cmd': " -t rsa -p 21 six.example.org",
|
|
'port': '21',
|
|
},
|
|
'ssh://[2001:DB8::abcd:abcd]/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': '[2001:DB8::abcd:abcd]',
|
|
'add_host_key_cmd': " -t rsa [2001:DB8::abcd:abcd]",
|
|
'port': None,
|
|
},
|
|
'ssh://[2001:DB8::abcd:abcd]:22/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': '[2001:DB8::abcd:abcd]',
|
|
'add_host_key_cmd': " -t rsa -p 22 [2001:DB8::abcd:abcd]",
|
|
'port': '22',
|
|
},
|
|
'username@[2001:DB8::abcd:abcd]/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': '[2001:DB8::abcd:abcd]',
|
|
'add_host_key_cmd': " -t rsa [2001:DB8::abcd:abcd]",
|
|
'port': None,
|
|
},
|
|
'username@[2001:DB8::abcd:abcd]:path/example.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': '[2001:DB8::abcd:abcd]',
|
|
'add_host_key_cmd': " -t rsa [2001:DB8::abcd:abcd]",
|
|
'port': None,
|
|
},
|
|
'ssh://internal.git.server:7999/repos/repo.git': {
|
|
'is_ssh_url': True,
|
|
'get_fqdn': 'internal.git.server',
|
|
'add_host_key_cmd': " -t rsa -p 7999 internal.git.server",
|
|
'port': '7999',
|
|
},
|
|
}
|
|
|
|
|
|
@pytest.mark.parametrize('url, is_ssh_url', ((k, URLS[k]['is_ssh_url']) for k in sorted(URLS)))
|
|
def test_is_ssh_url(url, is_ssh_url):
|
|
assert known_hosts.is_ssh_url(url) == is_ssh_url
|
|
|
|
|
|
@pytest.mark.parametrize('url, fqdn, port', ((k, URLS[k]['get_fqdn'], URLS[k]['port']) for k in sorted(URLS)))
|
|
def test_get_fqdn_and_port(url, fqdn, port):
|
|
assert known_hosts.get_fqdn_and_port(url) == (fqdn, port)
|
|
|
|
|
|
@pytest.mark.parametrize('fqdn, port, add_host_key_cmd',
|
|
((URLS[k]['get_fqdn'], URLS[k]['port'], URLS[k]['add_host_key_cmd'])
|
|
for k in sorted(URLS) if URLS[k]['is_ssh_url']))
|
|
def test_add_host_key(mocker, fqdn, port, add_host_key_cmd):
|
|
am = mocker.MagicMock()
|
|
|
|
get_bin_path = mocker.MagicMock()
|
|
get_bin_path.return_value = keyscan_cmd = "/custom/path/ssh-keyscan"
|
|
am.get_bin_path = get_bin_path
|
|
|
|
run_command = mocker.MagicMock()
|
|
run_command.return_value = (0, "Needs output, otherwise thinks ssh-keyscan timed out'", "")
|
|
am.run_command = run_command
|
|
|
|
append_to_file = mocker.MagicMock()
|
|
append_to_file.return_value = (None,)
|
|
am.append_to_file = append_to_file
|
|
|
|
mocker.patch('os.path.isdir', return_value=True)
|
|
mocker.patch('os.path.exists', return_value=True)
|
|
|
|
known_hosts.add_host_key(am, fqdn, port=port)
|
|
run_command.assert_called_with(keyscan_cmd + add_host_key_cmd)
|