mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-02-04 07:51:50 +00:00
218 lines
9.8 KiB
Python
218 lines
9.8 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 unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from ansible_collections.community.general.plugins.module_utils.source_control.bitbucket import BitbucketHelper
|
|
from ansible_collections.community.general.plugins.modules import bitbucket_pipeline_known_host
|
|
from ansible_collections.community.general.plugins.modules.bitbucket_pipeline_known_host import HAS_PARAMIKO
|
|
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import (
|
|
AnsibleExitJson,
|
|
ModuleTestCase,
|
|
set_module_args,
|
|
)
|
|
|
|
|
|
class TestBucketPipelineKnownHostModule(ModuleTestCase):
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.module = bitbucket_pipeline_known_host
|
|
|
|
@pytest.mark.skipif(not HAS_PARAMIKO, reason="paramiko must be installed to test key creation")
|
|
@patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
|
|
@patch.object(bitbucket_pipeline_known_host, "get_existing_known_host", return_value=None)
|
|
def test_create_known_host(self, *args):
|
|
with patch.object(self.module, "create_known_host") as create_known_host_mock:
|
|
with self.assertRaises(AnsibleExitJson) as exec_info:
|
|
with set_module_args(
|
|
{
|
|
"client_id": "ABC",
|
|
"client_secret": "XXX",
|
|
"workspace": "name",
|
|
"repository": "repo",
|
|
"name": "bitbucket.org",
|
|
"state": "present",
|
|
}
|
|
):
|
|
self.module.main()
|
|
|
|
self.assertEqual(create_known_host_mock.call_count, 1)
|
|
self.assertEqual(exec_info.exception.args[0]["changed"], True)
|
|
|
|
@patch.object(BitbucketHelper, "request", return_value=(dict(status=201), dict()))
|
|
@patch.object(bitbucket_pipeline_known_host, "get_existing_known_host", return_value=None)
|
|
def test_create_known_host_with_key(self, *args):
|
|
with patch.object(self.module, "get_host_key") as get_host_key_mock:
|
|
with self.assertRaises(AnsibleExitJson) as exec_info:
|
|
with set_module_args(
|
|
{
|
|
"user": "ABC",
|
|
"password": "XXX",
|
|
"workspace": "name",
|
|
"repository": "repo",
|
|
"name": "bitbucket.org",
|
|
"key": "ssh-rsa public",
|
|
"state": "present",
|
|
}
|
|
):
|
|
self.module.main()
|
|
|
|
self.assertEqual(get_host_key_mock.call_count, 0)
|
|
self.assertEqual(exec_info.exception.args[0]["changed"], True)
|
|
|
|
@pytest.mark.skipif(not HAS_PARAMIKO, reason="paramiko must be installed to test key creation")
|
|
@patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
|
|
@patch.object(
|
|
bitbucket_pipeline_known_host,
|
|
"get_existing_known_host",
|
|
return_value={
|
|
"type": "pipeline_known_host",
|
|
"uuid": "{21cc0590-bebe-4fae-8baf-03722704119a7}",
|
|
"hostname": "bitbucket.org",
|
|
"public_key": {
|
|
"type": "pipeline_ssh_public_key",
|
|
"md5_fingerprint": "md5:97:8c:1b:f2:6f:14:6b:4b:3b:ec:aa:46:46:74:7c:40",
|
|
"sha256_fingerprint": "SHA256:zzXQOXSFBEiUtuE8AikoYKwbHaxvSc0ojez9YXaGp1A",
|
|
"key_type": "ssh-rsa",
|
|
"key": "AAAAB3NzaC1yc2EAAAABIwAAAQEAubiN81eDcafrgMeLzaFPsw2kN...seeFVBoGqzHM9yXw==",
|
|
},
|
|
},
|
|
)
|
|
def test_dont_create_same_value(self, *args):
|
|
with patch.object(self.module, "create_known_host") as create_known_host_mock:
|
|
with self.assertRaises(AnsibleExitJson) as exec_info:
|
|
with set_module_args(
|
|
{
|
|
"client_id": "ABC",
|
|
"client_secret": "XXX",
|
|
"workspace": "name",
|
|
"repository": "repo",
|
|
"name": "bitbucket.org",
|
|
"state": "present",
|
|
}
|
|
):
|
|
self.module.main()
|
|
|
|
self.assertEqual(create_known_host_mock.call_count, 0)
|
|
self.assertEqual(exec_info.exception.args[0]["changed"], False)
|
|
|
|
@pytest.mark.skipif(not HAS_PARAMIKO, reason="paramiko must be installed to test key creation")
|
|
@patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
|
|
@patch.object(bitbucket_pipeline_known_host, "get_existing_known_host", return_value=None)
|
|
def test_create_known_host_check_mode(self, *args):
|
|
with patch.object(self.module, "create_known_host") as create_known_host_mock:
|
|
with self.assertRaises(AnsibleExitJson) as exec_info:
|
|
with set_module_args(
|
|
{
|
|
"client_id": "ABC",
|
|
"client_secret": "XXX",
|
|
"workspace": "name",
|
|
"repository": "repo",
|
|
"name": "bitbucket.org",
|
|
"state": "present",
|
|
"_ansible_check_mode": True,
|
|
}
|
|
):
|
|
self.module.main()
|
|
|
|
self.assertEqual(create_known_host_mock.call_count, 0)
|
|
self.assertEqual(exec_info.exception.args[0]["changed"], True)
|
|
|
|
@pytest.mark.skipif(not HAS_PARAMIKO, reason="paramiko must be installed to test key creation")
|
|
@patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
|
|
@patch.object(
|
|
bitbucket_pipeline_known_host,
|
|
"get_existing_known_host",
|
|
return_value={
|
|
"type": "pipeline_known_host",
|
|
"uuid": "{21cc0590-bebe-4fae-8baf-03722704119a7}",
|
|
"hostname": "bitbucket.org",
|
|
"public_key": {
|
|
"type": "pipeline_ssh_public_key",
|
|
"md5_fingerprint": "md5:97:8c:1b:f2:6f:14:6b:4b:3b:ec:aa:46:46:74:7c:40",
|
|
"sha256_fingerprint": "SHA256:zzXQOXSFBEiUtuE8AikoYKwbHaxvSc0ojez9YXaGp1A",
|
|
"key_type": "ssh-rsa",
|
|
"key": "AAAAB3NzaC1yc2EAAAABIwAAAQEAubiN81eDcafrgMeLzaFPsw2kN...seeFVBoGqzHM9yXw==",
|
|
},
|
|
},
|
|
)
|
|
def test_delete_known_host(self, *args):
|
|
with patch.object(self.module, "delete_known_host") as delete_known_host_mock:
|
|
with self.assertRaises(AnsibleExitJson) as exec_info:
|
|
with set_module_args(
|
|
{
|
|
"client_id": "ABC",
|
|
"client_secret": "XXX",
|
|
"workspace": "name",
|
|
"repository": "repo",
|
|
"name": "bitbucket.org",
|
|
"state": "absent",
|
|
}
|
|
):
|
|
self.module.main()
|
|
|
|
self.assertEqual(delete_known_host_mock.call_count, 1)
|
|
self.assertEqual(exec_info.exception.args[0]["changed"], True)
|
|
|
|
@pytest.mark.skipif(not HAS_PARAMIKO, reason="paramiko must be installed to test key creation")
|
|
@patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
|
|
@patch.object(bitbucket_pipeline_known_host, "get_existing_known_host", return_value=None)
|
|
def test_delete_absent_known_host(self, *args):
|
|
with patch.object(self.module, "delete_known_host") as delete_known_host_mock:
|
|
with self.assertRaises(AnsibleExitJson) as exec_info:
|
|
with set_module_args(
|
|
{
|
|
"client_id": "ABC",
|
|
"client_secret": "XXX",
|
|
"workspace": "name",
|
|
"repository": "repo",
|
|
"name": "bitbucket.org",
|
|
"state": "absent",
|
|
}
|
|
):
|
|
self.module.main()
|
|
|
|
self.assertEqual(delete_known_host_mock.call_count, 0)
|
|
self.assertEqual(exec_info.exception.args[0]["changed"], False)
|
|
|
|
@pytest.mark.skipif(not HAS_PARAMIKO, reason="paramiko must be installed to test key creation")
|
|
@patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
|
|
@patch.object(
|
|
bitbucket_pipeline_known_host,
|
|
"get_existing_known_host",
|
|
return_value={
|
|
"type": "pipeline_known_host",
|
|
"uuid": "{21cc0590-bebe-4fae-8baf-03722704119a7}",
|
|
"hostname": "bitbucket.org",
|
|
"public_key": {
|
|
"type": "pipeline_ssh_public_key",
|
|
"md5_fingerprint": "md5:97:8c:1b:f2:6f:14:6b:4b:3b:ec:aa:46:46:74:7c:40",
|
|
"sha256_fingerprint": "SHA256:zzXQOXSFBEiUtuE8AikoYKwbHaxvSc0ojez9YXaGp1A",
|
|
"key_type": "ssh-rsa",
|
|
"key": "AAAAB3NzaC1yc2EAAAABIwAAAQEAubiN81eDcafrgMeLzaFPsw2kN...seeFVBoGqzHM9yXw==",
|
|
},
|
|
},
|
|
)
|
|
def test_delete_known_host_check_mode(self, *args):
|
|
with patch.object(self.module, "delete_known_host") as delete_known_host_mock:
|
|
with self.assertRaises(AnsibleExitJson) as exec_info:
|
|
with set_module_args(
|
|
{
|
|
"client_id": "ABC",
|
|
"client_secret": "XXX",
|
|
"workspace": "name",
|
|
"repository": "repo",
|
|
"name": "bitbucket.org",
|
|
"state": "absent",
|
|
"_ansible_check_mode": True,
|
|
}
|
|
):
|
|
self.module.main()
|
|
|
|
self.assertEqual(delete_known_host_mock.call_count, 0)
|
|
self.assertEqual(exec_info.exception.args[0]["changed"], True)
|