mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||
|
|
|
||
|
|
from __future__ import absolute_import, division, print_function
|
||
|
|
|
||
|
|
__metaclass__ = type
|
||
|
|
|
||
|
|
from ansible_collections.community.general.plugins.modules.net_tools import dnsimple as dnsimple_module
|
||
|
|
from ansible_collections.community.general.tests.unit.plugins.modules.utils import AnsibleFailJson, ModuleTestCase, set_module_args
|
||
|
|
from ansible_collections.community.general.tests.unit.compat.mock import patch
|
||
|
|
import pytest
|
||
|
|
import sys
|
||
|
|
|
||
|
|
dnsimple = pytest.importorskip('dnsimple')
|
||
|
|
mandatory_py_version = pytest.mark.skipif(
|
||
|
|
sys.version_info < (3, 6),
|
||
|
|
reason='The dnsimple dependency requires python3.6 or higher'
|
||
|
|
)
|
||
|
|
|
||
|
|
from dnsimple import DNSimpleException
|
||
|
|
|
||
|
|
|
||
|
|
class TestDNSimple(ModuleTestCase):
|
||
|
|
"""Main class for testing dnsimple module."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""Setup."""
|
||
|
|
super(TestDNSimple, self).setUp()
|
||
|
|
self.module = dnsimple_module
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
"""Teardown."""
|
||
|
|
super(TestDNSimple, self).tearDown()
|
||
|
|
|
||
|
|
def test_without_required_parameters(self):
|
||
|
|
"""Failure must occurs when all parameters are missing"""
|
||
|
|
with self.assertRaises(AnsibleFailJson):
|
||
|
|
set_module_args({})
|
||
|
|
self.module.main()
|
||
|
|
|
||
|
|
@patch('dnsimple.service.Identity.whoami')
|
||
|
|
def test_account_token(self, mock_whoami):
|
||
|
|
mock_whoami.return_value.data.account = 42
|
||
|
|
ds = self.module.DNSimpleV2('fake', 'fake', True, self.module)
|
||
|
|
self.assertEquals(ds.account, 42)
|
||
|
|
|
||
|
|
@patch('dnsimple.service.Accounts.list_accounts')
|
||
|
|
@patch('dnsimple.service.Identity.whoami')
|
||
|
|
def test_user_token_multiple_accounts(self, mock_whoami, mock_accounts):
|
||
|
|
mock_accounts.return_value.data = [1, 2, 3]
|
||
|
|
mock_whoami.return_value.data.account = None
|
||
|
|
with self.assertRaises(DNSimpleException):
|
||
|
|
self.module.DNSimpleV2('fake', 'fake', True, self.module)
|
||
|
|
|
||
|
|
@patch('dnsimple.service.Accounts.list_accounts')
|
||
|
|
@patch('dnsimple.service.Identity.whoami')
|
||
|
|
def test_user_token_single_account(self, mock_whoami, mock_accounts):
|
||
|
|
mock_accounts.return_value.data = [42]
|
||
|
|
mock_whoami.return_value.data.account = None
|
||
|
|
ds = self.module.DNSimpleV2('fake', 'fake', True, self.module)
|
||
|
|
self.assertEquals(ds.account, 42)
|